def last_email():
conn = sqlite_connect()
cur = conn.cursor()
cur.execute("SELECT id FROM outbox")
row = cur.fetchone()
assert row is not None, "Failed to get last email from SQLite."
return PendingSend(row[0])
def delete_db():
conn = sqlite_connect()
cur = conn.cursor()
cur.execute("DELETE FROM registrations")
cur.execute("DELETE FROM email_changes")
cur.execute("DELETE FROM outbox")
conn.commit()
def delete_db():
conn = sqlite_connect()
cur = conn.cursor()
cur.execute("DELETE FROM registrations")
cur.execute("DELETE FROM email_changes")
cur.execute("DELETE FROM outbox")
conn.commit()
def last_email():
conn = sqlite_connect()
cur = conn.cursor()
cur.execute("SELECT id FROM outbox")
row = cur.fetchone()
assert row is not None, "Failed to get last email from SQLite."
return PendingSend(row[0])
def last_n_emails(num):
conn = sqlite_connect()
cur = conn.cursor()
cur.execute("SELECT id FROM outbox LIMIT ?", (num,))
rows = cur.fetchall()
assert len(rows) == num, "Failed to get last %d emails from SQLite." % (num,)
mails = []
for row in rows:
mails.append(PendingSend(row[0]))
return mails
def last_n_emails(num):
conn = sqlite_connect()
cur = conn.cursor()
cur.execute("SELECT id FROM outbox LIMIT ?", (num, ))
rows = cur.fetchall()
assert len(
rows) == num, "Failed to get last %d emails from SQLite." % (num, )
mails = []
for row in rows:
mails.append(PendingSend(row[0]))
return mails
def test_verify_outdated_request():
conn = sqlite_connect()
cur = conn.cursor()
statement = "INSERT INTO email_changes (username, new_email, request_time, verify_code) VALUES (?,?,?, ?)"
old = datetime.datetime.now() - datetime.timedelta(days = 4)
arguments = ('abc', '*****@*****.**', old.strftime('%Y-%m-%d %H:%M:%S'), 'bees')
cur.execute(statement, arguments)
conn.commit()
r,data = test_helpers.server_get("/verify/abc/bees")
status = r.status
assert status == 410, data
def test_verify_outdated_request(self):
with sqlite_connect() as conn:
cur = conn.cursor()
statement = "INSERT INTO password_resets (username, requestor_username, request_time, verify_code) VALUES (?,?,?, ?)"
days = config.getint('nemesis', 'password_reset_days')
old = datetime.datetime.now() - datetime.timedelta(days = days + 2)
arguments = ('abc', 'blueshirt', old.strftime('%Y-%m-%d %H:%M:%S'), 'bees')
cur.execute(statement, arguments)
conn.commit()
r, data = test_helpers.server_get("/reset_password/abc/bees")
self.assertEqual(410, r.status, data)
def test_verify_outdated_request(self):
with sqlite_connect() as conn:
cur = conn.cursor()
statement = "INSERT INTO password_resets (username, requestor_username, request_time, verify_code) VALUES (?,?,?, ?)"
days = config.getint('nemesis', 'password_reset_days')
old = datetime.datetime.now() - datetime.timedelta(days=days + 2)
arguments = ('abc', 'blueshirt', old.strftime('%Y-%m-%d %H:%M:%S'),
'bees')
cur.execute(statement, arguments)
conn.commit()
r, data = test_helpers.server_get("/reset_password/abc/bees")
self.assertEqual(410, r.status, data)
def test_verify_outdated_request():
conn = sqlite_connect()
cur = conn.cursor()
statement = "INSERT INTO email_changes (username, new_email, request_time, verify_code) VALUES (?,?,?, ?)"
days = config.getint('nemesis', 'email_change_days')
old = datetime.datetime.now() - datetime.timedelta(days=days + 2)
arguments = ('abc', '*****@*****.**', old.strftime('%Y-%m-%d %H:%M:%S'),
'bees')
cur.execute(statement, arguments)
conn.commit()
r, data = test_helpers.server_get("/verify/abc/bees")
status = r.status
assert status == 410, data
def assert_no_emails():
conn = sqlite_connect()
cur = conn.cursor()
cur.execute("SELECT id FROM outbox")
row = cur.fetchone()
assert row is None, "Should not be any emails in SQLite."
def get_registrations():
conn = sqlite_connect()
cur = conn.cursor()
cur.execute("SELECT * FROM registrations")
return list(cur)
def _exec(self, statement, arguments):
conn = sqlite_connect()
cur = conn.cursor()
cur.execute(statement, arguments)
conn.commit()
def _exec(self, statement, arguments):
conn = sqlite_connect()
cur = conn.cursor()
cur.execute(statement, arguments)
conn.commit()
def assert_no_emails():
conn = sqlite_connect()
cur = conn.cursor()
cur.execute("SELECT id FROM outbox")
row = cur.fetchone()
assert row is None, "Should not be any emails in SQLite."
def get_registrations():
conn = sqlite_connect()
cur = conn.cursor()
cur.execute("SELECT * FROM registrations")
return list(cur)
