comparison planemo/lib/python3.7/site-packages/cwltool/tests/test_trs.py @ 0:d30785e31577 draft

"planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"
author guerler
date Fri, 31 Jul 2020 00:18:57 -0400
parents
children
comparison
equal deleted inserted replaced
-1:000000000000 0:d30785e31577
1 from __future__ import absolute_import
2
3 import mock
4
5 from cwltool.main import main
6 from .util import get_data
7
8 def mocked_requests_head(*args):
9
10 class MockResponse:
11 def __init__(self, json_data, status_code, raise_for_status=None):
12 self.json_data = json_data
13 self.status_code = status_code
14 self.raise_for_status = mock.Mock()
15 self.raise_for_status.side_effect = raise_for_status
16
17 def json(self):
18 return self.json_data
19
20 return MockResponse(None, 200)
21
22
23 def mocked_requests_get(*args):
24
25 class MockResponse:
26 def __init__(self, json_data, status_code, raise_for_status=None):
27 self.json_data = json_data
28 self.text = json_data
29 self.status_code = status_code
30 self.raise_for_status = mock.Mock()
31 self.raise_for_status.side_effect = raise_for_status
32
33 def json(self):
34 return self.json_data
35
36 if args[0] == 'https://dockstore.org/api/api/ga4gh/v2/tools/quay.io%2Fbriandoconnor%2Fdockstore-tool-md5sum/versions/1.0.4/CWL/files':
37 return MockResponse(
38 [{"file_type": "CONTAINERFILE", "path": "Dockerfile"}, {"file_type": "PRIMARY_DESCRIPTOR", "path": "Dockstore.cwl"},
39 {"file_type": "TEST_FILE", "path": "test.json"}], 200)
40 elif args[0] == 'https://dockstore.org/api/api/ga4gh/v2/tools/quay.io%2Fbriandoconnor%2Fdockstore-tool-md5sum/versions/1.0.4/plain-CWL/descriptor/Dockstore.cwl':
41 string = open(get_data("tests/trs/Dockstore.cwl"), "r").read()
42 return MockResponse(string, 200)
43 elif args[0] == 'https://dockstore.org/api/api/ga4gh/v2/tools/%23workflow%2Fgithub.com%2Fdockstore-testing%2Fmd5sum-checker/versions/develop/plain-CWL/descriptor/md5sum-tool.cwl':
44 string = open(get_data("tests/trs/md5sum-tool.cwl"), "r").read()
45 return MockResponse(string, 200)
46 elif args[0] == 'https://dockstore.org/api/api/ga4gh/v2/tools/%23workflow%2Fgithub.com%2Fdockstore-testing%2Fmd5sum-checker/versions/develop/plain-CWL/descriptor/md5sum-workflow.cwl':
47 string = open(get_data("tests/trs/md5sum-workflow.cwl"), "r").read()
48 return MockResponse(string, 200)
49 elif args[
50 0] == 'https://dockstore.org/api/api/ga4gh/v2/tools/%23workflow%2Fgithub.com%2Fdockstore-testing%2Fmd5sum-checker/versions/develop/CWL/files':
51 return MockResponse(
52 [{"file_type": "TEST_FILE", "path": "md5sum-input-cwl.json"}, {"file_type": "SECONDARY_DESCRIPTOR", "path": "md5sum-tool.cwl"},
53 {"file_type": "PRIMARY_DESCRIPTOR", "path": "md5sum-workflow.cwl"}], 200)
54
55 print ("A mocked call to TRS missed, target was %s", args[0])
56 return MockResponse(None, 404)
57
58
59 @mock.patch('requests.Session.head', side_effect=mocked_requests_head)
60 @mock.patch('requests.Session.get', side_effect=mocked_requests_get)
61 def test_tool_trs_template(mock_head, mock_get):
62 params = ["--make-template", r"quay.io/briandoconnor/dockstore-tool-md5sum:1.0.4"]
63 return_value = main(params)
64 mock_head.assert_called()
65 mock_get.assert_called()
66 assert return_value == 0
67
68
69 @mock.patch('requests.Session.head', side_effect=mocked_requests_head)
70 @mock.patch('requests.Session.get', side_effect=mocked_requests_get)
71 def test_workflow_trs_template(mock_head, mock_get):
72 params = ["--make-template", r"#workflow/github.com/dockstore-testing/md5sum-checker:develop"]
73 return_value = main(params)
74 mock_head.assert_called()
75 mock_get.assert_called()
76 assert return_value == 0