openstack
/
ironic-python-agent
Code Issues Proposed changes

Fix MediaType() call and add tests for / and /v1

This commit is contained in:
Jim Rollenhagen 2014-03-17 10:13:17 -07:00
parent 1cc511ca11
commit 078b67dfa5
2 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
teeth_agent/api/controllers/v1/__init__.py
View File

@ -96,7 +96,9 @@ class V1(base.APIBase):
'',
bookmark=True)
]
v1.media_types = [MediaType('application/json')]
v1.media_types = [MediaType('application/json',
('application/vnd.openstack.'
'ironic-python-agent.v1+json'))]
return v1

14
teeth_agent/tests/api.py
View File

@ -44,6 +44,20 @@ class TestTeethAPI(unittest.TestCase):
client = test.Client(api, wrappers.BaseResponse)
return client.open(self._get_env_builder(method, path, data, query))
def test_root(self):
mock_agent = mock.MagicMock()
api_server = app.setup_app(mock_agent)
response = self._make_request(api_server, 'GET', '/')
self.assertEqual(response.status, '200 OK')
def test_v1_root(self):
mock_agent = mock.MagicMock()
api_server = app.setup_app(mock_agent)
response = self._make_request(api_server, 'GET', '/v1')
self.assertEqual(response.status, '200 OK')
def test_get_agent_status(self):
status = agent.TeethAgentStatus('TEST_MODE', time.time(), 'v72ac9')
mock_agent = mock.MagicMock()