mirror of
https://github.com/projecthorus/sondehub-infra.git
synced 2025-01-18 18:56:26 +00:00
57 lines
2.4 KiB
Python
57 lines
2.4 KiB
Python
import config_handler
|
|
|
|
import unittest
|
|
from unittest.mock import MagicMock, call, patch
|
|
|
|
# Mock AWS API calls
|
|
secret_call = {
|
|
'ARN': 'arn:aws:secretsmanager:us-west-2:123456789012:secret:MyTestDatabaseSecret-a1b2c3',
|
|
'CreatedDate': 1523477145.713,
|
|
'Name': 'MyTestDatabaseSecret',
|
|
'SecretString': '{\n "PASSWORD":"test_password"\n}\n',
|
|
'VersionId': 'EXAMPLE1-90ab-cdef-fedc-ba987SECRET1',
|
|
'VersionStages': [
|
|
'AWSCURRENT',
|
|
],
|
|
'ResponseMetadata': {
|
|
'...': '...',
|
|
},
|
|
}
|
|
|
|
class TestConfigHandler(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
config_handler.get.cache_clear()
|
|
|
|
def test_env(self):
|
|
with patch.dict(config_handler.os.environ,{ "MQTT_PASSWORD": "test_password" }, clear=True):
|
|
return_value = config_handler.get("MQTT", "PASSWORD")
|
|
self.assertEqual(return_value, "test_password")
|
|
|
|
@patch('botocore.client.BaseClient._make_api_call', return_value=secret_call)
|
|
def test_sm(self, MockApiCall):
|
|
with patch.dict(config_handler.os.environ,{}, clear=True): #ensure that local env variables don't influence the tests
|
|
return_value = config_handler.get("MQTT", "PASSWORD")
|
|
MockApiCall.assert_called()
|
|
self.assertEqual(return_value, "test_password")
|
|
|
|
@patch('botocore.client.BaseClient._make_api_call', return_value=secret_call)
|
|
def test_cache(self, MockApiCall):
|
|
with patch.dict(config_handler.os.environ,{}, clear=True): #ensure that local env variables don't influence the tests
|
|
return_value = config_handler.get("MQTT", "PASSWORD")
|
|
return_value = config_handler.get("MQTT", "PASSWORD")
|
|
MockApiCall.assert_called_once()
|
|
self.assertEqual(return_value, "test_password")
|
|
|
|
@patch('botocore.client.BaseClient._make_api_call', return_value=secret_call)
|
|
def test_not_found(self, MockApiCall):
|
|
with patch.dict(config_handler.os.environ,{}, clear=True):
|
|
self.assertRaises(KeyError, config_handler.get, "MQTT", "NOTPASSWORD")
|
|
|
|
def test_default(self):
|
|
with patch.dict(config_handler.os.environ,{}, clear=True):
|
|
return_value = config_handler.get("MQTT", "PASSWORD", "test_password_abc")
|
|
self.assertEqual(return_value, "test_password_abc")
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|