diff --git a/scanpipe/api/serializers.py b/scanpipe/api/serializers.py index faa27c1ee..4e2e67705 100644 --- a/scanpipe/api/serializers.py +++ b/scanpipe/api/serializers.py @@ -384,6 +384,7 @@ def get_model_serializer(model_class): CodebaseResource: CodebaseResourceSerializer, DiscoveredPackage: DiscoveredPackageSerializer, DiscoveredDependency: DiscoveredDependencySerializer, + CodebaseRelation: CodebaseRelationSerializer, ProjectError: ProjectErrorSerializer, }.get(model_class, None) diff --git a/scanpipe/pipes/output.py b/scanpipe/pipes/output.py index 8dfd997d6..caea7704b 100644 --- a/scanpipe/pipes/output.py +++ b/scanpipe/pipes/output.py @@ -272,6 +272,7 @@ def to_json(project): "discoveredpackage": "PACKAGES", "discovereddependency": "DEPENDENCIES", "codebaseresource": "RESOURCES", + "codebaserelation": "RELATIONS", "projecterror": "ERRORS", } @@ -447,6 +448,7 @@ def to_xlsx(project): "discoveredpackage", "discovereddependency", "codebaseresource", + "codebaserelation", "projecterror", ] diff --git a/scanpipe/tests/pipes/test_output.py b/scanpipe/tests/pipes/test_output.py index 5d49e4ee4..095473c27 100644 --- a/scanpipe/tests/pipes/test_output.py +++ b/scanpipe/tests/pipes/test_output.py @@ -194,7 +194,7 @@ def test_scanpipe_pipes_outputs_to_xlsx(self): # Make sure the output can be generated even if the work_directory was wiped shutil.rmtree(project.work_directory) - with self.assertNumQueries(7): + with self.assertNumQueries(8): output_file = output.to_xlsx(project=project) self.assertIn(output_file.name, project.output_root)