VhtdZddlmZddlZddlmZddlmZddlm Z m Z m Z Gd d e Z Gd d e Z y) z#Vultr plugin for integration tests.) annotationsN)display)IntegrationConfig)CloudEnvironmentCloudEnvironmentConfig CloudProviderc0eZdZdZdfd Zdfd ZxZS)VultrCloudProviderz[Checks if a configuration file has been passed or fixtures are going to be used for testingc2t||d|_y)NT)super__init__ uses_config)selfargs __class__s g/home/dcms/DCMS/lib/python3.12/site-packages/ansible_test/_internal/commands/integration/cloud/vultr.pyrzVultrCloudProvider.__init__s cBt||jy)zKSetup the cloud resource before delegation and register a cleanup callback.N)rsetup_use_static_config)rrs rrzVultrCloudProvider.setups   !r)rrreturnNone)rr)__name__ __module__ __qualname____doc__rr __classcell__)rs@rr r se ""rr ceZdZdZddZy)VultrCloudEnvironmentz_Updates integration test environment after delegation. Will setup the config file as parameter.c"tj}|j|jt |j dd}t jj|dt |j}t||S)zRReturn environment configuration for use in the test environment after delegation.defaultkey) VULTR_API_KEYr%)vultr_resource_prefix)env_vars ansible_vars) configparser ConfigParserread config_pathdictgetr sensitiveaddresource_prefixr )rparserr'r(s rget_environment_configz,VultrCloudEnvironment.get_environment_config's}**, D$$% **Y6  h78"&"6"6 &%  rN)rr )rrrrr3rrr!r!$s i rr!)r __future__rr)utilrconfigrrr r r r!r4rrr9s;)" " " , r