From 1906d75907e36b84609dd860e34cd1563a13f37a Mon Sep 17 00:00:00 2001 From: Mark Mercado Date: Wed, 12 May 2021 19:07:21 -0400 Subject: Adding DigitalOcean cloud support to ansible-test (#74222) --- .../commands/integration/cloud/digitalocean.py | 55 ++++++++++++++++++++++ 1 file changed, 55 insertions(+) create mode 100644 test/lib/ansible_test/_internal/commands/integration/cloud/digitalocean.py (limited to 'test/lib/ansible_test/_internal') diff --git a/test/lib/ansible_test/_internal/commands/integration/cloud/digitalocean.py b/test/lib/ansible_test/_internal/commands/integration/cloud/digitalocean.py new file mode 100644 index 0000000000..0ea771dc6c --- /dev/null +++ b/test/lib/ansible_test/_internal/commands/integration/cloud/digitalocean.py @@ -0,0 +1,55 @@ +"""DigitalOcean plugin for integration tests.""" +from __future__ import (absolute_import, division, print_function) +__metaclass__ = type + +from ....util import ( + ConfigParser, + display, +) + +from ....config import ( + IntegrationConfig, +) + +from . import ( + CloudEnvironment, + CloudEnvironmentConfig, + CloudProvider, +) + + +class DigitalOceanCloudProvider(CloudProvider): + """Checks if a configuration file has been passed or fixtures are going to be used for testing""" + def __init__(self, args): # type: (IntegrationConfig) -> None + super(DigitalOceanCloudProvider, self).__init__(args) + + self.uses_config = True + + def setup(self): # type: () -> None + """Setup the cloud resource before delegation and register a cleanup callback.""" + super(DigitalOceanCloudProvider, self).setup() + + self._use_static_config() + + +class DigitalOceanCloudEnvironment(CloudEnvironment): + """Updates integration test environment after delegation. Will setup the config file as parameter.""" + def get_environment_config(self): # type: () -> CloudEnvironmentConfig + """Return environment configuration for use in the test environment after delegation.""" + parser = ConfigParser() + parser.read(self.config_path) + + env_vars = dict( + DO_API_KEY=parser.get('default', 'key'), + ) + + display.sensitive.add(env_vars['DO_API_KEY']) + + ansible_vars = dict( + resource_prefix=self.resource_prefix, + ) + + return CloudEnvironmentConfig( + env_vars=env_vars, + ansible_vars=ansible_vars, + ) -- cgit v1.2.1