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])