diff --git a/hm_pyhelper/tests/test_miner_param.py b/hm_pyhelper/tests/test_miner_param.py index 486c179..7b59a6e 100644 --- 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())