Add unit test test_scanpipe_pipes_make_relationship #688

Signed-off-by: Thomas Druez <tdruez@nexb.com>

scanpipe/tests/pipes/test_pipes.py

@@ -29,6 +29,7 @@
 from django.test import TransactionTestCase
 
 from scanpipe import pipes
+from scanpipe.models import CodebaseRelation
 from scanpipe.models import CodebaseResource
 from scanpipe.models import DiscoveredPackage
 from scanpipe.models import Project
@@ -137,6 +138,25 @@ def test_scanpipe_pipes_update_or_create_dependency(self):
         dependency = pipes.update_or_create_dependency(p1, dependency_data)
         self.assertEqual(dependency.scope, "install")
 
+    def test_scanpipe_pipes_make_relationship(self):
+        p1 = Project.objects.create(name="Analysis")
+        from_resource = CodebaseResource.objects.create(project=p1, path="Name.java")
+        to_resource = CodebaseResource.objects.create(project=p1, path="Name.class")
+
+        relation = pipes.make_relationship(
+            from_resource=from_resource,
+            to_resource=to_resource,
+            relationship=CodebaseRelation.Relationship.COMPILED,
+            match_type="java_to_class",
+            extra_data={"extra": "data"},
+        )
+
+        self.assertEqual(from_resource, relation.from_resource)
+        self.assertEqual(to_resource, relation.to_resource)
+        self.assertEqual(CodebaseRelation.Relationship.COMPILED, relation.relationship)
+        self.assertEqual("java_to_class", relation.match_type)
+        self.assertEqual({"extra": "data"}, relation.extra_data)
+
 
 class ScanPipePipesTransactionTest(TransactionTestCase):
     """