Add tests for core.secrets

.github/workflows/test-python.yml

@@ -26,8 +26,7 @@ jobs:
         run: ruff check .
       - name: Run tests
         run: |
-          coverage run --include='src/**/*' -m unittest
+          coverage run --include='src/**/*' -m pytest
           coverage report
         env:
-          __SKIP_SECRETS__: true
           AWS_REGION: us-east-1

chat/src/core/secrets.py

@@ -2,8 +2,7 @@
 import json
 import os
 
-def load_secrets():
-  SecretsPath = os.getenv('SECRETS_PATH')
+def load_secrets(SecretsPath=os.getenv('SECRETS_PATH')):
   EnvironmentMap = [
     ['API_TOKEN_SECRET', 'dcapi', 'api_token_secret'],
     ['OPENSEARCH_ENDPOINT', 'index', 'endpoint'],

chat/test/core/test_secrets.py

@@ -0,0 +1,29 @@
+import boto3
+import os
+import pytest
+from moto import mock_aws
+from unittest import TestCase
+
+from core.secrets import load_secrets
+
+@mock_aws
+@mock_aws
+@pytest.mark.filterwarnings("ignore::DeprecationWarning")
+class TestSecrets(TestCase):
+    def setUp(self):
+        client = boto3.client("secretsmanager", region_name="us-east-1")
+        client.create_secret(
+            Name="mock/infrastructure/index",
+            SecretString='{"endpoint": "https://opensearch-endpoint", "embedding_model": "opensearch-model"}')
+        client.create_secret(
+            Name="mock/config/dcapi",
+            SecretString='{"api_token_secret": "dcapi-token"}')
+
+    def test_load_secrets(self):
+        self.assertNotEqual('dcapi-token', os.getenv('API_TOKEN_SECRET'))
+        self.assertNotEqual('https://opensearch-endpoint', os.getenv('OPENSEARCH_ENDPOINT'))
+        self.assertNotEqual('opensearch-model', os.getenv('OPENSEARCH_MODEL_ID'))
+        load_secrets('mock')
+        self.assertEqual('dcapi-token', os.getenv('API_TOKEN_SECRET'))
+        self.assertEqual('https://opensearch-endpoint', os.getenv('OPENSEARCH_ENDPOINT'))
+        self.assertEqual('opensearch-model', os.getenv('OPENSEARCH_MODEL_ID'))