tests.system.providers.amazon.aws.utils

Package Contents

Classes

SystemTestContextBuilder

This builder class ultimately constructs a TaskFlow task which is run at

Functions

fetch_variable(key[, default_value, test_name])

Given a Parameter name: first check for an existing Environment Variable,

set_env_id()

Retrieves or generates an Environment ID, validate that it is suitable,

purge_logs(test_logs[, force_delete, retry, retry_times])

Accepts a tuple in the format: ('log group name', 'log stream prefix').

split_string(string)

Attributes

ENV_ID_ENVIRON_KEY

ENV_ID_KEY

DEFAULT_ENV_ID_PREFIX

DEFAULT_ENV_ID_LEN

DEFAULT_ENV_ID

PURGE_LOGS_INTERVAL_PERIOD

TEST_FILE_IDENTIFIER

INVALID_ENV_ID_MSG

LOWERCASE_ENV_ID_MSG

NO_VALUE_MSG

log

tests.system.providers.amazon.aws.utils.ENV_ID_ENVIRON_KEY :str = SYSTEM_TESTS_ENV_ID[source]
tests.system.providers.amazon.aws.utils.ENV_ID_KEY :str = ENV_ID[source]
tests.system.providers.amazon.aws.utils.DEFAULT_ENV_ID_PREFIX :str = env[source]
tests.system.providers.amazon.aws.utils.DEFAULT_ENV_ID_LEN :int = 8[source]
tests.system.providers.amazon.aws.utils.DEFAULT_ENV_ID :str[source]
tests.system.providers.amazon.aws.utils.PURGE_LOGS_INTERVAL_PERIOD = 5[source]
tests.system.providers.amazon.aws.utils.TEST_FILE_IDENTIFIER :str = example[source]
tests.system.providers.amazon.aws.utils.INVALID_ENV_ID_MSG :str = In order to maximize compatibility, the SYSTEM_TESTS_ENV_ID must be an alphanumeric string which...[source]
tests.system.providers.amazon.aws.utils.LOWERCASE_ENV_ID_MSG :str = The provided Environment ID contains uppercase letters and will be converted to lowercase for...[source]
tests.system.providers.amazon.aws.utils.NO_VALUE_MSG :str = No Value Found: Variable {key} could not be found and no default value was provided.[source]
tests.system.providers.amazon.aws.utils.log[source]
class tests.system.providers.amazon.aws.utils.SystemTestContextBuilder[source]

This builder class ultimately constructs a TaskFlow task which is run at runtime (task execution time). This task generates and stores the test ENV_ID as well as any external resources requested (e.g.g IAM Roles, VPC, etc)

add_variable(variable_name, split_string=False, delimiter=None, **kwargs)[source]

Register a variable to fetch from environment or cloud parameter store

build()[source]

Build and return a TaskFlow task which will create an env_id and fetch requested variables. Storing everything in xcom for downstream tasks to use.

tests.system.providers.amazon.aws.utils.fetch_variable(key, default_value=None, test_name=None)[source]

Given a Parameter name: first check for an existing Environment Variable, then check SSM for a value. If neither are available, fall back on the optional default value.

Parameters
  • key (str) – The name of the Parameter to fetch a value for.

  • default_value (Optional[str]) – The default value to use if no value can be found.

  • test_name (Optional[str]) – The system test name.

Returns

The value of the parameter.

Return type

str

tests.system.providers.amazon.aws.utils.set_env_id()[source]

Retrieves or generates an Environment ID, validate that it is suitable, export it as an Environment Variable, and return it.

If an Environment ID has already been generated, use that. Otherwise, try to fetch it and export it as an Environment Variable. If there is not one available to fetch then generate one and export it as an Environment Variable.

Returns

A valid System Test Environment ID.

Return type

str

tests.system.providers.amazon.aws.utils.purge_logs(test_logs, force_delete=False, retry=False, retry_times=3)[source]

Accepts a tuple in the format: (‘log group name’, ‘log stream prefix’). For each log group, it will delete any log streams matching the provided prefix then if the log group is empty, delete the group. If the group is not empty that indicates there are logs not generated by the test and those are left intact. If check_log_streams is True, it will simply delete the log group regardless of log streams within that log group.

Parameters
  • test_logs (List[Tuple[str, Optional[str]]]) – A list of log_group/stream_prefix tuples to delete.

  • force_delete (bool) – Whether to check log streams within the log group before removal. If True, removes the log group and all its log streams inside it

  • retry (bool) – Whether to retry if the log group/stream was not found. In some cases, the log group/stream is created seconds after the main resource has been created. By default, it retries for 3 times with a 5s waiting period

  • retry_times (int) – Number of retries

tests.system.providers.amazon.aws.utils.split_string(string)[source]

Was this entry helpful?