def test_get_prev_ipv6_oses():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
GetData._get_prev_ipv6_oses(endpoint)
def test_get_last_seen():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
endpoint.unknown()
GetData._get_last_seen(endpoint)
def test_get_prev_state():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
endpoint.p_prev_state = None
prev_state = GetData._get_prev_state(endpoint)
assert prev_state == None
endpoint.unknown()
GetData._get_prev_state(endpoint)
endpoint.queue()
GetData._get_prev_state(endpoint)
def test_get_ipv6_os():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1',
'ipv6': '1212::1'
}
endpoint.metadata = {'ipv6_addresses': {}}
ipv6_os = GetData._get_ipv6_os(endpoint)
assert ipv6_os == NO_DATA
endpoint.metadata = {'ipv6_addresses': {'1212::1': {'short_os': 'foo'}}}
ipv6_os = GetData._get_ipv6_os(endpoint)
assert ipv6_os == 'foo'
def test_get_vlan():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'vlan': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
vlan = GetData._get_vlan(endpoint)
assert vlan == 'foo'
def test_get_port():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
port = GetData._get_port(endpoint)
assert port == '1'
def test_get_name():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
name = GetData._get_name(endpoint)
assert name == 'foo'
def test_get_next_state():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
state = GetData._get_next_state(endpoint)
assert state == 'None'
def test_get_history():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
history = GetData._get_history(endpoint)
assert history == 'No history recorded yet.'
def test_get_ignored():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
ignored = GetData._get_ignored(endpoint)
assert ignored == 'False'
def test_get_switch():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
switch = GetData._get_switch(endpoint)
assert switch == 'foo'
def test_get_pcap_labels():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
endpoint.metadata = {'mac_addresses': {}}
pcap_labels = GetData._get_pcap_labels(endpoint)
assert pcap_labels == NO_DATA
endpoint.metadata = {
'mac_addresses': {
'00:00:00:00:00:00': {
'pcap_labels': 'foo'
}
}
}
pcap_labels = GetData._get_pcap_labels(endpoint)
assert pcap_labels == 'foo'
def test_get_controller_type():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1',
'controller_type': 'foo'
}
controller_type = GetData._get_controller_type(endpoint)
assert controller_type == 'foo'
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
controller_type = GetData._get_controller_type(endpoint)
assert controller_type == NO_DATA
def test_get_ipv4():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1',
'ipv4': '0.0.0.0'
}
ipv4 = GetData._get_ipv4(endpoint)
assert ipv4 == '0.0.0.0'
def test_get_ipv6():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1',
'ipv6': '1212::1'
}
ipv6 = GetData._get_ipv6(endpoint)
assert ipv6 == '1212::1'
def test_get_acls():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1',
'ipv4': '0.0.0.0'
}
endpoint.acl_data = []
acls = GetData._get_acls(endpoint)
assert acls == '[]'
def test_get_ipv6_subnet():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1',
'ipv6': '1212::1'
}
ipv6_subnet = GetData._get_ipv6_subnet(endpoint)
assert ipv6_subnet == NO_DATA
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1',
'ipv6': '1212::1',
'ipv6_subnet': '1212::1/64'
}
ipv6_subnet = GetData._get_ipv6_subnet(endpoint)
assert ipv6_subnet == '1212::1/64'
def test_get_role_confidence():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
endpoint.metadata = {'mac_addresses': {}}
confidence = GetData._get_role_confidence(endpoint)
assert confidence == '0'
endpoint.metadata = {
'mac_addresses': {
'00:00:00:00:00:00': {
'classification': {
'confidences': [10.0, 9.0, 8.0]
}
}
}
}
confidence = GetData._get_role_confidence(endpoint)
assert confidence == '10.0'
def test_get_role():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
endpoint.metadata = {'mac_addresses': {}}
role = GetData._get_role(endpoint)
assert role == NO_DATA
endpoint.metadata = {
'mac_addresses': {
'00:00:00:00:00:00': {
'classification': {
'labels': ['foo', 'bar', 'baz']
}
}
}
}
role = GetData._get_role(endpoint)
assert role == 'foo'
def test_get_ipv6_rdns():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1',
'ipv4': '0.0.0.0',
'ipv4_subnet': '0.0.0.0/24'
}
ipv6_rdns = GetData._get_ipv6_rdns(endpoint)
assert ipv6_rdns == NO_DATA
endpoint.endpoint_data = {
'ipv6_rdns': 'foo.internal',
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1',
'ipv4': '0.0.0.0',
'ipv4_subnet': '0.0.0.0/24'
}
ipv6_rdns = GetData._get_ipv6_rdns(endpoint)
assert ipv6_rdns == 'foo.internal'
def test_get_ether_vendor():
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1',
'ipv4': '0.0.0.0',
'ipv4_subnet': '0.0.0.0/24'
}
ether_vendor = GetData._get_ether_vendor(endpoint)
assert ether_vendor == NO_DATA
endpoint.endpoint_data = {
'ether_vendor': 'VENDOR',
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1',
'ipv4': '0.0.0.0',
'ipv4_subnet': '0.0.0.0/24'
}
ether_vendor = GetData._get_ether_vendor(endpoint)
assert ether_vendor == 'VENDOR'