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