Teste adição de Porco

renzon · renzon · commit bfb8ae8529d1 · 2014-07-16T22:09:27.000-03:00
diff --git a/testes/fase_testes.py b/testes/fase_testes.py
@@ -2,7 +2,7 @@
 
 from __future__ import unicode_literals
 from unittest.case import TestCase
-from atores import Obstaculo
+from atores import Obstaculo, Porco
 from fase import Fase
 
 
@@ -17,4 +17,15 @@ def teste_adicionar_obstaculo(self):
         obstaculo1, obstaculo2 = Obstaculo(), Obstaculo()
         fase.adicionar_obstaculo(obstaculo1, obstaculo2)
         self.assertListEqual([obstaculo, obstaculo1, obstaculo2], fase._obstaculos)
+        
+    def teste_adicionar_porco(self):
+        fase = Fase()
+        self.assertListEqual([], fase._porcos)
+        porco = Porco()
+        fase.adicionar_porco(porco)
+        self.assertListEqual([porco], fase._porcos)
+
+        porco1, porco2 = Porco(), Porco()
+        fase.adicionar_porco(porco1, porco2)
+        self.assertListEqual([porco, porco1, porco2], fase._porcos)
 
