1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
# Copyright 2018 The Chromium Authors
# 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):
# pylint: disable=super-with-arguments
def __init__(self, methodName, module):
super(CommandLineTest, self).__init__(methodName)
self._module = module
# pylint: disable=super-with-arguments
def setUp(self):
self.temp_dir = tempfile.mkdtemp(prefix='common_merge_script_tests')
def tearDown(self):
shutil.rmtree(self.temp_dir)
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')
os.makedirs(shard0_dir)
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,
shard0_json,
]
self.assertEqual(0, self._module.main(raw_args))
|
