class TestUserManager(TestCase):
def setUp(self):
self._user_manager = UserManager()
UserModel.objects.create(user_id=1775634455973687126, access_token="token1", account="account1", password="******")
UserModel.objects.create(user_id=3589683559412082469, access_token="token2", account="account2", password="******")
def test_get_user_from_token(self):
user = self._user_manager.get_user_from_token("token1")
self.assertEqual(user.account, 'account1')
def test_get_user_from_token_exception(self):
try:
user = self._user_manager.get_user_from_token("no this token")
self.assertTrue(False)
except AuthorizationError:
pass
def test_login(self):
user = self._user_manager.login("account1", "password1")
self.assertEqual(user["accessToken"], "token1")
self.assertEqual(user["account"], "account1")
def test_login_fail(self):
try:
user = self._user_manager.login("no this account", "no this password")
self.assertTrue(False)
except AuthorizationError:
pass
def tearDown(self):
UserModel.objects.all().delete()
class TestUserManager(TestCase):
def setUp(self):
self._user_manager = UserManager()
def test_register(self):
fb_token = Utility.generate_token()
user = self._user_manager.register(fb_token)
self.assertEqual(fb_token, user.fb_token, "fb token error")
def tearDown(self):
pass
def register(request):
try:
request = RequestChecker(request)
# header
# POST data
data = {
"fb_token": request.get_data("fb_token"),
}
# action
user = UserManager.register(data["fb_token"])
return JSONResponse.output({"access_token": user.token})
except Error as error:
return JSONResponse.output(error)
def setUp(self):
self._user_manager = UserManager()
def setUp(self):
self._user_manager = UserManager()
UserModel.objects.create(user_id=1775634455973687126, access_token="token1", account="account1", password="******")
UserModel.objects.create(user_id=3589683559412082469, access_token="token2", account="account2", password="******")