Python setval示例

示例#1

文件： test_environ.py 项目： arizvisa/saltstack-salt

    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"
            )

示例#2

文件： test_environ.py 项目： arizvisa/saltstack-salt

    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"
            )

示例#3

    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))

示例#4

文件： environ_test.py 项目： HowardMei/saltstack

    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))

示例#5

文件： environ_test.py 项目： HowardMei/saltstack

    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')

示例#6

文件： test_environ.py 项目： bsemar/IntCont

    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')

示例#7

    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")