From 4c0dc2490cc240addbd5e9fc8fde384b3cd3b771 Mon Sep 17 00:00:00 2001 From: Russell Haering Date: Fri, 20 Dec 2013 16:08:07 -0800 Subject: [PATCH] split agent command test --- teeth_agent/tests/api.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/teeth_agent/tests/api.py b/teeth_agent/tests/api.py index 3adf70cab..446bd0ffd 100644 --- a/teeth_agent/tests/api.py +++ b/teeth_agent/tests/api.py @@ -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',