Python CheckSSH._collect_metadata Examples

Example #1

File: test_ssh.py Project: zhangjm12/integrations-core

def test_collect_metadata(version, metadata, datadog_agent):
    client = MagicMock()
    client.get_transport = MagicMock(
        return_value=namedtuple('Transport', ['remote_version'])(version))

    ssh = CheckSSH('ssh_check', {}, {}, list(common.INSTANCES.values()))
    ssh.check_id = 'test:123'
    ssh._collect_metadata(client)
    datadog_agent.assert_metadata('test:123', metadata)

Example #2

File: test_ssh.py Project: zhangjm12/integrations-core

def test_collect_bad_metadata(datadog_agent):
    client = MagicMock()
    client.get_transport = MagicMock(return_value=namedtuple(
        'Transport', ['remote_version'])('Cannot parse this'))

    ssh = CheckSSH('ssh_check', {}, {}, list(common.INSTANCES.values()))
    ssh.check_id = 'test:123'
    ssh._collect_metadata(client)
    datadog_agent.assert_metadata_count(1)
    datadog_agent.assert_metadata('test:123', {'flavor': 'unknown'})