1 # ============LICENSE_START====================================================
3 # =============================================================================
4 # Copyright (c) 2019 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======================================================
19 from testing_helpers import get_json_fixture, get_fixture_path
20 from aoconversion import docker_gen, scanner
23 TEST_META = get_json_fixture("models/example-model/metadata.json")
26 def test_generate_dockerfile():
28 docker_gen._generate_dockerfile(TEST_META, "example-model")
32 ENV MODELNAME example-model
36 ADD ./example-model /app/example-model
37 ADD ./requirements.txt /app
39 RUN pip install -r /app/requirements.txt && \
40 pip install acumos_dcae_model_runner
45 ENTRYPOINT ["acumos_dcae_model_runner"]
46 CMD ["/app/example-model"]
51 def test_build_and_push_docker(monkeypatch):
52 model_repo_path = get_fixture_path('models')
53 config = 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', tmpdir=model_repo_path)
54 monkeypatch.setattr(docker_gen, 'APIClient', test_fed._mockdocker.APIClient)
55 assert(docker_gen.build_and_push_docker(config, 'example-model', model_version="latest") == 'dockerregistry/example-model:latest')