mirror of
https://github.com/Lynnesbian/FediBooks/
synced 2024-09-28 23:23:04 +00:00
43 lines
1.9 KiB
Python
43 lines
1.9 KiB
Python
|
def test_homepage(client):
|
||
|
|
response = client.get("/")
|
||
|
|
assert response.status_code == 200
|
||
|
|
|
||
|
|
|
||
|
|
def test_homepage_logged_in_no_bots(client, database):
|
||
|
|
with client.session_transaction() as session:
|
||
|
|
session["user_id"] = "123"
|
||
|
|
response = client.get("/")
|
||
|
|
assert response.status_code == 200
|
||
|
|
assert b"Hi there! You have 0 bots." in response.data
|
||
|
|
|
||
|
|
|
||
|
|
def test_home_logged_in_has_bots(client, database):
|
||
|
|
cursor = database.cursor()
|
||
|
|
|
||
|
|
# TODO move these lines into a factory
|
||
|
|
cursor.execute("INSERT INTO `users` (email, password) VALUES (%s, %s)", ("user1@localhost", "pass1"))
|
||
|
|
user1_id = cursor.lastrowid
|
||
|
|
cursor.execute("INSERT INTO `users` (email, password) VALUES (%s, %s)", ("user2@localhost", "pass2"))
|
||
|
|
user2_id = cursor.lastrowid
|
||
|
|
|
||
|
|
cursor.execute("INSERT INTO `credentials` (client_id, client_secret, secret) VALUES (%s, %s, %s)", ("123", "123", "123"))
|
||
|
|
fake_credentials = cursor.lastrowid
|
||
|
|
|
||
|
|
cursor.execute("INSERT INTO `bots` (handle, user_id, enabled, credentials_id, push_public_key, push_private_key, push_secret) VALUES (%s, %s, %s, %s, %s, %s, %s)",
|
||
|
|
("handle1", user1_id, False, fake_credentials, "pubkey", "privkey", "pushsecret"))
|
||
|
|
cursor.execute("INSERT INTO `bots` (handle, user_id, enabled, credentials_id, push_public_key, push_private_key, push_secret) VALUES (%s, %s, %s, %s, %s, %s, %s)",
|
||
|
|
("handle2", user2_id, True, fake_credentials, "pubkey", "privkey", "pushsecret"))
|
||
|
|
|
||
|
|
|
||
|
|
with client.session_transaction() as session:
|
||
|
|
session["user_id"] = user1_id
|
||
|
|
response = client.get("/")
|
||
|
|
assert response.status_code == 200
|
||
|
|
assert b"Hi there! You have 1 bot, 0 of which are currently active." in response.data
|
||
|
|
|
||
|
|
cursor.execute("UPDATE `bots` SET enabled = %s WHERE user_id = %s", (True, user1_id))
|
||
|
|
|
||
|
|
response = client.get("/")
|
||
|
|
assert response.status_code == 200
|
||
|
|
assert b"Hi there! You have 1 bot, 1 of which is currently active." in response.data
|