Update test_miner_param.py

NebraLtd · Apr 7, 2023 · bcd2ad4 · bcd2ad4
1 parent a6f7c28
commit bcd2ad4
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/hm_pyhelper/tests/test_miner_param.py b/hm_pyhelper/tests/test_miner_param.py
@@ -315,6 +315,16 @@ def test_get_ecc_key_missing(self):
         with self.assertRaises(UnknownVariantAttributeException):
             get_ecc_location()
 
+    @patch.dict('os.environ', {"VARIANT": "MISSING"})
+    def test_get_onboarding_key_missing_variant(self):
+        with self.assertRaises(UnknownVariantException):
+            get_onboarding_location()
+
+    @patch.dict('os.environ', {"VARIANT": "MISSING"})
+    def test_get_ecc_key_missing_variant(self):
+        with self.assertRaises(UnknownVariantException):
+            get_ecc_location()
+
     @patch('hm_pyhelper.miner_param.get_gateway_mfr_command',
            return_value=['gateway_mfr', 'arg1', 'arg2'])
     @patch('subprocess.run', side_effect=FileNotFoundError())