1 # ============LICENSE_START====================================================
3 # =============================================================================
4 # Copyright (c) 2019-2020 AT&T Intellectual Property. All rights reserved.
5 # =============================================================================
6 # Licensed under the Apache License, Version 2.0 (the "License");
7 # you may not use this file except in compliance with the License.
8 # You may obtain a copy of the License at
10 # http://www.apache.org/licenses/LICENSE-2.0
12 # Unless required by applicable law or agreed to in writing, software
13 # distributed under the License is distributed on an "AS IS" BASIS,
14 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 # See the License for the specific language governing permissions and
16 # limitations under the License.
17 # ============LICENSE_END======================================================
22 from testing_helpers import get_fixture_path as get_test_file
24 from aoconversion import docker_gen as aoc_docker_gen
25 from aoconversion import scanner as aoc_scanner
34 Class to mock packages.
35 Just a quick way to create an object with specified attributes
37 def __init__(self, **kwargs):
38 for k, v in kwargs.items():
43 # Mocking for package "docker"
49 Class to mock docker.APIClient class.
51 def __init__(self, base_url, version=None, user_agent='xxx'):
54 def build(self, path, rm, tag):
55 return [b'some message', b'another message']
57 def push(self, repository, auth_config, stream):
58 return [b'some message', b'another message']
64 def _mock_kwargs_from_env(**kwargs):
66 Method to mock docker.utils.kwargs_from_env method.
68 return {'base_url': None}
71 _mockdocker = _MockModule(
72 APIClient=_MockAPIClient,
73 utils=_MockModule(kwargs_from_env=_mock_kwargs_from_env))
77 # Mocking for requests.get
83 Fake responses for mocking requests.get
85 def __init__(self, json=None, file=None, data=None):
89 self.status_code = 200
93 return self._raw().decode()
95 def raise_for_status(self):
100 if self.fx is not None:
101 with open(self.fx, 'rb') as f:
103 elif self.jx is not None:
104 self.dx = json.dumps(self.jx, sort_keys=True).encode()
109 def iter_content(self, bsize=-1):
115 while pos + bsize < lim:
116 yield buf[pos:pos + bsize]
122 self.jx = json.loads(self._raw().decode())
126 def _mockwww(responses):
127 def _op(path, json=None, auth=None, cert=None, verify=None, stream=False):
128 return responses[path]
133 'https://onboarding/dataformats': _r({'dataFormatUrl': 'https://onboarding/dataformats/somedfid'}),
134 'https://onboarding/components': _r({'componentUrl': 'https://onboarding/components/somedxid'}),
138 'https://onboarding/dataformats/somedfid': _r({}),
139 'https://onboarding/components/somedxid': _r({}),
143 'https://acumos/catalogs': _r({'content': [{'catalogId': 'c1'}]}),
144 'https://acumos/solutions?catalogId=c1': _r({'content': [{'solutionId': 's1', 'name': 'example-model', 'ratingAverageTenths': 17}]}),
145 'https://acumos/solutions/s1/revisions': _r({'content': [{'revisionId': 'r1'}]}),
146 'https://acumos/solutions/s1/revisions/r1': _r({'content': {
148 'modified': '2019-01-01T00:00:00Z',
150 {'artifactId': 'a1', 'name': 'xxx.other'},
151 {'artifactId': 'a2', 'name': 'xxx.proto'},
152 {'artifactId': 'a3', 'name': 'xxx.zip'},
153 {'artifactId': 'a4', 'name': 'xxx.json'},
156 'https://acumos/artifacts/a2/content': _r(file=get_test_file('models/example-model/model.proto')),
157 'https://acumos/artifacts/a3/content': _r(data=b'dummy zip archive data'),
158 'https://acumos/artifacts/a4/content': _r(file=get_test_file('models/example-model/metadata.json')),
167 def test_aoconversion(mock_schemas, tmpdir, monkeypatch):
168 config = aoc_scanner.Config(dcaeurl='http://dcaeurl', dcaeuser='dcaeuser', onboardingurl='https://onboarding', onboardinguser='obuser', onboardingpass='obpass', acumosurl='https://acumos', certfile=None, dockerregistry='dockerregistry', dockeruser='registryuser', dockerpass='registrypassword')
169 monkeypatch.setattr(aoc_docker_gen, 'APIClient', _mockdocker.APIClient)
170 monkeypatch.setattr(requests, 'get', _mockwww(_mockwebdata))
171 monkeypatch.setattr(requests, 'post', _mockwww(_mockpostdata))
172 monkeypatch.setattr(requests, 'patch', _mockwww(_mockpatchdata))
173 aoc_scanner.scan(config)
174 aoc_scanner.scan(config)