def test_check_ancestor_trace(self): fake_nodes = [ ('a', '1.0.0'), ('b', '2.0.0'), ('c', '1.4.0'), ('d', '2.0.0'), ('e', '1.6.0'), ('f', '210.0'), ] fake_edges = [ [('root', '0.0.0'), fake_nodes[0]], [('root', '0.0.0'), fake_nodes[1]], [fake_nodes[0], fake_nodes[1]], [fake_nodes[1], fake_nodes[2]], [fake_nodes[2], fake_nodes[3]], [fake_nodes[3], fake_nodes[4]], [fake_nodes[4], fake_nodes[5]], ] f_venv = MagicMock() f_venv.nodes = fake_nodes f_venv.edges = fake_edges p = Package(fake_nodes[0][0], fake_nodes[0][1]) ret = p.ancestor_trace(f_venv) self.assertEqual(ret, {('root', '0.0.0'): 1, ('a', '1.0.0'): 0}) p2 = Package(fake_nodes[1][0], fake_nodes[1][1]) ret = p2.ancestor_trace(f_venv) self.assertEqual(ret, { ('root', '0.0.0'): 1, ('b', '2.0.0'): 0, ('a', '1.0.0'): 1 }) p5 = Package(fake_nodes[5][0], fake_nodes[5][1]) ret = p5.ancestor_trace(f_venv) self.assertEqual( ret, { ('f', '210.0'): 0, ('e', '1.6.0'): 1, ('d', '2.0.0'): 2, ('c', '1.4.0'): 3, ('b', '2.0.0'): 4, ('a', '1.0.0'): 5, ('root', '0.0.0'): 5, })
def test_check_ancestor_trace(self): fake_nodes = [('a', '1.0.0'), ('b', '2.0.0'), ('c', '1.4.0'), ('d', '2.0.0'), ('e', '1.6.0'), ('f', '210.0'), ] fake_edges = [[('root', '0.0.0'), fake_nodes[0]], [('root', '0.0.0'), fake_nodes[1]], [fake_nodes[0], fake_nodes[1]], [fake_nodes[1], fake_nodes[2]], [fake_nodes[2], fake_nodes[3]], [fake_nodes[3], fake_nodes[4]], [fake_nodes[4], fake_nodes[5]], ] f_venv = MagicMock() f_venv.nodes = fake_nodes f_venv.edges = fake_edges p = Package(fake_nodes[0][0], fake_nodes[0][1]) ret = p.ancestor_trace(f_venv) self.assertEqual(ret, {('root', '0.0.0'): 1, ('a', '1.0.0'): 0}) p2 = Package(fake_nodes[1][0], fake_nodes[1][1]) ret = p2.ancestor_trace(f_venv) self.assertEqual(ret, {('root', '0.0.0'): 1, ('b', '2.0.0'): 0, ('a', '1.0.0'): 1}) p5 = Package(fake_nodes[5][0], fake_nodes[5][1]) ret = p5.ancestor_trace(f_venv) self.assertEqual(ret, {('f', '210.0'): 0, ('e', '1.6.0'): 1, ('d', '2.0.0'): 2, ('c', '1.4.0'): 3, ('b', '2.0.0'): 4, ('a', '1.0.0'): 5, ('root', '0.0.0'): 5, })
def test_returns_dict(self): p = Package("Django") at = p.ancestor_trace(self.venv) self.assertEqual(type(at), dict)