11 from .mock_discovery import get_rootDesc
13 class TestMakeOutput(unittest.TestCase):
15 self.api = mock.MagicMock()
16 self.api._rootDesc = get_rootDesc()
18 @mock.patch("arvados.collection.Collection")
19 @mock.patch("arvados.collection.CollectionReader")
20 def test_make_output_collection(self, reader, col):
21 keep_client = mock.MagicMock()
22 runner = arvados_cwl.ArvCwlRunner(self.api, keep_client=keep_client)
23 runner.project_uuid = 'zzzzz-j7d0g-zzzzzzzzzzzzzzz'
25 final = mock.MagicMock()
26 col.return_value = final
27 readermock = mock.MagicMock()
28 reader.return_value = readermock
30 cwlout = StringIO.StringIO()
31 openmock = mock.MagicMock()
32 final.open.return_value = openmock
33 openmock.__enter__.return_value = cwlout
35 _, runner.final_output_collection = runner.make_output_collection("Test output", {
38 "location": "keep:99999999999999999999999999999991+99/foo.txt",
44 "location": "keep:99999999999999999999999999999992+99/bar.txt",
45 "basename": "baz.txt",
50 final.copy.assert_has_calls([mock.call('bar.txt', 'baz.txt', overwrite=False, source_collection=readermock)])
51 final.copy.assert_has_calls([mock.call('foo.txt', 'foo.txt', overwrite=False, source_collection=readermock)])
52 final.save_new.assert_has_calls([mock.call(ensure_unique_name=True, name='Test output', owner_uuid='zzzzz-j7d0g-zzzzzzzzzzzzzzz')])
56 "location": "baz.txt",
61 "location": "foo.txt",
64 }""", cwlout.getvalue())
66 self.assertIs(final, runner.final_output_collection)