1717sys .path .append (project_dir )
1818
1919from atores import Obstaculo , Porco , PassaroVermelho , PassaroAmarelo , DESTRUIDO , ATIVO
20- from fase import Fase , Ponto
20+ from fase import Fase , Ponto , EM_ANDAMENTO , VITORIA , DERROTA
2121
2222
2323class FaseTestes (TestCase ):
@@ -57,7 +57,7 @@ def teste_adicionar_passaro(self):
5757
5858 def teste_acabou_sem_porcos (self ):
5959 fase = Fase ()
60- self .assertTrue ( fase .acabou ())
60+ self .assertEqual ( VITORIA , fase .status ())
6161
6262 def teste_acabou_com_porcos_e_passaros (self ):
6363 fase = Fase ()
@@ -66,52 +66,52 @@ def teste_acabou_com_porcos_e_passaros(self):
6666 fase .adicionar_porco (* porcos )
6767 fase .adicionar_passaro (* passaros )
6868
69- self .assertFalse ( fase .acabou ())
69+ self .assertEqual ( EM_ANDAMENTO , fase .status ())
7070
7171 # colidindo cada passaro com um porco no tempo 3
7272 for passaro , porco in zip (passaros , porcos ):
7373 passaro .colidir (porco , 3 )
7474
75- self .assertTrue ( fase .acabou ())
75+ self .assertEqual ( VITORIA , fase .status ())
7676
7777 fase .adicionar_obstaculo (Obstaculo ())
78- self .assertTrue ( fase .acabou (), 'Obstáculo não interfere no fim do jogo' )
78+ self .assertEqual ( VITORIA , fase .status (), 'Obstáculo não interfere no fim do jogo' )
7979
8080 fase .adicionar_porco (Porco ())
81- self .assertTrue ( fase .acabou (), 'Com Porco ativo e sem pássaro para lançar, o jogo deveria acabar' )
81+ self .assertEqual ( DERROTA , fase .status (), 'Com Porco ativo e sem pássaro para lançar, o jogo deveria acabar' )
8282
8383 fase .adicionar_passaro (PassaroAmarelo ())
84- self .assertFalse ( fase .acabou (), 'Com Porco ativo e com pássaro para lançar, o jogo não deveria acabar' )
84+ self .assertEqual ( EM_ANDAMENTO , fase .status (), 'Com Porco ativo e com pássaro para lançar, o jogo não deveria acabar' )
8585
8686 def teste_status (self ):
8787 fase = Fase ()
8888 porcos = [Porco (1 , 1 ) for i in range (2 )]
8989 passaros = [PassaroAmarelo (1 , 1 ) for i in range (2 )]
9090 fase .adicionar_porco (* porcos )
9191 fase .adicionar_passaro (* passaros )
92- self .assertEqual ('Jogo em andamento.' , fase .status ())
92+ self .assertEqual (EM_ANDAMENTO , fase .status ())
9393
9494 for passaro , porco in zip (passaros , porcos ):
9595 passaro .colidir (porco , 3 )
9696
97- self .assertEqual ('Jogo em encerrado. Você ganhou!' , fase .status (),
97+ self .assertEqual (VITORIA , fase .status (),
9898 'Sem porcos ativos o jogo deveria terminar com vitória' )
9999
100100 fase .adicionar_obstaculo (Obstaculo ())
101- self .assertEqual ('Jogo em encerrado. Você ganhou!' , fase .status (),
101+ self .assertEqual (VITORIA , fase .status (),
102102 'Obstáculo não interfere para definir vitória' )
103103
104104 porco = Porco ()
105105 fase .adicionar_porco (porco )
106- self .assertEqual ('Jogo em encerrado. Você perdeu!' , fase .status (),
106+ self .assertEqual (DERROTA , fase .status (),
107107 'Com Porco ativo e sem pássaro para lançar, o jogo deveria acabar em derrota' )
108108
109109 fase .adicionar_passaro (PassaroAmarelo ())
110- self .assertEqual ('Jogo em andamento.' , fase .status (),
110+ self .assertEqual (EM_ANDAMENTO , fase .status (),
111111 'Com Porco ativo e com pássaro para lançar, o jogo não deveria acabar' )
112112
113113 porco .colidir (porco , 3 )
114- self .assertEqual ('Jogo em encerrado. Você ganhou!' , fase .status (),
114+ self .assertEqual (VITORIA , fase .status (),
115115 'Sem porco ativo, o jogo deveria acabar com vitória' )
116116
117117 def teste_lancar_passaro_sem_erro_quando_nao_existe_passaro (self ):
@@ -202,7 +202,7 @@ def teste_calcular_pontos(self):
202202
203203 self .assertListEqual (expected , fase_exemplo .calcular_pontos (8.5 ))
204204
205- self .assertTrue ( fase_exemplo .acabou ())
205+ self .assertEqual ( VITORIA , fase_exemplo .status ())
206206
207207
208208def criar_fase_exemplo (multiplicador = 1 ):
0 commit comments