def test_reset(self, os_remove_mock, os_walk_mock, sys_exit_mock, config_parser_mock, open_mock, os_path_join_mock, os_rmdir_mock):
# Agent config update
config_mock = MagicMock()
os_walk_mock.return_value = [('/', ('',), ('file1.txt', 'file2.txt'))]
config_parser_mock.return_value= config_mock
config_mock.get('server', 'hostname').return_value = "old_host"
main.reset_agent(["test", "reset", "new_hostname"])
self.assertEqual(config_mock.get.call_count, 3)
self.assertEqual(config_mock.set.call_count, 1)
self.assertEqual(os_remove_mock.call_count, 2)
self.assertTrue(sys_exit_mock.called)
def test_reset(self, os_remove_mock, os_walk_mock, sys_exit_mock, config_parser_mock, open_mock, os_path_join_mock, os_rmdir_mock):
# Agent config update
config_mock = MagicMock()
os_walk_mock.return_value = [('/', ('',), ('file1.txt', 'file2.txt'))]
config_parser_mock.return_value= config_mock
config_mock.get('server', 'hostname').return_value = "old_host"
main.reset_agent(["test", "reset", "new_hostname"])
self.assertEqual(config_mock.get.call_count, 3)
self.assertEqual(config_mock.set.call_count, 1)
self.assertEqual(os_remove_mock.call_count, 2)
self.assertTrue(sys_exit_mock.called)
def test_reset_invalid_path(self, os_remove_mock, os_walk_mock,
os_exit_mock, config_parser_mock, open_mock,
os_path_join_mock, os_rmdir_mock):
# Agent config file cannot be accessed
config_mock = MagicMock()
os_walk_mock.return_value = [('/', ('', ), ('file1.txt', 'file2.txt'))]
config_parser_mock.return_value = config_mock
config_mock.get('server', 'hostname').return_value = "old_host"
open_mock.side_effect = Exception("Invalid Path!")
try:
main.reset_agent(["test", "reset", "new_hostname"])
self.fail("Should have thrown exception!")
except:
self.assertTrue(True)
def test_reset_invalid_path(self, os_remove_mock, os_walk_mock, sys_exit_mock,
config_parser_mock, open_mock, os_path_join_mock, os_rmdir_mock):
# Agent config file cannot be accessed
config_mock = MagicMock()
os_walk_mock.return_value = [('/', ('',), ('file1.txt', 'file2.txt'))]
config_parser_mock.return_value= config_mock
config_mock.get('server', 'hostname').return_value = "old_host"
open_mock.side_effect = Exception("Invalid Path!")
try:
main.reset_agent(["test", "reset", "new_hostname"])
self.fail("Should have thrown exception!")
except:
self.assertTrue(True)
self.assertTrue(sys_exit_mock.called)