Add unittests for new d2d modules

Signed-off-by: Ayan Sinha Mahapatra <[email protected]>

Author: AyanSinhaMahapatra

scanpipe/tests/data/d2d-macho/from-ollama.zip

@@ -1580,6 +1580,73 @@ def test_scanpipe_pipes_d2d_map_ruby(self):
             ).count(),
         )
 
+    @skipIf(sys.platform == "darwin", "Test is failing on macOS")
+    def test_scanpipe_pipes_d2d_extract_binary_symbols_from_resources(self):
+        input_dir = self.project1.input_path
+        input_resources = [
+            self.data / "d2d-macho/to-ollama.zip",
+            self.data / "d2d-macho/from-ollama.zip",
+        ]
+        copy_inputs(input_resources, input_dir)
+        self.from_files, self.to_files = d2d.get_inputs(self.project1)
+        inputs_with_codebase_path_destination = [
+            (self.from_files, self.project1.codebase_path / d2d.FROM),
+            (self.to_files, self.project1.codebase_path / d2d.TO),
+        ]
+        for input_files, codebase_path in inputs_with_codebase_path_destination:
+            for input_file_path in input_files:
+                scancode.extract_archive(input_file_path, codebase_path)
+
+        scancode.extract_archives(
+            self.project1.codebase_path,
+            recurse=True,
+        )
+        pipes.collect_and_create_codebase_resources(self.project1)
+        buffer = io.StringIO()
+
+        binary_resource = self.project1.codebaseresources.get(
+            path="to/libggml-cpu-skylakex.so"
+        )
+        d2d.extract_binary_symbols_from_resources(
+            resources=[binary_resource],
+            binary_symbols_func=d2d.collect_and_parse_macho_symbols,
+            logger=buffer.write,
+        )
+        symbols = binary_resource.extra_data.get("macho_symbols")
+        self.assertNotEqual(symbols, [])
+
+    @skipIf(sys.platform == "darwin", "Test is failing on macOS")
+    def test_scanpipe_pipes_d2d_extract_binary_symbols(self):
+        input_dir = self.project1.input_path
+        input_resources = [
+            self.data / "d2d-macho/to-ollama.zip",
+            self.data / "d2d-macho/from-ollama.zip",
+        ]
+        copy_inputs(input_resources, input_dir)
+        self.from_files, self.to_files = d2d.get_inputs(self.project1)
+        inputs_with_codebase_path_destination = [
+            (self.from_files, self.project1.codebase_path / d2d.FROM),
+            (self.to_files, self.project1.codebase_path / d2d.TO),
+        ]
+        for input_files, codebase_path in inputs_with_codebase_path_destination:
+            for input_file_path in input_files:
+                scancode.extract_archive(input_file_path, codebase_path)
+
+        scancode.extract_archives(
+            self.project1.codebase_path,
+            recurse=True,
+        )
+        pipes.collect_and_create_codebase_resources(self.project1)
+        buffer = io.StringIO()
+        d2d.extract_binary_symbols(
+            project=self.project1, options=["Go"], logger=buffer.write
+        )
+        binary_resource = self.project1.codebaseresources.get(
+            path="to/libggml-cpu-skylakex.so"
+        )
+        symbols = binary_resource.extra_data.get("macho_symbols")
+        self.assertNotEqual(symbols, [])
+
     @skipIf(sys.platform == "darwin", "Test is failing on macOS")
     def test_scanpipe_pipes_d2d_map_rust_symbols(self):
         input_dir = self.project1.input_path
@@ -1620,6 +1687,40 @@ def test_scanpipe_pipes_d2d_map_rust_symbols(self):
             ).count(),
         )
 
+    @skipIf(sys.platform == "darwin", "Test is failing on macOS")
+    def test_scanpipe_pipes_d2d_map_go_symbols(self):
+        input_dir = self.project1.input_path
+        input_resources = [
+            self.data / "d2d-macho/to-ollama.zip",
+            self.data / "d2d-macho/from-ollama.zip",
+        ]
+        copy_inputs(input_resources, input_dir)
+        self.from_files, self.to_files = d2d.get_inputs(self.project1)
+        inputs_with_codebase_path_destination = [
+            (self.from_files, self.project1.codebase_path / d2d.FROM),
+            (self.to_files, self.project1.codebase_path / d2d.TO),
+        ]
+        for input_files, codebase_path in inputs_with_codebase_path_destination:
+            for input_file_path in input_files:
+                scancode.extract_archive(input_file_path, codebase_path)
+
+        scancode.extract_archives(
+            self.project1.codebase_path,
+            recurse=True,
+        )
+        pipes.collect_and_create_codebase_resources(self.project1)
+        buffer = io.StringIO()
+        d2d.extract_binary_symbols(
+            project=self.project1, options=["Go"], logger=buffer.write
+        )
+        d2d.map_go_binaries_with_symbols(project=self.project1, logger=buffer.write)
+        self.assertEqual(
+            1,
+            CodebaseRelation.objects.filter(
+                project=self.project1, map_type="macho_symbols"
+            ).count(),
+        )
+
     @skipIf(sys.platform == "darwin", "Test is failing on macOS")
     def test_scanpipe_pipes_d2d_map_elf_symbols(self):
         input_dir = self.project1.input_path