adapter/acumos/tests/test_docker.py

   1 # ============LICENSE_START====================================================
   2 # org.onap.dcae
   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
   9 #
  10 #      http://www.apache.org/licenses/LICENSE-2.0
  11 #
  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======================================================
  18
  19 from testing_helpers import get_json_fixture, get_fixture_path
  20 from aoconversion import docker_gen, scanner
  21 import test_fed
  22
  23 TEST_META = get_json_fixture("models/example-model/metadata.json")
  24
  25
  26 def test_generate_dockerfile():
  27     assert (
  28         docker_gen._generate_dockerfile(TEST_META, "example-model")
  29         == """
  30     FROM python:3.6.8
  31
  32     ENV MODELNAME example-model
  33     RUN mkdir /app
  34     WORKDIR /app
  35
  36     ADD ./example-model /app/example-model
  37     ADD ./requirements.txt /app
  38
  39     RUN pip install -r /app/requirements.txt && \
  40         pip install acumos_dcae_model_runner
  41
  42     ENV DCAEPORT=10000
  43     EXPOSE $DCAEPORT
  44
  45     ENTRYPOINT ["acumos_dcae_model_runner"]
  46     CMD ["/app/example-model"]
  47     """
  48     )
  49
  50
  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')