move get_numero_escuela tests to their own class

1 files changed, 25 insertions, 15 deletions

diff --git a/webapp/polls/tests/poll_result_file_tests.py b/webapp/polls/tests/poll_result_file_tests.py
index 2e11d8f..628a5b1 100644
--- a/webapp/polls/tests/poll_result_file_tests.py
+++ b/webapp/polls/tests/poll_result_file_tests.py
@@ -20,21 +20,6 @@ class PollResultFileTest(MongoTestCase):
         data['result']['0']['polled'] = {}
         self.data = data
 
-    def test_it_should_respond_to_numero_escuela(self):
-        data = self.data
-        numero_escuela = '1'
-        data['result']['0']['polled']['NUM_ESC'] = numero_escuela
-        json_str = json_construc(data)
-
-        file_ = tempfile.NamedTemporaryFile(suffix='.poll_result',
-                                            delete=False)
-        file_.write(json_str)
-        file_.close()
-
-        file_path = file_.name
-        result = PollResultFile(file_path)
-        self.assertEqual(numero_escuela, result.get_numero_escuela())
-
     def test_it_should_respond_to_pollster_username(self):
         data = self.data
         pollster_username = "encuestador1"
@@ -154,3 +139,28 @@ class PollResultFileTest(MongoTestCase):
         expected_url = os.path.join(
             settings.RESULT_BCK_URL, poll_id, result_file.name)
         self.assertEqual(expected_url, result_file.get_absolute_url())
+
+
+class NumeroEscuelaTest(MongoTestCase):
+
+    def setUp(self):
+        data = {}
+        data['result'] = {}
+        data['result']['0'] = {}
+        data['result']['0']['polled'] = {}
+        self.data = data
+
+    def test_it_should_respond_to_numero_escuela(self):
+        data = self.data
+        numero_escuela = '1'
+        data['result']['0']['polled']['NUM_ESC'] = numero_escuela
+        json_str = json_construc(data)
+
+        file_ = tempfile.NamedTemporaryFile(suffix='.poll_result',
+                                            delete=False)
+        file_.write(json_str)
+        file_.close()
+
+        file_path = file_.name
+        result = PollResultFile(file_path)
+        self.assertEqual(numero_escuela, result.get_numero_escuela())