class TestCliente(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.catalogo = {
"Comum": Plano("comum"),
"FaleMais30": Plano("FaleMais30", 30),
"FaleMais60": Plano("FaleMais60", 60),
"FaleMais120": Plano("FaleMais120", 120),
}
def setUp(self):
self.c_1 = Cliente("Exemplo", "011", self.catalogo["Comum"])
def test_init_cliente_valido(self):
self.assertEqual(self.c_1.get_status()["nome"], "Exemplo")
self.assertEqual(self.c_1.get_status()["DDD"], "011")
self.assertEqual(self.c_1.get_status()["Plano"].get_nome(), "comum")
def test_init_cliente_invalido(self):
try:
Cliente("Exemplo", "012", self.catalogo["Comum"])
created = True
except:
created = False
self.assertFalse(created)
def test_alterar_plano_cliente(self):
self.c_1.alterar_plano(self.catalogo["FaleMais30"])
from cliente import Cliente
from plano import Plano
if __name__ == "__main__":
catalogo = {
"Comum": Plano("comum"),
"FaleMais30": Plano("FaleMais30", 30),
"FaleMais60": Plano("FaleMais60", 60),
"FaleMais120": Plano("FaleMais120", 120),
}
c = Cliente("Zé", "011", catalogo["Comum"])
c.solicitar_simulacao(catalogo["FaleMais30"], "011", "016", 20)
c.solicitar_simulacao(catalogo["FaleMais60"], "011", "017", 80)
c.solicitar_simulacao(catalogo["FaleMais120"], "018", "011", 200)
c.solicitar_simulacao(catalogo["FaleMais30"], "018", "017", 200)
c.alterar_plano(catalogo["FaleMais30"])