Adicao de Obstaculos Ok

renzon · renzon · commit ab0735b77bc1 · 2014-07-16T22:08:00.000-03:00
diff --git a/fase.py b/fase.py
@@ -0,0 +1,11 @@
+# -*- coding: utf-8 -*-
+
+from __future__ import unicode_literals
+
+
+class Fase():
+    def __init__(self):
+        self._obstaculos = []
+
+    def adicionar_obstaculo(self, *obstaculos):
+        self._obstaculos.extend(obstaculos)
diff --git a/testes/fase_testes.py b/testes/fase_testes.py
@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+
+from __future__ import unicode_literals
+from unittest.case import TestCase
+from atores import Obstaculo
+from fase import Fase
+
+
+class FaseTestes(TestCase):
+    def teste_adicionar_obstaculo(self):
+        fase = Fase()
+        self.assertListEqual([], fase._obstaculos)
+        obstaculo = Obstaculo()
+        fase.adicionar_obstaculo(obstaculo)
+        self.assertListEqual([obstaculo], fase._obstaculos)
+
+        obstaculo1, obstaculo2 = Obstaculo(), Obstaculo()
+        fase.adicionar_obstaculo(obstaculo1, obstaculo2)
+        self.assertListEqual([obstaculo, obstaculo1, obstaculo2], fase._obstaculos)
+
