"""
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