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()