test_student.py

from datetime import timedelta
import unittest
from student import Student
from unittest.mock import patch

class TestStudent(unittest.TestCase):
    """
    - Create a new test method called test_apply_extension
    - Inside test_apply_extension, store the current end_date for our student
    instance in a variable called old_end_date
    - Call a method named apply_extension that takes a no.
    of days as an arg on the student instance to update the end_date
    - Assert whether the instance's end_date equals the old end date plus the days arg that was passed in using timedelta
    - Run the tests to confirm that the new method is failing
    - In the Student class, create a new method called apply_extension that has a parameter called days
    - Use the timedelta method from datetime to update the end_date property
    - Run the tests to confirm they are working, with the new method now passing
    """
    @classmethod
    def setUpClass(cls):
        print('setUpClass...')

    @classmethod
    def tearDownClass(cls):
        print('tearDownClass...')

    def setUp(self):
        print('setUp...')
        self.student = Student('John', 'Doe')

    def tearDown(self):
        print('tearDown...')

    def test_full_name(self):
        print('test_full_name')
        self.assertEqual(self.student.full_name, 'John Doe')

    def test_email(self):
        print('test_email')
        self.assertEqual(self.student.email, 'john.doe@email.com')

    def test_alert_santa(self):
        print('test_alert_santa')
        self.student.alert_santa()
        self.assertTrue(self.student.naughty_list)

    def test_apply_extension(self):
        print('test_apply_extension')
        old_end_date = self.student.end_date
        self.student.apply_extension(5)
        self.assertEqual(self.student.end_date, old_end_date + timedelta(days=5))

    def test_course_schedule_success(self):
        with patch("student.requests.get") as mocked_get:
            mocked_get.return_value.ok = True
            mocked_get.return_value.text = "Success!!!"

            schedule = self.student.course_schedule()
            self.assertEqual(schedule, "Success!!!")

    def test_course_schedule_failure(self):
        with patch("student.requests.get") as mocked_get:
            mocked_get.return_value.ok = False

            schedule = self.student.course_schedule()
            self.assertEqual(schedule, "Something went wrong with your request!")


if __name__ == '__main__':
    unittest.main()