split agent command test
This commit is contained in:
@@ -56,7 +56,7 @@ class TestTeethAPI(unittest.TestCase):
|
||||
self.assertEqual(data['started_at'], status.started_at)
|
||||
self.assertEqual(data['version'], status.version)
|
||||
|
||||
def test_execute_agent_command(self):
|
||||
def test_execute_agent_command_success(self):
|
||||
mock_agent = mock.MagicMock()
|
||||
api_server = api.TeethAgentAPIServer(mock_agent)
|
||||
|
||||
@@ -78,6 +78,10 @@ class TestTeethAPI(unittest.TestCase):
|
||||
data = json.loads(response.data)
|
||||
self.assertEqual(data, {'result': 'success'})
|
||||
|
||||
def test_execute_agent_command_validation(self):
|
||||
mock_agent = mock.MagicMock()
|
||||
api_server = api.TeethAgentAPIServer(mock_agent)
|
||||
|
||||
invalid_command = {}
|
||||
response = self._make_request(api_server,
|
||||
'POST',
|
||||
|
||||
Reference in New Issue
Block a user