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))