def test_admitir_funcionario(self): empresa1 = Empresa("Ricks Rocks","12345678900") funcionario1 = Funcionario("Astrobaldo","02966644729", 3000) empresa1.admitir_funcionario(funcionario1) empresa1.quadro_funcionarios[0].nome |should| equal_to("Astrobaldo") funcionario2 = Funcionario("Centreflexiano","086", 2000) empresa1.admitir_funcionario(funcionario2) empresa1.quadro_funcionarios[1].nome |should| equal_to("Centreflexiano") empresa1.quadro_funcionarios |should| have(2).itens
def test_buscar_funcionario(self): empresa1 = Empresa("Ricks Rocks","12345678900") funcionario1 = Funcionario("Astrobaldo","02966644729", 3000) empresa1.admitir_funcionario(funcionario1) funcionario2 = Funcionario("Centreflexiano","086", 2000) empresa1.admitir_funcionario(funcionario2) empresa1.buscar_funcionario("Centreflexiano") |should| equal_to(1) empresa1.buscar_funcionario("Benzocriol") |should| equal_to(None)
def test_demitir_funcionario(self): empresa1 = Empresa("Ricks Rocks","12345678900") funcionario1 = Funcionario("Astrobaldo","02966644729", 3000) empresa1.admitir_funcionario(funcionario1) empresa1.demitir_funcionario("Astrobaldo", 2011, 10, 25) empresa1.quadro_funcionarios[0].data_demissao |should| equal_to(date(2011, 10, 25)) funcionario2 = Funcionario("Centreflexiano","086", 2000) empresa1.admitir_funcionario(funcionario2) empresa1.demitir_funcionario("Centreflexiano", 2010, 3, 25) empresa1.quadro_funcionarios[1].data_demissao |should| equal_to(None)