class TokenGeneratoryFactoryTests(FantasticoUnitTestsCase):
'''This class provides the tests suite for token generator factory.'''
_factory = None
def init(self):
'''This method is invoked automatically in order to set common dependencies for test cases.'''
self._factory = TokenGeneratorFactory()
def test_logintokengenerator_get(self):
'''This test case ensures the tokens generator factory can correctly instantiate a login token generator.'''
login_generator = self._factory.get_generator(TokenGeneratorFactory.LOGIN_TOKEN, Mock())
access_generator = self._factory.get_generator(TokenGeneratorFactory.ACCESS_TOKEN, Mock())
self.assertIsNotNone(login_generator)
self.assertIsNotNone(access_generator)
self.assertIsInstance(login_generator, LoginTokenGenerator)
self.assertIsInstance(access_generator, AccessTokenGenerator)
def test_unknonwgenerator_get_exception(self):
'''This test case ensures a concrete exception is raised if the requested token generator is not supported.'''
with self.assertRaises(OAuth2InvalidTokenTypeError) as ctx:
self._factory.get_generator("Unknown", Mock())
self.assertEqual("Unknown", ctx.exception.token_type)
def init(self):
'''This method is invoked automatically in order to set common dependencies for test cases.'''
self._factory = TokenGeneratorFactory()
