shared/datastore/tests/unit/connect_minio_test.py

@@ -0,0 +1,33 @@
+"""Definition of unittests for connect_minio function."""
+
+import os
+import unittest
+
+from shared.datastore import connect_minio
+
+
+class TestConnectMinio(unittest.TestCase):
+
+    def setUp(self):
+        # define relevant env vars
+        self.env_var_map = {
+            'MINIO_ENDPOINT': '192.168.1.2',
+            'MINIO_ACCESS_KEY': 'randomAccess_key',
+            'MINIO_SECRET_KEY': 'randomSecret_key',
+            'MINIO_BUCKET_NAME': 'test-bucket-name',
+        }
+        # set env vars
+        for key, val in self.env_var_map.items():
+            os.environ[key] = val
+
+    def tearDown(self):
+        # clear env vars
+        for key in self.env_var_map:
+            _ = os.environ.pop(key, default=None)
+
+    def test_should_fail_when_env_not_set(self):
+        # ensure env not set
+        self.tearDown()
+        # run test
+        with self.assertRaises(AssertionError):
+            connect_minio()