class TestStudent(unittest.TestCase):
@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_alert_santa(self):
print("test_alert_santa")
self.student.alert_santa()
self.assertTrue(self.student.naughty_list)
def test_email(self):
print("test_email")
self.assertEqual(self.student.email, "*****@*****.**")
def test_apply_extension(self):
initial = self.student.end_date
self.student.apply_extension(10)
self.assertEqual(self.student.end_date, initial + timedelta(10))
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_fail(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 the request!")
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 will take a number of days as an argument on the student instance to update the end_date
* Assert whether the instance’s end_date equals the old end date plus the days argument 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.
"""
@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, '*****@*****.**')
def test_alert_santa(self):
print('test_alert_santa')
self.student.alert_santa()
self.assertTrue(self.student.naughty_list)
def test_apply_extension(self):
old_end_date = self.student.end_date
self.student.apply_extension(5)
self.assertEqual(self.student.end_date,
old_end_date + timedelta(days=5))
