tests: update imports in tests

.github/workflows/test.yml

@@ -10,7 +10,7 @@ on:
 env:
   PYTHONPATH: ./src # Needed for tests to discover whispr package
 jobs:
-  semgrep-scan:
+  test:
     runs-on: ubuntu-22.04
 
     steps:

tests/test_aws.py

@@ -6,11 +6,11 @@
 import botocore.exceptions
 import structlog
 
-from src.whispr.vault import SimpleVault
-from src.whispr.aws import AWSVault
+from whispr.vault import SimpleVault
+from whispr.aws import AWSVault
 
 
-class TestAWSVault(unittest.TestCase):
+class AWSVaultTestCase(unittest.TestCase):
     """Unit tests for AWSVault class, which fetches secrets from AWS Secrets Manager."""
 
     def setUp(self):
@@ -47,7 +47,7 @@ def test_fetch_secrets_resource_not_found(self):
             secret_name="non_existent_secret",
         )
 
-    @patch("src.whispr.aws.AWSVault.fetch_secrets")
+    @patch("whispr.aws.AWSVault.fetch_secrets")
     def test_fetch_secrets_unrecognized_client_exception(self, mock_fetch_secrets):
         """Test fetch_secrets handles UnrecognizedClientException gracefully."""
         mock_fetch_secrets.side_effect = botocore.exceptions.ClientError(

tests/test_gcp.py

@@ -6,11 +6,11 @@
 import google.api_core.exceptions
 import structlog
 
-from src.whispr.vault import SimpleVault
-from src.whispr.gcp import GCPVault
+from whispr.vault import SimpleVault
+from whispr.gcp import GCPVault
 
 
-class TestGCPVault(unittest.TestCase):
+class GCPVaultTestCase(unittest.TestCase):
     """Unit tests for GCPVault class, which fetches secrets from GCP Secrets Manager."""
 
     def setUp(self):

tests/test_vault.py

@@ -2,12 +2,13 @@
 
 import unittest
 from unittest.mock import Mock, patch
+
 import structlog
 
-from src.whispr.vault import SimpleVault
+from whispr.vault import SimpleVault
 
 
-class SimpleVaultTest(unittest.TestCase):
+class SimpleVaultTestCase(unittest.TestCase):
     """Tests for Vault"""
 
     def setUp(self):