add unit tests of agent API

teeth_agent/tests/api.py

@@ -0,0 +1,91 @@
+"""
+Copyright 2013 Rackspace, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+   http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+import json
+import mock
+import time
+import unittest
+
+from werkzeug import test
+from werkzeug import wrappers
+
+from teeth_agent import api
+from teeth_agent import base
+
+
+class TestTeethAPI(unittest.TestCase):
+    def _get_env_builder(self, method, path, data=None, query=None):
+        if data is not None:
+            data = json.dumps(data)
+
+        return test.EnvironBuilder(method=method,
+                                   path=path,
+                                   data=data,
+                                   content_type='application/json',
+                                   query_string=query)
+
+    def _make_request(self, api, method, path, data=None, query=None):
+        client = test.Client(api, wrappers.BaseResponse)
+        return client.open(self._get_env_builder(method, path, data, query))
+
+    def test_get_agent_status(self):
+        status = base.TeethAgentStatus('TEST_MODE', time.time(), 'v72ac9')
+        mock_agent = mock.MagicMock()
+        mock_agent.get_status.return_value = status
+        api_server = api.TeethAgentAPIServer(mock_agent)
+
+        response = self._make_request(api_server, 'GET', '/v1.0/status')
+        mock_agent.get_status.assert_called_once_with()
+
+        self.assertEqual(response.status_code, 200)
+        data = json.loads(response.data)
+        self.assertEqual(data['mode'], status.mode)
+        self.assertEqual(data['started_at'], status.started_at)
+        self.assertEqual(data['version'], status.version)
+
+    def test_execute_agent_command(self):
+        mock_agent = mock.MagicMock()
+        api_server = api.TeethAgentAPIServer(mock_agent)
+
+        valid_command = {
+            'name': 'do_things',
+            'params': {'key': 'value'},
+        }
+
+        response = self._make_request(api_server,
+                                      'POST',
+                                      '/v1.0/command',
+                                      data=valid_command)
+
+        self.assertEqual(mock_agent.execute_command.call_count, 1)
+        args, kwargs = mock_agent.execute_command.call_args
+        self.assertEqual(len(args), 1)
+        self.assertEqual(len(kwargs), 0)
+        self.assertIsInstance(args[0], api.AgentCommand)
+        self.assertEqual(args[0].name, valid_command['name'])
+        self.assertEqual(args[0].params, valid_command['params'])
+        self.assertEqual(response.status_code, 200)
+        data = json.loads(response.data)
+        self.assertEqual(data, {'result': 'success'})
+
+        invalid_command = {}
+        response = self._make_request(api_server,
+                                      'POST',
+                                      '/v1.0/command',
+                                      data=invalid_command)
+        self.assertEqual(response.status_code, 400)
+        data = json.loads(response.data)
+        self.assertEqual(data['details'], 'Missing command \'name\' field.')