Builder design pattern raw test suite

0--key · 0--key · commit 66aeceaa3a45 · 2017-01-25T20:44:31.000+02:00
diff --git a/tests/test_builder.py b/tests/test_builder.py
@@ -0,0 +1,34 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+import unittest
+from creational.builder import Director, BuilderHouse,  BuilderFlat
+
+
+class TestHouseBuilding(unittest.TestCase):
+
+    def setUp(self):
+        self.director = Director()
+        self.director.builder = BuilderHouse()
+        self.director.construct_building()
+        self.building = self.director.get_building()
+
+    def test_house_size(self):
+        self.assertEqual(self.building.size, 'Big')
+
+    def test_num_floor_in_house(self):
+        self.assertEqual(self.building.floor, 'One')
+
+
+class TestFlatBuilding(unittest.TestCase):
+
+    def setUp(self):
+        self.director = Director()
+        self.director.builder = BuilderFlat()
+        self.director.construct_building()
+        self.building = self.director.get_building()
+
+    def test_house_size(self):
+        self.assertEqual(self.building.size, 'Small')
+
+    def test_num_floor_in_house(self):
+        self.assertEqual(self.building.floor, 'More than One')
