test_student.py

import unittest
from employee import Student
from unittest.mock import patch


class TestStudent(unittest.TestCase):

    def setUp(self):
        print("Setup has been run")
        self.stud_1 = Student('kelvin', 'onkundi', 300)
        self.stud_2 = Student('novak', 'onkundi', 300)
        self.parent_name = "Barrack Obama"

    def tearDown(self):
        print("teardown has been run")
    

    def test_email(self):

        self.assertEqual(self.stud_1.email, '[email protected]')
        self.assertEqual(self.stud_2.email, "[email protected]")

    def test_fullname(self):
        self.assertEqual(self.stud_1.fullname, "kelvin onkundi")

    def test_send_parent_message(self):
        """ test if the right message was sent to the parent """
        self.assertIn(
            f"Dear {self.parent_name}, Your son {self.stud_1.first} {self.stud_1.last} managed to score {self.stud_1.marks}", self.stud_1.send_parent_message(self.parent_name))