tests.py

from config import TestConfig, db_generator
from api import (
    Announcements,
    CourseAliases,
    Courses,
    OrgUnits,
    Guardians,
    GuardianInvites,
    Invitations,
    Topics,
    Students,
    Teachers,
)
import os
import pandas as pd
from test_responses import (
    FakeService,
    ALIAS_SOLUTION,
    ORG_UNIT_SOLUTION,
    GUARDIAN_SOLUTION,
    GUARDIAN_INVITE_SOLUTION,
    COURSE_SOLUTION,
    TOPIC_SOLUTION,
    STUDENT_SOLUTION,
    TEACHER_SOLUTION,
    INVITATION_SOLUTION,
    ANNOUNCEMENT_SOLUTION,
)

# TODO: Add tests for Coursework and Submissions.


class TestEndToEnd:
    def setup(self):
        self.config = TestConfig
        self.sql = db_generator(self.config)
        self.service = FakeService()

    def teardown(self):
        if os.path.exists(self.config.DB):
            os.remove(self.config.DB)

    def test_get_org_units(self):
        self.generic_get_test(
            OrgUnits(self.service, self.sql, self.config), ORG_UNIT_SOLUTION
        )

    def test_get_guardians(self):
        self.generic_get_test(
            Guardians(self.service, self.sql, self.config), GUARDIAN_SOLUTION
        )

    def test_get_guardian_invites(self):
        self.generic_get_test(
            GuardianInvites(self.service, self.sql, self.config),
            GUARDIAN_INVITE_SOLUTION,
        )

    def test_get_courses(self):
        self.generic_get_test(
            Courses(self.service, self.sql, self.config), COURSE_SOLUTION
        )

    def test_get_topics(self):
        self.generic_get_test(
            Topics(self.service, self.sql, self.config),
            TOPIC_SOLUTION,
            course_ids=[0, 1],
        )

    def test_get_students(self):
        self.generic_get_test(
            Students(self.service, self.sql, self.config),
            STUDENT_SOLUTION,
            course_ids=[0, 1],
        )

    def test_get_teachers(self):
        self.generic_get_test(
            Teachers(self.service, self.sql, self.config),
            TEACHER_SOLUTION,
            course_ids=[0, 1],
        )

    def test_get_aliases(self):
        self.generic_get_test(
            CourseAliases(self.service, self.sql, self.config),
            ALIAS_SOLUTION,
            course_ids=[0, 1],
        )

    def test_get_invitations(self):
        self.generic_get_test(
            Invitations(self.service, self.sql, self.config),
            INVITATION_SOLUTION,
            course_ids=[0, 1],
        )

    def test_get_announcements(self):
        self.generic_get_test(
            Announcements(self.service, self.sql, self.config),
            ANNOUNCEMENT_SOLUTION,
            course_ids=[0, 1],
        )

    def generic_get_test(self, endpoint, solution, course_ids=[None]):
        endpoint.batch_pull_data(course_ids=course_ids)
        result = pd.read_sql_table(
            endpoint.table_name, con=self.sql.engine, schema=self.sql.schema
        )
        assert result.equals(solution)