class TestMain(unittest.TestCase):
def setUp(self):
self.emp_1 = Employee("José", "111.111.111-11", "2000.00")
self.emp_2 = Employee("Mario", "444.444.444-44", "1800.00")
self.emp_3 = Employee("César", "555.555.555-55", "2200.00")
self.man_1 = Manager("Maria", "222.222.222-22", "5000.00", "3542", "3")
self.man_2 = Manager("Paula", "666.666.666-66", "6000.00", "8546", "4")
self.man_3 = Manager("Carolina", "777.777.777-77", "7000.00", "9852",
"6")
self.cli_1 = Client("Luiza", "333.333.333-33", "3854")
def tearDown(self):
pass
def test_name(self):
self.assertEqual(self.emp_1._name, "José")
self.assertEqual(self.man_1._name, "Maria")
self.assertEqual(self.cli_1._name, "Luiza")
def test_get_bonus(self):
self.assertEqual(self.emp_1.get_bonus(), 200)
self.assertEqual(self.emp_2.get_bonus(), 180)
self.assertEqual(self.emp_3.get_bonus(), 220)
self.assertEqual(self.man_1.get_bonus(), 1500)
self.assertEqual(self.man_2.get_bonus(), 1600)
self.assertEqual(self.man_3.get_bonus(), 1700)
def generate_users(user_range):
for i in range(1, user_range):
emp = Employee()
emp.username = fake.name()
emp.password = "******"
print(f" The value of i is {i}")
#emp.id = i
try:
print(f" The employee being added is {emp}")
db.session.add(emp)
db.session.commit()
except Exception as e:
print(f" Error while commiting to db {e} ")
class EmployeeUnitTest(unittest.TestCase):
"""Used to test the employee class in the module main.py"""
#Unit Test Constructor
def setUp(self):
"""Used to try reduce redundancy in each test method."""
self.testEmployee = Employee("Ryan", "M", 20000)
def test_default_raise(self):
"""Used to test that the classes default raise value is correct."""
self.testEmployee.award_raise()
self.assertEquals(self.testEmployee.salary, 25000)
def test_alt_raise(self):
"""Used to test that method raise executes properly when not using default raise value."""
self.testEmployee.award_raise(7500)
self.assertEquals(self.testEmployee.salary, 27500)
def setUp(self):
self.emp_1 = Employee("José", "111.111.111-11", "2000.00")
self.emp_2 = Employee("Mario", "444.444.444-44", "1800.00")
self.emp_3 = Employee("César", "555.555.555-55", "2200.00")
self.man_1 = Manager("Maria", "222.222.222-22", "5000.00", "3542", "3")
self.man_2 = Manager("Paula", "666.666.666-66", "6000.00", "8546", "4")
self.man_3 = Manager("Carolina", "777.777.777-77", "7000.00", "9852",
"6")
self.cli_1 = Client("Luiza", "333.333.333-33", "3854")
def test_employer_class(self):
with pytest.raises(TypeError):
Employee(123, 123, 123)
def setUp(self):
"""Used to try reduce redundancy in each test method."""
self.testEmployee = Employee("Ryan", "M", 20000)
from main import Employee, Manager, Director
if __name__ == "__main__":
director = Director("joao", "111.111.111-11", "4000.0")
print(director.get_bonus())
# Esse teste vai dar erro! É uma classe abstrata.. Não conseguimos instanciar a classe Employee MAS podemos
# instanciar as classes filhas que são objetos que realmente existem no sistema.
# Abstract class (é como se ela não existisse no nosso programa) - Tipo conta / conta corrente e conta poupança.
employee = Employee("maria", "222.222.222-22", "5000.0")
print(employee.get_bonus())
from main import Employee, Manager, Bonus_Control, Client
if __name__ == '__main__':
employee = Employee("João", "111.111.111-11", "2000.00")
print(f"Employee bonus: {employee.get_bonus()}")
manager1 = Manager("Helena", "222.222.222-22", "5000.00", "1234", "0")
print(f"Manager bonus: {manager1.get_bonus()}")
manager2 = Manager("Isabel", "333.333.333-33", "5000.00", "1245", "0")
print(f"Manager bonus: {manager2.get_bonus()}")
client1 = Client("Marisa", "444.444.444-44", "8721")
control = Bonus_Control()
control.register(employee)
control.register(manager1)
control.register(manager2)
control.register(client1)
print(f"Total: {control.bonus_total}")