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)