diff options
Diffstat (limited to 'testapi/opnfv_testapi/tests/unit/handlers/test_token.py')
-rw-r--r-- | testapi/opnfv_testapi/tests/unit/handlers/test_token.py | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/testapi/opnfv_testapi/tests/unit/handlers/test_token.py b/testapi/opnfv_testapi/tests/unit/handlers/test_token.py new file mode 100644 index 0000000..e8b746d --- /dev/null +++ b/testapi/opnfv_testapi/tests/unit/handlers/test_token.py @@ -0,0 +1,52 @@ +# All rights reserved. This program and the accompanying materials +# are made available under the terms of the Apache License, Version 2.0 +# which accompanies this distribution, and is available at +# http://www.apache.org/licenses/LICENSE-2.0 + +import httplib +import unittest + +from tornado import web + +from opnfv_testapi.common import message +from opnfv_testapi.tests.unit import executor +from opnfv_testapi.tests.unit import fake_pymongo +from opnfv_testapi.tests.unit.handlers import test_result + + +class TestTokenCreateResult(test_result.TestResultBase): + def get_app(self): + from opnfv_testapi.router import url_mappings + return web.Application( + url_mappings.mappings, + db=fake_pymongo, + debug=True, + auth=True + ) + + def setUp(self): + super(TestTokenCreateResult, self).setUp() + fake_pymongo.tokens.insert({"access_token": "12345"}) + + @executor.create(httplib.FORBIDDEN, message.invalid_token()) + def test_resultCreateTokenInvalid(self): + self.headers['X-Auth-Token'] = '1234' + return self.req_d + + @executor.create(httplib.UNAUTHORIZED, message.unauthorized()) + def test_resultCreateTokenUnauthorized(self): + if 'X-Auth-Token' in self.headers: + self.headers.pop('X-Auth-Token') + return self.req_d + + @executor.create(httplib.OK, '_create_success') + def test_resultCreateTokenSuccess(self): + self.headers['X-Auth-Token'] = '12345' + return self.req_d + + def _create_success(self, body): + self.assertIn('CreateResponse', str(type(body))) + + +if __name__ == '__main__': + unittest.main() |