def test_set_val_permanent(self):
with patch.dict(os.environ, {}), patch.dict(
environ.__utils__,
{"reg.set_value": MagicMock(), "reg.delete_value": MagicMock()},
), patch("salt.utils.platform.is_windows", return_value=True):
environ.setval("key", "Test", permanent=True)
environ.__utils__["reg.set_value"].assert_called_with(
"HKCU", "Environment", "key", "Test"
)
def test_set_val_permanent_hklm(self):
with patch.dict(os.environ, {}), patch.dict(
environ.__utils__,
{"reg.set_value": MagicMock(), "reg.delete_value": MagicMock()},
), patch("salt.utils.platform.is_windows", return_value=True):
key = r"SYSTEM\CurrentControlSet\Control\Session Manager\Environment"
environ.setval("key", "Test", permanent="HKLM")
environ.__utils__["reg.set_value"].assert_called_with(
"HKLM", key, "key", "Test"
)
def test_setval(self):
"""
Test for set a single salt process environment variable. Returns True
on success.
"""
mock = MagicMock(return_value=None)
with patch.dict(os.environ, {}):
self.assertEqual(environ.setval("key", False, True), None)
mock = MagicMock(side_effect=Exception())
with patch.dict(os.environ, {}):
self.assertFalse(environ.setval("key", False, True))
mock_environ = {}
with patch.dict(os.environ, mock_environ):
self.assertEqual(environ.setval("key", False), "")
with patch.dict(os.environ, mock_environ):
self.assertFalse(environ.setval("key", True))
def test_setval(self):
'''
Test for set a single salt process environment variable. Returns True
on success.
'''
mock = MagicMock(return_value=None)
with patch.dict(os.environ, {}):
self.assertEqual(environ.setval('key', False, True), None)
mock = MagicMock(side_effect=Exception())
with patch.dict(os.environ, {}):
self.assertFalse(environ.setval('key', False, True))
mock_environ = {}
with patch.dict(os.environ, mock_environ):
self.assertEqual(environ.setval('key', False), '')
with patch.dict(os.environ, mock_environ):
self.assertFalse(environ.setval('key', True))
def test_set_val_permanent(self):
with patch.dict(os.environ, {}):
environ.__salt__['reg.set_value'] = MagicMock()
environ.__salt__['reg.delete_value'] = MagicMock()
environ.setval('key', 'Test', permanent=True)
environ.__salt__['reg.set_value'].assert_called_with('HKCU', 'Environment', 'key', 'Test')
environ.setval('key', False, false_unsets=True, permanent=True)
environ.__salt__['reg.set_value'].asset_not_called()
environ.__salt__['reg.delete_value'].assert_called_with('HKCU', 'Environment', 'key')
key = r'SYSTEM\CurrentControlSet\Control\Session Manager\Environment'
environ.setval('key', 'Test', permanent='HKLM')
environ.__salt__['reg.set_value'].assert_called_with('HKLM', key, 'key', 'Test')
def test_set_val_permanent(self):
with patch.dict(os.environ, {}), \
patch.dict(environ.__salt__, {'reg.set_value': MagicMock(),
'reg.delete_value': MagicMock()}), \
patch('salt.utils.is_windows', MagicMock(return_value=True)):
environ.setval('key', 'Test', permanent=True)
environ.__salt__['reg.set_value'].assert_called_with('HKCU', 'Environment', 'key', 'Test')
environ.setval('key', False, false_unsets=True, permanent=True)
environ.__salt__['reg.set_value'].asset_not_called()
environ.__salt__['reg.delete_value'].assert_called_with('HKCU', 'Environment', 'key')
key = r'SYSTEM\CurrentControlSet\Control\Session Manager\Environment'
environ.setval('key', 'Test', permanent='HKLM')
environ.__salt__['reg.set_value'].assert_called_with('HKLM', key, 'key', 'Test')
def test_set_val_permanent(self):
with patch.dict(os.environ, {}), patch.dict(
environ.__salt__,
{
"reg.set_value": MagicMock(),
"reg.delete_value": MagicMock()
},
), patch("salt.utils.platform.is_windows",
MagicMock(return_value=True)):
environ.setval("key", "Test", permanent=True)
environ.__salt__["reg.set_value"].assert_called_with(
"HKCU", "Environment", "key", "Test")
environ.setval("key", False, false_unsets=True, permanent=True)
environ.__salt__["reg.set_value"].asset_not_called()
environ.__salt__["reg.delete_value"].assert_called_with(
"HKCU", "Environment", "key")
key = r"SYSTEM\CurrentControlSet\Control\Session Manager\Environment"
environ.setval("key", "Test", permanent="HKLM")
environ.__salt__["reg.set_value"].assert_called_with(
"HKLM", key, "key", "Test")
