def test_execute_agent_command_with_token(self): agent_token = str('0123456789' * 10) command = { 'name': 'do_things', 'params': {'key': 'value', 'wait': False, 'agent_token': agent_token}, } result = base.SyncCommandResult(command['name'], command['params'], True, {'test': 'result'}) self.mock_agent.validate_agent_token.return_value = True self.mock_agent.execute_command.return_value = result with mock.patch.object(result, 'join', autospec=True) as join_mock: response = self.post_json( '/commands?wait=false?agent_token=%s' % agent_token, command) self.assertFalse(join_mock.called) self.assertEqual(200, response.status_code) self.assertEqual(1, self.mock_agent.execute_command.call_count) self.assertEqual(1, self.mock_agent.validate_agent_token.call_count) args, kwargs = self.mock_agent.execute_command.call_args self.assertEqual(('do_things',), args) expected_result = result.serialize() data = response.json self.assertEqual(expected_result, data)
def test_execute_agent_command_success_with_true_wait(self): command = { 'name': 'do_things', 'params': {'key': 'value'}, } result = base.SyncCommandResult(command['name'], command['params'], True, {'test': 'result'}) self.mock_agent.execute_command.return_value = result with mock.patch.object(result, 'join', autospec=True) as join_mock: response = self.post_json('/commands?wait=true', command) join_mock.assert_called_once_with() self.assertEqual(200, response.status_code) self.assertEqual(1, self.mock_agent.execute_command.call_count) self.assertEqual(1, self.mock_agent.validate_agent_token.call_count) args, kwargs = self.mock_agent.execute_command.call_args self.assertEqual(('do_things',), args) self.assertEqual({'key': 'value'}, kwargs) expected_result = result.serialize() data = response.json self.assertEqual(expected_result, data)
def test_execute_command_success(self): expected_result = base.SyncCommandResult('fake', None, True, None) fake_ext = self.agent.get_extension('fake') fake_ext.execute = mock.Mock() fake_ext.execute.return_value = expected_result result = self.agent.execute_command('fake.sleep', sleep_info={"time": 1}) self.assertEqual(expected_result, result)
def test_get_command_result(self): cmd_result = base.SyncCommandResult('do_things', {'key': 'value'}, True, {'test': 'result'}) serialized_cmd_result = cmd_result.serialize() self.mock_agent.get_command_result.return_value = cmd_result response = self.get_json('/commands/abc123') self.assertEqual(response.status_code, 200) data = response.json self.assertEqual(data, serialized_cmd_result)
def test_list_command_results(self): cmd_result = base.SyncCommandResult(u'do_things', {u'key': u'value'}, True, {u'test': u'result'}) self.mock_agent.list_command_results.return_value = [ cmd_result, ] response = self.get_json('/commands') self.assertEqual(response.status_code, 200) self.assertEqual(response.json, { u'commands': [ cmd_result.serialize(), ], })