def mock_ssh_client(monkeypatch):
monkeypatch.setattr(dcos_test_utils.ssh_client, 'open_tunnel',
mocked_context)
monkeypatch.setattr(dcos_test_utils.ssh_client.SshClient, 'command',
stub(b''))
monkeypatch.setattr(dcos_test_utils.ssh_client.SshClient, 'get_home_dir',
stub(b''))
def mock_bare_cluster_hosts(monkeypatch, mocked_aws_cf, mocked_test_runner,
mock_ssh_client):
monkeypatch.setattr(dcos_test_utils.onprem.OnpremCluster,
'setup_installer_server', stub(None))
monkeypatch.setattr(dcos_test_utils.onprem.OnpremCluster,
'start_bootstrap_zk', stub(None))
monkeypatch.setattr(dcos_test_utils.onprem, 'DcosInstallerApiSession',
MockInstaller)
monkeypatch.setattr(dcos_launch.onprem.OnpremLauncher, 'get_last_state',
stub(None))
def mocked_azure(monkeypatch, mocked_test_runner):
monkeypatch.setenv('AZURE_CLIENT_ID', 'AEF234DFLDWQMNEZ2')
monkeypatch.setenv('AZURE_CLIENT_SECRET', 'ASDPFOKAWEFN123')
monkeypatch.setenv('AZURE_TENANT_ID', 'ASDPFOKAWEFN123')
monkeypatch.setenv('AZURE_SUBSCRIPTION_ID', 'ASDPFOKAWEFN123')
monkeypatch.setattr(dcos_launch.platforms.arm.ServicePrincipalCredentials,
'__init__', stub(None))
monkeypatch.setattr(dcos_launch.platforms.arm.ResourceManagementClient,
'__init__', stub(None))
monkeypatch.setattr(dcos_launch.platforms.arm.NetworkManagementClient,
'__init__', stub(None))
monkeypatch.setattr(dcos_launch.platforms.arm.AzureWrapper,
'deploy_template_to_new_resource_group', stub(None))
monkeypatch.setattr(dcos_launch.platforms.arm.DcosAzureResourceGroup,
'wait_for_deployment', stub(None))
monkeypatch.setattr(dcos_launch.platforms.arm.DcosAzureResourceGroup,
'delete', stub(None))
monkeypatch.setattr(dcos_launch.platforms.arm.DcosAzureResourceGroup,
'get_master_ips', stub([mock_pub_priv_host]))
monkeypatch.setattr(dcos_launch.platforms.arm.DcosAzureResourceGroup,
'get_private_agent_ips', stub([mock_priv_host]))
monkeypatch.setattr(dcos_launch.platforms.arm.DcosAzureResourceGroup,
'get_public_agent_ips', stub([mock_pub_priv_host]))
monkeypatch.setattr(dcos_launch.platforms.arm.DcosAzureResourceGroup,
'public_agent_lb_fqdn', 'abc-foo-bar')
monkeypatch.setattr(dcos_launch.platforms.arm.DcosAzureResourceGroup,
'public_master_lb_fqdn', 'dead-beef')
def mocked_aws_cf(monkeypatch, mocked_test_runner):
"""Does not include SSH key mocking
"""
# mock credentials
monkeypatch.setenv('AWS_ACCESS_KEY_ID', 'AEF234DFLDWQMNEZ2')
monkeypatch.setenv('AWS_SECRET_ACCESS_KEY', 'ASDPFOKAWEFN123')
monkeypatch.setattr(dcos_launch.platforms.aws.DcosCfStack, '__init__',
stub(None))
monkeypatch.setattr(
dcos_launch.platforms.aws, 'fetch_stack',
lambda stack_name, bw: dcos_launch.platforms.aws.DcosCfStack(
stack_name, bw))
# mock create
monkeypatch.setattr(dcos_launch.platforms.aws.BotoWrapper, 'create_stack',
stub(MockStack()))
# mock wait
monkeypatch.setattr(dcos_launch.platforms.aws.CfStack, 'wait_for_complete',
stub(None))
# mock describe
monkeypatch.setattr(dcos_launch.platforms.aws.DcosCfStack,
'get_master_ips', stub([mock_pub_priv_host]))
monkeypatch.setattr(dcos_launch.platforms.aws.DcosCfStack,
'get_private_agent_ips', stub([mock_priv_host]))
monkeypatch.setattr(dcos_launch.platforms.aws.DcosCfStack,
'get_public_agent_ips', stub([mock_pub_priv_host]))
# mock delete
monkeypatch.setattr(dcos_launch.platforms.aws.DcosCfStack, 'delete',
stub(None))
monkeypatch.setattr(dcos_launch.platforms.aws.BotoWrapper,
'delete_key_pair', stub(None))
# mock config
monkeypatch.setattr(dcos_launch.platforms.aws.BotoWrapper,
'create_key_pair',
stub(dcos_launch.util.MOCK_SSH_KEY_DATA))
def mocked_aws_cfstack_bare_cluster(monkeypatch, mock_bare_cluster_hosts):
monkeypatch.setattr(dcos_launch.platforms.aws.BareClusterCfStack,
'__init__', stub(None))
monkeypatch.setattr(dcos_launch.platforms.aws.BareClusterCfStack, 'delete',
stub(None))
monkeypatch.setattr(dcos_launch.platforms.aws.BareClusterCfStack,
'get_host_ips', stub([mock_pub_priv_host] * 4))
monkeypatch.setattr(
dcos_launch.platforms.aws, 'fetch_stack',
lambda stack_name, bw: dcos_launch.platforms.aws.BareClusterCfStack(
stack_name, bw))
def mock_bare_cluster_hosts(monkeypatch, mocked_aws_cf, mocked_test_runner, mock_ssh_client):
monkeypatch.setattr(dcos_test_utils.aws.BareClusterCfStack, '__init__', stub(None))
monkeypatch.setattr(dcos_test_utils.aws.BareClusterCfStack, 'delete', stub(None))
monkeypatch.setattr(dcos_test_utils.aws.BareClusterCfStack, 'get_host_ips', stub([mock_pub_priv_host] * 4))
monkeypatch.setattr(
dcos_test_utils.aws, 'fetch_stack', lambda stack_name,
bw: dcos_test_utils.aws.BareClusterCfStack(stack_name, bw))
monkeypatch.setattr(dcos_test_utils.onprem.OnpremCluster, 'setup_installer_server', stub(None))
monkeypatch.setattr(dcos_test_utils.onprem.OnpremCluster, 'start_bootstrap_zk', stub(None))
monkeypatch.setattr(dcos_test_utils.onprem, 'DcosInstallerApiSession', MockInstaller)
monkeypatch.setattr(dcos_launch.onprem.OnpremLauncher, 'get_last_state', stub(None))
def mock_ssh_client(monkeypatch):
# monkeypatch.setattr(dcos_test_utils.ssh_client, 'Tunnelled', MockTunnelled)
monkeypatch.setattr(dcos_test_utils.ssh_client, 'open_tunnel',
mocked_context)
monkeypatch.setattr(dcos_test_utils.ssh_client.SshClient, 'command',
stub(b''))
monkeypatch.setattr(dcos_test_utils.ssh_client.SshClient, 'get_home_dir',
stub(b''))
# need to nullify platforms.onprem
monkeypatch.setattr(dcos_launch.platforms.onprem, 'prepare_bootstrap',
stub('foo'))
monkeypatch.setattr(dcos_launch.platforms.onprem, 'install_dcos',
stub(None))
def __init__(self, credentials_dict):
DeploymentManagerMock = namedtuple('DeploymentManagerMock',
'deployments')
DeploymentFunctionsMock = namedtuple('DeploymentFunctionsMock',
'insert delete get')
ApiRequestMock = namedtuple('ApiRequestMock', 'execute')
self.project_id = ''
api_request_mock = ApiRequestMock(
lambda: {'operation': {
'status': 'DONE'
}})
self.deployment_manager = DeploymentManagerMock(
lambda: DeploymentFunctionsMock(stub(api_request_mock),
stub(api_request_mock),
stub(api_request_mock)))
def mocked_aws_cfstack_bare_cluster(monkeypatch, mock_ssh_client,
mocked_aws_cf):
monkeypatch.setattr(dcos_launch.platforms.aws.BareClusterCfStack,
'__init__', stub(None))
monkeypatch.setattr(dcos_launch.platforms.aws.BareClusterCfStack, 'delete',
stub(None))
monkeypatch.setattr(
dcos_launch.aws.OnPremLauncher, 'get_onprem_cluster',
dcos_launch.onprem.AbstractOnpremLauncher.get_onprem_cluster)
monkeypatch.setattr(dcos_launch.platforms.aws.BareClusterCfStack,
'get_cluster_host_ips', stub([mock_pub_priv_host] * 4))
monkeypatch.setattr(dcos_launch.platforms.aws.BareClusterCfStack,
'get_bootstrap_ip', stub(mock_pub_priv_host))
monkeypatch.setattr(
dcos_launch.platforms.aws, 'fetch_stack',
lambda stack_name, bw: dcos_launch.platforms.aws.BareClusterCfStack(
stack_name, bw))
def mocked_test_runner(monkeypatch):
monkeypatch.setattr(dcos_launch.util, 'try_to_output_unbuffered', stub(0))
def mocked_aws_zen_cf(monkeypatch, mocked_aws_cf):
monkeypatch.setattr(dcos_launch.platforms.aws.DcosZenCfStack, '__init__',
stub(None))
monkeypatch.setattr(
dcos_launch.platforms.aws, 'fetch_stack',
lambda stack_name, bw: dcos_launch.platforms.aws.DcosZenCfStack(
stack_name, bw))
# mock create
monkeypatch.setattr(dcos_launch.platforms.aws.BotoWrapper,
'create_vpc_tagged',
stub(dcos_launch.util.MOCK_VPC_ID))
monkeypatch.setattr(dcos_launch.platforms.aws.BotoWrapper,
'create_internet_gateway_tagged',
stub(dcos_launch.util.MOCK_GATEWAY_ID))
monkeypatch.setattr(dcos_launch.platforms.aws.BotoWrapper,
'create_subnet_tagged',
stub(dcos_launch.util.MOCK_SUBNET_ID))
# mock delete
monkeypatch.setattr(dcos_launch.platforms.aws.BotoWrapper, 'delete_subnet',
stub(None))
monkeypatch.setattr(dcos_launch.platforms.aws.BotoWrapper, 'delete_vpc',
stub(None))
monkeypatch.setattr(dcos_launch.platforms.aws.BotoWrapper,
'delete_internet_gateway', stub(None))
# mock describe
monkeypatch.setattr(dcos_launch.platforms.aws.DcosZenCfStack,
'get_master_ips', stub([mock_pub_priv_host]))
monkeypatch.setattr(dcos_launch.platforms.aws.DcosZenCfStack,
'get_private_agent_ips', stub([mock_priv_host]))
monkeypatch.setattr(dcos_launch.platforms.aws.DcosZenCfStack,
'get_public_agent_ips', stub([mock_pub_priv_host]))
# mock delete
monkeypatch.setattr(dcos_launch.platforms.aws.DcosZenCfStack, 'delete',
stub(None))
