adjusting the tests

bridgecrewio · Oct 17, 2023 · a4a4b22 · a4a4b22
1 parent 130b2fc
commit a4a4b22
Showing 1 changed file with 41 additions and 32 deletions.
diff --git a/tests/sca_package_2/test_runner.py b/tests/sca_package_2/test_runner.py
@@ -1,3 +1,4 @@
+import os
 from pathlib import Path
 from mock.mock import MagicMock
 
@@ -74,40 +75,48 @@ def test_upload_scannable_files_exclude_go_and_requirements():
 
 
 def test_upload_scannable_files_file_config():
-    # when
-    input_output_paths = Runner().upload_package_files(
-        root_path=None,
-        files=[
-            str(EXAMPLES_DIR / 'requirements.txt'),
-            str(EXAMPLES_DIR / 'go.sum'),
-            str(EXAMPLES_DIR / 'package-lock.json'),
-            str(EXAMPLES_DIR / 'package.json'),
-            str(EXAMPLES_DIR / 'go.mod'),
-            str(EXAMPLES_DIR / 'Microsoft.NET.Sdk.csproj')
-        ],
-        excluded_paths=set(),
-        excluded_file_names=set()
-    )
-    # expected
-    expected_output = {
-        FileToPersist(full_file_path=str(EXAMPLES_DIR / 'requirements.txt'),
-                      s3_file_key='requirements.txt'),
-        FileToPersist(full_file_path=str(EXAMPLES_DIR / 'go.sum'),
-                      s3_file_key='go.sum'),
-        FileToPersist(full_file_path=str(EXAMPLES_DIR / 'package-lock.json'),
-                      s3_file_key='package-lock.json'),
-        FileToPersist(full_file_path=str(EXAMPLES_DIR / 'package.json'),
-                      s3_file_key='package.json'),
-        FileToPersist(full_file_path=str(EXAMPLES_DIR / 'go.mod'),
-                      s3_file_key='go.mod'),
-        FileToPersist(full_file_path=str(EXAMPLES_DIR / 'Microsoft.NET.Sdk.csproj'),
-                      s3_file_key='Microsoft.NET.Sdk.csproj')
-    }
+    origin_cwd = os.getcwd()
+    try:
+        # setup
+        os.chdir(str(Path(__file__).parent))
+
+        # when
+        input_output_paths = Runner().upload_package_files(
+            root_path=None,
+            files=[
+                str(EXAMPLES_DIR / 'requirements.txt'),
+                str(EXAMPLES_DIR / 'go.sum'),
+                str(EXAMPLES_DIR / 'package-lock.json'),
+                str(EXAMPLES_DIR / 'package.json'),
+                str(EXAMPLES_DIR / 'go.mod'),
+                str(EXAMPLES_DIR / 'Microsoft.NET.Sdk.csproj')
+            ],
+            excluded_paths=set(),
+            excluded_file_names=set()
+        )
+        # expected (paths are in related to the test-working-dir)
+        expected_output = {
+            FileToPersist(full_file_path=str(EXAMPLES_DIR / 'requirements.txt'),
+                          s3_file_key='examples/requirements.txt'),
+            FileToPersist(full_file_path=str(EXAMPLES_DIR / 'go.sum'),
+                          s3_file_key='examples/go.sum'),
+            FileToPersist(full_file_path=str(EXAMPLES_DIR / 'package-lock.json'),
+                          s3_file_key='examples/package-lock.json'),
+            FileToPersist(full_file_path=str(EXAMPLES_DIR / 'package.json'),
+                          s3_file_key='examples/package.json'),
+            FileToPersist(full_file_path=str(EXAMPLES_DIR / 'go.mod'),
+                          s3_file_key='examples/go.mod'),
+            FileToPersist(full_file_path=str(EXAMPLES_DIR / 'Microsoft.NET.Sdk.csproj'),
+                          s3_file_key='examples/Microsoft.NET.Sdk.csproj')
+        }
 
-    # then
-    assert len(input_output_paths) == 6
+        # then
+        assert len(input_output_paths) == 6
 
-    assert set(input_output_paths) == expected_output
+        assert set(input_output_paths) == expected_output
+    finally:
+        # teardown
+        os.chdir(origin_cwd)
 
 
 def test_run(sca_package_2_report):