CM-55749-Tests coverage

Author: mateusz-sterczewski

cycode/cli/apps/scan/code_scanner.py

@@ -56,8 +56,9 @@ def scan_disk_files(ctx: typer.Context, paths: tuple[str, ...]) -> None:
             is_cycodeignore_allowed=is_cycodeignore_allowed_by_scan_config(ctx),
         )
 
-        # Add entrypoint.cycode file at each root path to mark the scan root
-        for root_path in paths:
+        # Add entrypoint.cycode file at root path to mark the scan root (only for single path)
+        if len(paths) == 1:
+            root_path = paths[0]
             absolute_root_path = get_absolute_path(root_path)
             entrypoint_path = get_path_by_os(os.path.join(absolute_root_path, consts.CYCODE_ENTRYPOINT_FILENAME))
             entrypoint_document = Document(

tests/cli/commands/scan/test_code_scanner.py

@@ -1,6 +1,8 @@
 import os
+from unittest.mock import MagicMock, Mock, patch
 
 from cycode.cli import consts
+from cycode.cli.apps.scan.code_scanner import scan_disk_files
 from cycode.cli.files_collector.file_excluder import _is_file_relevant_for_sca_scan
 from cycode.cli.files_collector.path_documents import _generate_document
 from cycode.cli.models import Document
@@ -72,3 +74,47 @@ def test_generate_document() -> None:
     assert isinstance(generated_tfplan_document, Document)
     assert generated_tfplan_document.path.endswith('.tf')
     assert generated_tfplan_document.is_git_diff_format == is_git_diff
+
+
+@patch('cycode.cli.apps.scan.code_scanner.get_relevant_documents')
+@patch('cycode.cli.apps.scan.code_scanner.scan_documents')
+@patch('cycode.cli.apps.scan.code_scanner.get_scan_parameters')
+def test_entrypoint_cycode_added_to_documents(
+    mock_get_scan_parameters: Mock,
+    mock_scan_documents: Mock,
+    mock_get_relevant_documents: Mock,
+) -> None:
+    """Test that entrypoint.cycode file is added to documents in scan_disk_files."""
+    # Arrange
+    mock_ctx = MagicMock()
+    mock_ctx.obj = {
+        'scan_type': consts.SAST_SCAN_TYPE,
+        'progress_bar': MagicMock(),
+    }
+    mock_get_scan_parameters.return_value = {}
+    
+    mock_documents = [
+        Document('/test/path/file1.py', 'content1', is_git_diff_format=False),
+        Document('/test/path/file2.js', 'content2', is_git_diff_format=False),
+    ]
+    mock_get_relevant_documents.return_value = mock_documents.copy()
+    test_path = '/Users/test/repositories'
+
+    # Act
+    scan_disk_files(mock_ctx, (test_path,))
+
+    # Assert
+    call_args = mock_scan_documents.call_args
+    documents_passed = call_args[0][1]
+
+    # Verify entrypoint document was added
+    entrypoint_docs = [
+        doc for doc in documents_passed if doc.path.endswith(consts.CYCODE_ENTRYPOINT_FILENAME)
+    ]
+    assert len(entrypoint_docs) == 1
+
+    entrypoint_doc = entrypoint_docs[0]
+    assert entrypoint_doc.path == os.path.join(test_path, consts.CYCODE_ENTRYPOINT_FILENAME)
+    assert entrypoint_doc.content == ''
+    assert entrypoint_doc.is_git_diff_format is False
+    assert entrypoint_doc.absolute_path == entrypoint_doc.path