class TestManageRefresh(TestCase):
def setUp(self):
self.admin_strategy = MockRefreshTokenAdminStrategy()
self.user_strategy = MockRefreshTokenUserStrategy()
self.manage = ManageUser()
def tearDown(self):
self.admin_strategy = None
self.user_strategy = None
self.manage = None
def test_success_admin_refresh(self):
mock_return = dict(token_id="admin token",
refresh_token="admin refresh token")
self.admin_strategy.refresh = MagicMock(return_value=mock_return)
tokens = self.manage.refresh(self.admin_strategy, "refresh token")
self.assertDictEqual(tokens, mock_return)
def test_success_user_refresh(self):
mock_return = dict(token_id="user token",
refresh_token="user refresh token")
self.user_strategy.refresh = MagicMock(return_value=mock_return)
tokens = self.manage.refresh(self.user_strategy, "refresh token")
self.assertDictEqual(tokens, mock_return)
class TestManageLogin(TestCase):
def setUp(self):
self.admin_strategy = MockLoginAdminStrategy()
self.user_strategy = MockLoginUserStrategy()
self.manage = ManageUser()
def tearDown(self):
self.admin_strategy = None
self.user_strategy = None
self.manage = None
def test_success_admin_login(self):
mock_return = dict(token_id="admin token",
refresh_token="admin refresh token")
self.admin_strategy.login = MagicMock(return_value=mock_return)
tokens = self.manage.login(self.admin_strategy, dict())
self.assertDictEqual(tokens, mock_return)
def test_success_user_login(self):
mock_return = dict(token_id="user token",
refresh_token="user refresh token")
self.user_strategy.login = MagicMock(return_value=mock_return)
tokens = self.manage.login(self.user_strategy, dict())
self.assertDictEqual(tokens, mock_return)
def test_error_login(self):
self.user_strategy.login = MagicMock(return_value=None)
with self.assertRaises(AuthenticationError):
self.manage.login(self.user_strategy, dict())
def get(self):
refresh_token = request.headers.get("REFRESH_TOKEN").split(" ")[-1]
maneger = ManageUser()
strategy = RefreshTokenAdminStrategy()
tokens = maneger.refresh(strategy, refresh_token)
return Utils.response_auth(tokens["token_id"], tokens["refresh_token"])
def post(self): manage = ManageUser() strategy = LoginUserStrategy() try: tokens = manage.login(strategy, request.json) except InvalidParametersError as e: return Utils.response(status=e.status, data=e.errors, message="Invalid parameters") except AuthenticationError as e: return Utils.response(status=e.status, data=e.errors, message="Invalid credentials") return Utils.response_auth(tokens["token_id"], tokens["refresh_token"])
def post(self): manage = ManageUser() strategy = SignupAdminStrategy() try: admin = manage.signup(strategy, request.json) data = AdminCreateSchema().dumps(admin) except InvalidParametersError as e: return Utils.response(status=e.status, data=e.errors, message="Invalid parameters") except UserExistsError as e: return Utils.response(status=e.status, data=e.errors, message="Parameter must be unique") return Utils.response(status=201, data=json.loads(data), message="Request succesfull")
class TestManageSignup(TestCase):
def setUp(self):
self.admin_strategy = MockSignupAdminStrategy()
self.user_strategy = MockSignupUserStrategy()
self.manage = ManageUser()
def tearDown(self):
self.admin_strategy = None
self.user_strategy = None
self.manage = None
def test_success_admin_signup(self):
self.admin_strategy.signup = MagicMock(return_value=Admin())
admin = self.manage.signup(self.admin_strategy, dict())
self.assertIsInstance(admin, Admin)
def test_success_user_signup(self):
self.user_strategy.signup = MagicMock(return_value=User())
user = self.manage.signup(self.user_strategy, dict())
self.assertIsInstance(user, User)
def setUp(self):
self.admin_strategy = MockRefreshTokenAdminStrategy()
self.user_strategy = MockRefreshTokenUserStrategy()
self.manage = ManageUser()
def setUp(self):
self.admin_strategy = MockSignupAdminStrategy()
self.user_strategy = MockSignupUserStrategy()
self.manage = ManageUser()
def setUp(self):
self.admin_strategy = MockLoginAdminStrategy()
self.user_strategy = MockLoginUserStrategy()
self.manage = ManageUser()