Python stub Examples

Example #1

def mocked_aws_cf(monkeypatch, mocked_test_runner):
    """Does not include SSH key mocking
    """
    monkeypatch.setattr(test_util.aws.DcosCfStack, '__init__', stub(None))
    monkeypatch.setattr(
        test_util.aws, 'fetch_stack',
        lambda stack_name, bw: test_util.aws.DcosCfStack(stack_name, bw))
    # mock create
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'create_stack',
                        stub(MockStack()))
    # mock wait
    monkeypatch.setattr(test_util.aws.CfStack, 'wait_for_complete', stub(None))
    # mock describe
    monkeypatch.setattr(test_util.aws.DcosCfStack, 'get_master_ips',
                        stub([mock_pub_priv_host]))
    monkeypatch.setattr(test_util.aws.DcosCfStack, 'get_private_agent_ips',
                        stub([mock_priv_host]))
    monkeypatch.setattr(test_util.aws.DcosCfStack, 'get_public_agent_ips',
                        stub([mock_pub_priv_host]))
    # mock delete
    monkeypatch.setattr(test_util.aws.DcosCfStack, 'delete', stub(None))
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'delete_key_pair',
                        stub(None))
    # mock config
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'create_key_pair',
                        stub(launch.util.MOCK_SSH_KEY_DATA))

Example #2

def mock_bare_cluster_hosts(monkeypatch, mocked_aws_cf, mock_ssher):
    monkeypatch.setattr(test_util.aws.BareClusterCfStack, '__init__', stub(None))
    monkeypatch.setattr(test_util.aws.BareClusterCfStack, 'delete', stub(None))
    monkeypatch.setattr(test_util.aws.BareClusterCfStack, 'get_host_ips', stub([mock_pub_priv_host] * 4))
    monkeypatch.setattr(
        test_util.aws, 'fetch_stack', lambda stack_name, bw: test_util.aws.BareClusterCfStack(stack_name, bw))
    monkeypatch.setattr(test_util.onprem.OnpremCluster, 'setup_installer_server', stub(None))
    monkeypatch.setattr(test_util.onprem.OnpremCluster, 'start_bootstrap_zk', stub(None))
    monkeypatch.setattr(test_util.onprem, 'DcosInstallerApiSession', MockInstaller)
    monkeypatch.setattr(launch.onprem.OnpremLauncher, 'get_last_state', stub(None))

Example #3

File: conftest.py Project: tamarrow/dcos

def mocked_aws_cf(monkeypatch, mocked_test_runner):
    """Does not include SSH key mocking
    """
    monkeypatch.setattr(test_util.aws.DcosCfStack, '__init__', stub(None))
    monkeypatch.setattr(
        test_util.aws, 'fetch_stack', lambda stack_name, bw: test_util.aws.DcosCfStack(stack_name, bw))
    # mock create
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'create_stack', stub(MockStack()))
    # mock wait
    monkeypatch.setattr(test_util.aws.CfStack, 'wait_for_complete', stub(None))
    # mock describe
    monkeypatch.setattr(test_util.aws.DcosCfStack, 'get_master_ips',
                        stub([mock_pub_priv_host]))
    monkeypatch.setattr(test_util.aws.DcosCfStack, 'get_private_agent_ips',
                        stub([mock_priv_host]))
    monkeypatch.setattr(test_util.aws.DcosCfStack, 'get_public_agent_ips',
                        stub([mock_pub_priv_host]))
    # mock delete
    monkeypatch.setattr(test_util.aws.DcosCfStack, 'delete', stub(None))
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'delete_key_pair', stub(None))
    # mock config
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'create_key_pair', stub(launch.util.MOCK_SSH_KEY_DATA))

Example #4

def mocked_azure(monkeypatch, mocked_test_runner):
    monkeypatch.setattr(test_util.azure.ServicePrincipalCredentials, '__init__', stub(None))
    monkeypatch.setattr(test_util.azure.ResourceManagementClient, '__init__', stub(None))
    monkeypatch.setattr(test_util.azure.NetworkManagementClient, '__init__', stub(None))

    monkeypatch.setattr(test_util.azure.AzureWrapper, 'deploy_template_to_new_resource_group', stub(None))
    monkeypatch.setattr(test_util.azure.DcosAzureResourceGroup, 'wait_for_deployment', stub(None))
    monkeypatch.setattr(test_util.azure.DcosAzureResourceGroup, 'delete', stub(None))
    monkeypatch.setattr(test_util.azure.DcosAzureResourceGroup, 'get_master_ips',
                        stub([mock_pub_priv_host]))
    monkeypatch.setattr(test_util.azure.DcosAzureResourceGroup, 'get_private_agent_ips',
                        stub([mock_priv_host]))
    monkeypatch.setattr(test_util.azure.DcosAzureResourceGroup, 'get_public_agent_ips',
                        stub([mock_pub_priv_host]))
    monkeypatch.setattr(test_util.azure.DcosAzureResourceGroup, 'public_agent_lb_fqdn', 'abc-foo-bar')
    monkeypatch.setattr(test_util.azure.DcosAzureResourceGroup, 'public_master_lb_fqdn', 'dead-beef')

Example #5

def mocked_aws_zen_cf(monkeypatch, mocked_aws_cf):
    monkeypatch.setattr(test_util.aws.DcosZenCfStack, '__init__', stub(None))
    monkeypatch.setattr(
        test_util.aws, 'fetch_stack', lambda stack_name, bw: test_util.aws.DcosZenCfStack(stack_name, bw))
    # mock create
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'create_vpc_tagged', stub(launch.util.MOCK_VPC_ID))
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'create_internet_gateway_tagged', stub(launch.util.MOCK_GATEWAY_ID))
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'create_subnet_tagged', stub(launch.util.MOCK_SUBNET_ID))
    # mock delete
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'delete_subnet', stub(None))
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'delete_vpc', stub(None))
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'delete_internet_gateway', stub(None))
    # mock describe
    monkeypatch.setattr(test_util.aws.DcosZenCfStack, 'get_master_ips',
                        stub([Host('127.0.0.1', '12.34.56')]))
    monkeypatch.setattr(test_util.aws.DcosZenCfStack, 'get_private_agent_ips',
                        stub([Host('127.0.0.1', None)]))
    monkeypatch.setattr(test_util.aws.DcosZenCfStack, 'get_public_agent_ips',
                        stub([Host('127.0.0.1', '12.34.56')]))
    # mock delete
    monkeypatch.setattr(test_util.aws.DcosZenCfStack, 'delete', stub(None))

Example #6

def mocked_test_runner(monkeypatch):
    monkeypatch.setattr(ssh.tunnel, 'tunnel', mocked_context)
    monkeypatch.setattr(test_util.runner, 'integration_test', stub(0))

Example #7

File: conftest.py Project: tamarrow/dcos

def mocked_aws_zen_cf(monkeypatch, mocked_aws_cf):
    monkeypatch.setattr(test_util.aws.DcosZenCfStack, '__init__', stub(None))
    monkeypatch.setattr(
        test_util.aws, 'fetch_stack', lambda stack_name, bw: test_util.aws.DcosZenCfStack(stack_name, bw))
    # mock create
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'create_vpc_tagged', stub(launch.util.MOCK_VPC_ID))
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'create_internet_gateway_tagged', stub(launch.util.MOCK_GATEWAY_ID))
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'create_subnet_tagged', stub(launch.util.MOCK_SUBNET_ID))
    # mock delete
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'delete_subnet', stub(None))
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'delete_vpc', stub(None))
    monkeypatch.setattr(test_util.aws.BotoWrapper, 'delete_internet_gateway', stub(None))
    # mock describe
    monkeypatch.setattr(test_util.aws.DcosZenCfStack, 'get_master_ips',
                        stub([mock_pub_priv_host]))
    monkeypatch.setattr(test_util.aws.DcosZenCfStack, 'get_private_agent_ips',
                        stub([mock_priv_host]))
    monkeypatch.setattr(test_util.aws.DcosZenCfStack, 'get_public_agent_ips',
                        stub([mock_pub_priv_host]))
    # mock delete
    monkeypatch.setattr(test_util.aws.DcosZenCfStack, 'delete', stub(None))

Example #8

def mock_ssher(monkeypatch):
    monkeypatch.setattr(ssh.ssher, 'open_tunnel', mocked_context)
    monkeypatch.setattr(ssh.ssher.Ssher, 'command', stub(b''))
    monkeypatch.setattr(ssh.ssher.Ssher, 'get_home_dir', stub(b''))

Example #9

def mocked_test_runner(monkeypatch):
    monkeypatch.setattr(launch.util, 'try_to_output_unbuffered', stub(0))