blob: fd56905fbc412db19be9b85440287d0ea3fbfb5f [file] [log] [blame]
# Copyright 2018 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import json
import os
import shutil
import tempfile
import unittest
class CommandLineTest(unittest.TestCase):
def __init__(self, methodName, module):
super(CommandLineTest, self).__init__(methodName)
self._module = module
def setUp(self):
self.temp_dir = tempfile.mkdtemp(prefix='common_merge_script_tests')
def tearDown(self):
def test_accepts_task_output_dir(self):
task_output_dir = os.path.join(self.temp_dir, 'task_output_dir')
shard0_dir = os.path.join(task_output_dir, '0')
summary_json = os.path.join(task_output_dir, 'summary.json')
with open(summary_json, 'w') as summary_file:
summary_contents = {
u'shards': [
u'state': u'COMPLETED',
json.dump(summary_contents, summary_file)
shard0_json = os.path.join(shard0_dir, 'output.json')
with open(shard0_json, 'w') as shard0_file:
json.dump({}, shard0_file)
output_json = os.path.join(self.temp_dir, 'merged.json')
raw_args = [
'--task-output-dir', task_output_dir,
'--summary-json', summary_json,
'--output-json', output_json,
self.assertEquals(0, self._module.main(raw_args))