Files
raid-callouts/src/py/db_helper_tests.py
2025-10-17 11:05:30 -06:00

70 lines
2.4 KiB
Python

"""
Database functionality tests.
@author: Gabriella "Contrastellar" Agathon
"""
import datetime
import helper.db_helper
class TestClass():
"""
Test class for the database functionality.
These tests should only be assertions of type, not of return values.
"""
DATABASE_CONN = helper.db_helper.DBHelper("xiv-database.ini")
def test_add_registration(self) -> None:
registration = self.DATABASE_CONN.register_char_name(uid=1, char_name="test")
assert registration is None
def test_query_callouts(self) -> None:
callouts = self.DATABASE_CONN.query_callouts(7)
assert type(callouts) is list
def test_add_callout(self) -> None:
callout = self.DATABASE_CONN.add_callout(user_id=1, callout=datetime.date.today(), reason='test', nickname='test', char_name='test', potential_fill='test')
assert callout is None
def test_self_query_callouts(self) -> None:
self_callouts = self.DATABASE_CONN.query_self_callouts(user_id=1, days=365)
assert self_callouts is not None
def test_callouts(self) -> None:
callout = self.DATABASE_CONN.query_callouts(days=7)
assert callout is not None
def test_remove_callout(self) -> None:
remove_callout = self.DATABASE_CONN.remove_callout(user_id=1, callout=datetime.date.today())
assert remove_callout is None
def test_char_name(self) -> None:
char_name = self.DATABASE_CONN.return_char_name(uid=1)
assert type(char_name) is str
assert char_name == "test"
def test_remove_registration(self) -> None:
registration = self.DATABASE_CONN.remove_registration(uid=1, isOkay=True)
assert registration is None
def test_format_list_of_callouts(self) -> None:
callouts = self.DATABASE_CONN.query_callouts(days=7)
formatted_callouts = self.DATABASE_CONN.format_list_of_callouts(callouts=callouts)
assert formatted_callouts.__class__ is str
def test_add_break(self) -> None:
today: datetime.date = datetime.date.today()
new_break = self.DATABASE_CONN.add_break(user_id=1, break_start=today, break_end=today)
assert new_break is None
def test_remove_break(self) -> None:
today: datetime.date = datetime.date.today()
remove_break = self.DATABASE_CONN.remove_break(user_id=1, start_date=today)
assert remove_break is None