1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
from tests import SiteTest, manager
class TestOauthBackend(SiteTest):
""" Test cases for the oauth.py file """
def test_get(self):
""" Make sure the get function returns nothing """
self.assertIsNone(manager.oauth_backend.get())
def test_delete(self):
""" Make sure the delete function returns nothing """
self.assertIsNone(manager.oauth_backend.delete(None))
def test_logout(self):
""" Make sure at least apart of logout is working :/ """
self.assertIsNone(manager.oauth_backend.logout())
def test_add_user(self):
""" Make sure function adds values to database and session """
from flask import session
from pysite.constants import OAUTH_DATABASE
sess_id = "hey bro wazup"
fake_token = {"access_token": "access_token", "id": sess_id, "refresh_token": "refresh_token", "expires_at": 5}
fake_user = {"id": 1235678987654321, "username": "Zwacky", "discriminator": "#6660", "email": "[email protected]"}
manager.db.conn = manager.db.get_connection()
manager.oauth_backend.add_user(fake_token, fake_user, sess_id)
self.assertEqual(sess_id, session["session_id"])
fake_token["snowflake"] = fake_user["id"]
fake_user["user_id"] = fake_user["id"]
del fake_user["id"]
self.assertEqual(fake_token, manager.db.get(OAUTH_DATABASE, sess_id))
self.assertEqual(fake_user, manager.db.get("users", fake_user["user_id"]))
manager.db.delete(OAUTH_DATABASE, sess_id)
manager.db.delete("users", fake_user["user_id"])
manager.db.teardown_request(None)
|
