def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) fields = ["name", "path"] project_codebase = codebase.ProjectCodebase(self.object) context["tree_data"] = [codebase.get_tree(project_codebase.root, fields)] return context
def test_scanpipe_pipes_codebase_get_tree(self): fixtures = self.data_location / "asgiref-3.3.0_fixtures.json" call_command("loaddata", fixtures, **{"verbosity": 0}) project = Project.objects.get(name="asgiref") scan_results = self.data_location / "asgiref-3.3.0_scan.json" virtual_codebase = scancode.get_virtual_codebase(project, scan_results) project_codebase = codebase.ProjectCodebase(project) fields = ["name", "path"] virtual_tree = codebase.get_tree( virtual_codebase.root, fields, codebase=virtual_codebase ) project_tree = codebase.get_tree(project_codebase.root, fields) with open(self.data_location / "asgiref-3.3.0_tree.json") as f: expected = json.loads(f.read()) self.assertEqual(expected, project_tree) self.assertEqual(expected, virtual_tree)