test_user_profile.py

import unittest
from user_profile import Profile


class TestProfile(unittest.TestCase):

    def setUp(self):
        self.__user = Profile()

    # we test if attributes exist
    def test_has_get_email(self):
        # use built in hasattr() to check if attribute
        # get_managers exist
        test_result = hasattr(self.__user, "get_email")
        self.assertIs(test_result, True)

    # we test if attributes exist
    def test_has_set_email(self):
        # use built in hasattr() to check if attribute
        # get_managers exist
        test_result = hasattr(self.__user, "set_email")
        self.assertIs(test_result, True)

    # we test if attributes exist
    def test_has_get_role(self):
        # use built in hasattr() to check if attribute
        # get_managers exist
        test_result = hasattr(self.__user, "get_role")
        self.assertIs(test_result, True)

    # we test if attributes exist
    def test_has_get_role(self):
        # use built in hasattr() to check if attribute
        # get_managers exist
        test_result = hasattr(self.__user, "set_role")
        self.assertIs(test_result, True)