mirror of https://github.com/ansible/ansible.git
Adding DigitalOcean cloud support to ansible-test (#74222)
parent
d6e28e6859
commit
1906d75907
@ -0,0 +1,2 @@
|
||||
minor_changes:
|
||||
- ansible-test - Adding DigitalOcean cloud support to ansible-test (https://github.com/ansible/ansible/pull/74222).
|
@ -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,
|
||||
)
|
Loading…
Reference in New Issue