def test_vk_auth_after_normal_registartion(self):
os.environ["VK_APP_SECRET_KEY"] = "12345"
vk_id = 42
vk_concated_string = "blablabla=1&blabla=2"
sig = md5(vk_concated_string.replace("&", "") + "12345")
# Обычная регистрация, получаем токен:
credentials = Credentials()
credentials.phone = "+79263435016"
credentials.password = "******"
register_result = self.service.register(credentials)
verification_code = register_result["verification"]["send_code"]
self.service.verify_phone(credentials, verification_code)
auth_result = self.service.authenticate(credentials)
self.assertEqual(1, auth_result[0])
token = auth_result[1]
# Используем токен и укажем vk_id:
new_credentials = Credentials()
new_credentials.token = token
new_credentials.vk_id = vk_id
vk_auth_result = self.service.authenticate_vk(new_credentials, vk_concated_string, sig)
self.assertTrue(isinstance(vk_auth_result, tuple))
self.assertEqual(1, vk_auth_result[0])
self.assertEqual(token, vk_auth_result[1])
# Теперь можно и без токена:
new_credentials = Credentials()
new_credentials.vk_id = vk_id
vk_auth_result = self.service.authenticate_vk(new_credentials, vk_concated_string, sig)
self.assertTrue(isinstance(vk_auth_result, tuple))
self.assertEqual(1, vk_auth_result[0])
self.assertEqual(32, len(vk_auth_result[1]))
self.assertNotEqual(token, vk_auth_result[1])
def test_vk_auth_fail(self):
os.environ["VK_APP_SECRET_KEY"] = "12345"
vk_id = 42
vk_concated_string = "blablabla=1&blabla=2"
sig = md5(vk_concated_string.replace("&", "") + "12345")
credentials = Credentials()
credentials.vk_id = vk_id
self.assertRaises(
IncorrectOAuthSignature,
self.service.authenticate_vk, credentials, vk_concated_string, sig+"1"
)
def build_credentials(cls, request: Request):
""" Метод для создания экземпляра класса Credentials на основе предоставленного объекта Request
:param request:
:return:
"""
credentials = Credentials()
credentials.email = request.get("email").lower() if request.get("email", None) else None
credentials.phone = normalize_phone_number(request.get("phone")) if request.get("phone", False) else None
credentials.token_name = request.get("token_name", "")
credentials.token = request.get("%stoken" % credentials.token_name, None)
credentials.password = request.get("password", None)
credentials.vk_id = request.get("vk_id", None)
return credentials
def test_authenticate_vk(self):
os.environ["VK_APP_SECRET_KEY"] = "12345"
vk_id = 42
vk_concated_string = "blablabla=1&blabla=2"
sig = md5(vk_concated_string.replace("&", "") + "12345")
credentials = Credentials()
credentials.vk_id = vk_id
response = self.app.get(
"/v1/authenticate_vk/", {"vk_id": "42", "vk_concated_string": vk_concated_string, "signature": sig}
)
response = json.loads(response.body.decode())
self.assertTrue(isinstance(response, list))
self.assertTrue(len(response), 2)
self.assertEqual(1, response[0])
self.assertEqual(32, len(response[1]))
def build_credentials(cls, request: Request):
""" Метод для создания экземпляра класса Credentials на основе предоставленного объекта Request
:param request:
:return:
"""
credentials = Credentials()
credentials.email = request.get("email").lower() if request.get(
"email", None) else None
credentials.phone = normalize_phone_number(
request.get("phone")) if request.get("phone", False) else None
credentials.token_name = request.get("token_name", "")
credentials.token = request.get("%stoken" % credentials.token_name,
None)
credentials.password = request.get("password", None)
credentials.vk_id = request.get("vk_id", None)
return credentials
def test_init_registration_with_vk(self):
os.environ["VK_APP_SECRET_KEY"] = "12345"
vk_id = 42
vk_concated_string = "blablabla=1&blabla=2"
sig = md5(vk_concated_string.replace("&", "") + "12345")
credentials = Credentials()
credentials.vk_id = vk_id
result = self.service.authenticate_vk(credentials, vk_concated_string, sig)
self.assertTrue(isinstance(result, tuple))
self.assertEqual(1, result[0])
self.assertEqual(32, len(result[1]))
token = result[1]
# Используя токен, полученный при регистрации через вк авторизуемся стандартным путем:
credentials.token = token
auth_result = self.service.authenticate(credentials)
self.assertTrue(isinstance(auth_result, tuple))
self.assertEqual(1, auth_result[0])
self.assertEqual(token, auth_result[1])
