From 996dcb989c8c6c9bedf7b764578d1f2e196b16f4 Mon Sep 17 00:00:00 2001 From: jctanner Date: Thu, 22 Jun 2017 13:49:58 -0400 Subject: [PATCH] Add the cloud provider script for the vcenter simulator container (#25964) --- .../targets/vmware_guest_find/aliases | 3 + .../targets/vmware_guest_find/tasks/main.yml | 56 +++++++ test/runner/lib/cloud/vcenter.py | 152 ++++++++++++++++++ 3 files changed, 211 insertions(+) create mode 100644 test/integration/targets/vmware_guest_find/aliases create mode 100644 test/integration/targets/vmware_guest_find/tasks/main.yml create mode 100644 test/runner/lib/cloud/vcenter.py diff --git a/test/integration/targets/vmware_guest_find/aliases b/test/integration/targets/vmware_guest_find/aliases new file mode 100644 index 00000000000..b26008ed574 --- /dev/null +++ b/test/integration/targets/vmware_guest_find/aliases @@ -0,0 +1,3 @@ +posix/ci/cloud/vcenter +cloud/vcenter +skip/python3 diff --git a/test/integration/targets/vmware_guest_find/tasks/main.yml b/test/integration/targets/vmware_guest_find/tasks/main.yml new file mode 100644 index 00000000000..526d47cfb5e --- /dev/null +++ b/test/integration/targets/vmware_guest_find/tasks/main.yml @@ -0,0 +1,56 @@ +- name: make sure pyvmomi is installed + pip: + name: pyvmomi + state: latest + +- name: store the vcenter container ip + set_fact: + vcsim: "{{ lookup('env', 'vcenter_host') }}" +- debug: var=vcsim + +- name: kill vcsim + uri: + url: "{{ 'http://' + vcsim + ':5000/killall' }}" +- name: start vcsim + uri: + url: "{{ 'http://' + vcsim + ':5000/spawn?cluster=2' }}" + register: vcsim_instance + +- name: get a list of VMS from vcsim + uri: + url: "{{ 'http://' + vcsim + ':5000/govc_find?filter=_VM' }}" + register: vmlist + +- debug: var=vcsim_instance +- debug: var=vmlist + +# vcsim embeds the datacenter into the path and +# the VM name, so it can be chopped out for later +# use or verification +- name: show the datacenter(s) + debug: + msg: "{{ (item|basename).split('_')[0] }}" + with_items: "{{ vmlist['json'] }}" + +- name: find folders for each vm + vmware_guest_find: + validate_certs: False + hostname: "{{ vcsim }}" + username: "{{ vcsim_instance['json']['username'] }}" + password: "{{ vcsim_instance['json']['password'] }}" + name: "{{ item|basename }}" + datacenter: "{{ (item|basename).split('_')[0] }}" + with_items: "{{ vmlist['json'] }}" + register: folders + +- debug: var=item + with_items: "{{ folders.results }}" + +# We only care that each VM was found, not that the folder path +# is completely accurate. Eventually the test should be extended +# to validate the full path for each VM. +- assert: + that: + - "{{ 'folders' in item }}" + - "{{ item['folders']|length == 1 }}" + with_items: "{{ folders.results }}" diff --git a/test/runner/lib/cloud/vcenter.py b/test/runner/lib/cloud/vcenter.py new file mode 100644 index 00000000000..cd312b98596 --- /dev/null +++ b/test/runner/lib/cloud/vcenter.py @@ -0,0 +1,152 @@ +"""VMware vCenter plugin for integration tests.""" +from __future__ import absolute_import, print_function + +import os + +from lib.cloud import ( + CloudProvider, + CloudEnvironment, +) + +from lib.util import ( + find_executable, + display, +) + +from lib.docker_util import ( + docker_run, + docker_rm, + docker_inspect, + docker_pull, + get_docker_container_id, +) + +try: + # noinspection PyPep8Naming + import ConfigParser as configparser +except ImportError: + # noinspection PyUnresolvedReferences + import configparser + + +class VcenterProvider(CloudProvider): + """VMware vcenter/esx plugin. Sets up cloud resources for tests.""" + DOCKER_SIMULATOR_NAME = 'vcenter-simulator' + + def __init__(self, args): + """ + :type args: TestConfig + """ + super(VcenterProvider, self).__init__(args, config_extension='.ini') + + self.image = 'ansible/ansible:vcenter-simulator' + self.container_name = '' + + def filter(self, targets, exclude): + """Filter out the cloud tests when the necessary config and resources are not available. + :type targets: tuple[TestTarget] + :type exclude: list[str] + """ + if os.path.isfile(self.config_static_path): + return + + docker = find_executable('docker') + + if docker: + return + + super(VcenterProvider, self).filter(targets, exclude) + + def setup(self): + """Setup the cloud resource before delegation and register a cleanup callback.""" + super(VcenterProvider, self).setup() + + if self._use_static_config(): + self._setup_static() + else: + self._setup_dynamic() + + def get_docker_run_options(self): + """Get any additional options needed when delegating tests to a docker container. + :rtype: list[str] + """ + if self.managed: + return ['--link', self.DOCKER_SIMULATOR_NAME] + + return [] + + def cleanup(self): + """Clean up the cloud resource and any temporary configuration files after tests complete.""" + if self.container_name: + docker_rm(self.args, self.container_name) + + super(VcenterProvider, self).cleanup() + + def _setup_dynamic(self): + """Create a vcenter simulator using docker.""" + container_id = get_docker_container_id() + + if container_id: + display.info('Running in docker container: %s' % container_id, verbosity=1) + + self.container_name = self.DOCKER_SIMULATOR_NAME + + results = docker_inspect(self.args, self.container_name) + + if results and not results[0].get('State', {}).get('Running'): + docker_rm(self.args, self.container_name) + results = [] + + if results: + display.info('Using the existing vCenter simulator docker container.', verbosity=1) + else: + display.info('Starting a new vCenter simulator docker container.', verbosity=1) + + if not self.args.docker and not container_id: + # publish the simulator ports when not running inside docker + publish_ports = [ + '-p', '80:80', + '-p', '443:443', + '-p', '8080:8080', + '-p', '8989:8989', + '-p', '5000:5000', # control port for flask app in simulator + ] + else: + publish_ports = [] + + docker_pull(self.args, self.image) + docker_run( + self.args, + self.image, + ['-d', '--name', self.container_name] + publish_ports, + ) + + if self.args.docker: + vcenter_host = self.DOCKER_SIMULATOR_NAME + elif container_id: + vcenter_host = self._get_simulator_address() + display.info('Found vCenter simulator container address: %s' % vcenter_host, verbosity=1) + else: + vcenter_host = 'localhost' + + self._set_cloud_config('vcenter_host', vcenter_host) + + def _get_simulator_address(self): + results = docker_inspect(self.args, self.container_name) + ipaddress = results[0]['NetworkSettings']['IPAddress'] + return ipaddress + + def _setup_static(self): + raise NotImplementedError() + + +class VcenterEnvironment(CloudEnvironment): + """VMware vcenter/esx environment plugin. Updates integration test environment after delegation.""" + def configure_environment(self, env, cmd): + """ + :type env: dict[str, str] + :type cmd: list[str] + """ + + # Send the container IP down to the integration test(s) + env['vcenter_host'] = self._get_cloud_config('vcenter_host')