refactorin para passaro informar se foi lançado ou nao

renzon · renzon · commit a678883dc6f1 · 2014-07-16T23:47:39.000-03:00
diff --git a/atores.py b/atores.py
@@ -71,6 +71,9 @@ def __init__(self, x=0, y=0):
         self._tempo_de_lancamento = None
         self._angulo_de_lancamento = None  # radianos
 
+    def foi_lancado(self):
+        return self._tempo_de_lancamento is not None
+
     def colidir_com_chao(self, tempo):
         if self.y <= 0:
             self._tempo_de_colisao = tempo
@@ -107,6 +110,7 @@ class PassaroAmarelo(Passaro):
     _velocidade_scalar = 30  # m/s
     _caracter_ativo = '>'
 
+
 class PassaroVermelho(Passaro):
     _velocidade_scalar = 20  # m/s
     _caracter_ativo = '»'
diff --git a/fase.py b/fase.py
@@ -36,6 +36,12 @@ def status(self, tempo):
             return 'Jogo em encerrado. Você perdeu!'
         return 'Jogo em encerrado. Você ganhou!'
 
+    def lancar(self, angulo, tempo):
+        for passaro in self._passaros:
+            if ATIVO == passaro.status(tempo):
+                passaro.lancar(angulo, tempo)
+                return
+
     def _existe_porco_ativo(self, tempo):
         return self._verificar_se_existe_ator_ativo(self._porcos, tempo)
 
diff --git a/testes/atores_testes.py b/testes/atores_testes.py
@@ -113,13 +113,19 @@ def assert_passaro_posicao(self, x_esperado, y_esperado, status_esperado, passar
 
 
 class PassaroVermelhoTests(PassaroBaseTests):
+    def teste_foi_lancado(self):
+        passaro_vermelho = PassaroVermelho(1, 1)
+        self.assertFalse(passaro_vermelho.foi_lancado())
+        passaro_vermelho.lancar(0, 0)
+        self.assertTrue(passaro_vermelho.foi_lancado())
+
     def teste_posicao_antes_do_lancamento(self):
-        passaro_amarelo = PassaroVermelho(1, 1)
-        passaro_amarelo.lancar(90, 2)  # passaro lancado a 90 graus no tempo 2 segundos
+        passaro_vermelho = PassaroVermelho(1, 1)
+        passaro_vermelho.lancar(90, 2)  # passaro lancado a 90 graus no tempo 2 segundos
         #
         for t in range(20):
             t /= 10
-            self.assertEqual((1, 1), passaro_amarelo.calcular_posicao(t),
+            self.assertEqual((1, 1), passaro_vermelho.calcular_posicao(t),
                              'Não deveria se mover no tempo %s < 2 segundtos' % t)
 
     def teste_status(self):
diff --git a/testes/fase_testes.py b/testes/fase_testes.py
@@ -2,6 +2,7 @@
 
 from __future__ import unicode_literals
 from unittest.case import TestCase
+import math
 from atores import Obstaculo, Porco, PassaroVermelho, PassaroAmarelo
 from fase import Fase
 
@@ -106,16 +107,18 @@ def teste_lancar_passaro_sem_erro_quando_nao_existe_passaro(self):
 
     def teste_lancar_passaro_sem_erro_quando_nao_existe_passaro(self):
         passaro_vermelho, passaro_amarelo = PassaroVermelho(1, 1), PassaroAmarelo(1, 1)
-        fase = Fase(passaro_vermelho, passaro_amarelo)
+        fase = Fase()
+        fase.adicionar_passaro(passaro_vermelho, passaro_amarelo)
+
         fase.lancar(90, 1)
-        fase.lancar(45, 3)
-        fase.lancar(31, 5)  # testando que lançar passaros depos de todos lançados não causa erro
-        self.assertEqual(90, passaro_vermelho._angulo_de_lancamento)
+        self.assertEqual(math.radians(90), passaro_vermelho._angulo_de_lancamento)
         self.assertEqual(1, passaro_vermelho._tempo_de_lancamento)
 
-        self.assertEqual(45, passaro_amarelo._angulo_de_lancamento)
+        fase.lancar(45, 3)
+        self.assertEqual(math.radians(45), passaro_amarelo._angulo_de_lancamento)
         self.assertEqual(3, passaro_amarelo._tempo_de_lancamento)
 
+        fase.lancar(31, 5)  # testando que lançar passaros depos de todos lançados não causa erro
 
         # def teste_calcular_pontos(self):
         # fase=Fase()
