Teste se jogo acabou com todos atores Ok

renzon · renzon · commit 7b20ecd1a472 · 2014-07-16T23:04:11.000-03:00
diff --git a/fase.py b/fase.py
@@ -1,3 +1,6 @@
+from atores import ATIVO
+
+
 class Fase():
     def __init__(self):
         self._passaros = []
@@ -13,5 +16,17 @@ def adicionar_porco(self, *porcos):
     def adicionar_passaro(self, *passaros):
         self._passaros.extend(passaros)
 
-    def acabou(self,tempo):
-        return True
+    def acabou(self, tempo):
+        return not self._existe_porco_ativo(tempo) or not self._existe_passaro_ativo(tempo)
+
+    def _existe_porco_ativo(self, tempo):
+        return self._verificar_se_existe_ator_ativo(self._porcos, tempo)
+
+    def _verificar_se_existe_ator_ativo(self, atores, tempo):
+        for a in atores:
+            if a.status(tempo) == ATIVO:
+                return True
+        return False
+
+    def _existe_passaro_ativo(self, tempo):
+        return self._verificar_se_existe_ator_ativo(self._passaros,tempo)
diff --git a/testes/fase_testes.py b/testes/fase_testes.py
@@ -41,5 +41,33 @@ def teste_adicionar_passaro(self):
         self.assertListEqual([passaro, passaro1, passaro2], fase._passaros)
 
     def teste_acabou_sem_porcos(self):
-        fase=Fase()
-        self.assertTrue(fase.acabou(0))
+        fase = Fase()
+        self.assertTrue(fase.acabou(0))
+
+    def teste_acabou_com_porcos_e_passaros(self):
+        fase = Fase()
+        porcos = [Porco(1, 1) for i in range(2)]
+        passaros = [PassaroAmarelo(1, 1) for i in range(2)]
+        fase.adicionar_porco(*porcos)
+        fase.adicionar_passaro(*passaros)
+        self.assertFalse(fase.acabou(0))
+        self.assertFalse(fase.acabou(2.9))
+        self.assertFalse(fase.acabou(3))
+
+        for passaro, porco in zip(passaros, porcos):
+            passaro.colidir(porco, 3)
+
+        self.assertFalse(fase.acabou(0))
+        self.assertFalse(fase.acabou(2.9))
+        self.assertTrue(fase.acabou(3))
+
+        fase.adicionar_obstaculo(Obstaculo())
+        self.assertTrue(fase.acabou(3), 'Obstáculo não interfere no fim do jogo')
+
+        fase.adicionar_porco(Porco())
+        self.assertTrue(fase.acabou(3), 'Com Porco ativo e sem pássaro para lançar, o jogo deveria acabar')
+
+        fase.adicionar_porco(PassaroAmarelo())
+        self.assertTrue(fase.acabou(3), 'Com Porco ativo e com pássaro para lançar, o jogo não deveria acabar')
+
+
