def test_add_reverse_iptag_then_locate_tag(self):
"""
check that asking for a reverse iptag for a speific partitioned vertex
works
:return:
"""
tag_info = Tags()
reverse_iptag = ReverseIPTag("", 1, 23, 0, 0, 1, 1)
parttiioned_vertex = PartitionedVertex(None, "")
tag_info.add_reverse_ip_tag(reverse_iptag, parttiioned_vertex)
gotton_tag = tag_info.get_reverse_ip_tags_for_vertex(parttiioned_vertex)
self.assertEqual(gotton_tag[0], reverse_iptag)
def test_add_reverse_iptag_then_not_locate_tag(self):
"""
check that asking for a reverse iptag with a incorrect partitioned vertex
will cause a none returned
:return:
"""
tag_info = Tags()
reverse_iptag = ReverseIPTag("", 1, 23, 0, 0, 1, 1)
parttiioned_vertex = PartitionedVertex(None, "")
parttiioned_vertex2 = PartitionedVertex(None, "")
tag_info.add_reverse_ip_tag(reverse_iptag, parttiioned_vertex2)
gotton_tag = tag_info.get_reverse_ip_tags_for_vertex(parttiioned_vertex)
self.assertEqual(gotton_tag, None)
