diff options
Diffstat (limited to 'tests/test_oauth_backend.py')
-rw-r--r-- | tests/test_oauth_backend.py | 39 |
1 files changed, 0 insertions, 39 deletions
diff --git a/tests/test_oauth_backend.py b/tests/test_oauth_backend.py deleted file mode 100644 index 473f5c54..00000000 --- a/tests/test_oauth_backend.py +++ /dev/null @@ -1,39 +0,0 @@ -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"} - 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) |