def test_retrieve_geoip4(self):
packets = ScapySource.load(['test.pcap'])
g = GraphManager(packets, layer=4, args=args)
node = list(g.data.keys())[0]
g._retrieve_node_info(node)
self.assertIn('country', g.data[node])
def test_retrieve_geoip(self):
packets = ScapySource.load(['test.pcap'])
g = GraphManager(packets, layer=4)
node = g.graph.nodes()[0]
g._retrieve_node_info(node)
self.assertIn('country', g.data[node])
def test_retrieve_geoip(self):
packets = ScapySource.load(['test.pcap'])
g = GraphManager(packets, layer=4)
node = g.graph.nodes()[0]
g._retrieve_node_info(node)
self.assertIn('country', g.data[node])
