diff options
author | Mark Mercado <mamercad@gmail.com> | 2021-05-12 19:07:21 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-05-12 16:07:21 -0700 |
commit | 1906d75907e36b84609dd860e34cd1563a13f37a (patch) | |
tree | ac07a3ac968237a7fdcd57c33b89f7b5dcc0416f /test/lib/ansible_test/_internal | |
parent | d6e28e68599e703c153914610152cf4492851eb3 (diff) | |
download | ansible-1906d75907e36b84609dd860e34cd1563a13f37a.tar.gz |
Adding DigitalOcean cloud support to ansible-test (#74222)
Diffstat (limited to 'test/lib/ansible_test/_internal')
-rw-r--r-- | test/lib/ansible_test/_internal/commands/integration/cloud/digitalocean.py | 55 |
1 files changed, 55 insertions, 0 deletions
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, + ) |