From 97855bb4c137ca27dd0abdc33868866ece8edf7b Mon Sep 17 00:00:00 2001
From: Vasyl Nakvasiuk <vaxxxa@gmail.com>
Date: Fri, 19 Apr 2013 14:03:13 +0300
Subject: [PATCH] add test_good_ajax_request for WordCloud

---
 .../lib/xmodule/xmodule/tests/test_logic.py   | 35 +++++++++++++++----
 1 file changed, 29 insertions(+), 6 deletions(-)

diff --git a/common/lib/xmodule/xmodule/tests/test_logic.py b/common/lib/xmodule/xmodule/tests/test_logic.py
index e49fef599a..c977758e6c 100644
--- a/common/lib/xmodule/xmodule/tests/test_logic.py
+++ b/common/lib/xmodule/xmodule/tests/test_logic.py
@@ -8,6 +8,15 @@ from xmodule.conditional_module import ConditionalDescriptor
 from xmodule.word_cloud_module import WordCloudDescriptor
 
 
+class PostData:
+    """Class which emulate postdata."""
+    def __init__(self, dict_data):
+        self.dict_data = dict_data
+
+    def getlist(self, key):
+        return self.dict_data.get(key)
+
+
 class LogicTest(unittest.TestCase):
     """Base class for testing xmodule logic."""
     descriptor_class = None
@@ -75,19 +84,33 @@ class WordCloudModuleTest(LogicTest):
     raw_model_data = {
         'all_words': {'cat': 10, 'dog': 5, 'mom': 1, 'dad': 2},
         'top_words': {'cat': 10, 'dog': 5, 'dad': 2},
-        'submitted': False,
-        'student_words': ['mom', 'dad', 'cat']
+        'submitted': False
     }
 
     def test_bad_ajax_request(self):
 
-        # TODO: move top global test.
+        # TODO: move top global test. Formalize all Xmodule errors.
         response = self.ajax_request('bad_dispatch', {})
         self.assertDictEqual(response, {
             'status': 'fail',
             'error': 'Unknown Command!'
         })
 
-    # TODO
-    # def test_good_ajax_request(self):
-    #     response = self.ajax_request('submit', {})
+    def test_good_ajax_request(self):
+        post_data = PostData({'student_words[]': ['cat', 'cat', 'dog', 'sun']})
+        response = self.ajax_request('submit', post_data)
+        self.assertEqual(response['status'], 'success')
+        self.assertEqual(response['submitted'], True)
+        self.assertEqual(response['total_count'], 22)
+        self.assertDictEqual(
+            response['student_words'],
+            {'sun': 1, 'dog': 6, 'cat': 12}
+        )
+        self.assertListEqual(
+            response['top_words'],
+            [{'text': 'dad', 'size': 2},
+             {'text': 'sun', 'size': 1},
+             {'text': 'dog', 'size': 6},
+             {'text': 'mom', 'size': 1},
+             {'text': 'cat', 'size': 12}]
+        )