From 61ba91025596d50752dbf09e79dbdf581435380b Mon Sep 17 00:00:00 2001
From: Calen Pennington <cale@edx.org>
Date: Tue, 16 Dec 2014 16:38:01 -0500
Subject: [PATCH] Add a test of pure xblock and xmodule event publishing

---
 .../courseware/tests/test_module_render.py    | 50 ++++++++++++++++++-
 1 file changed, 49 insertions(+), 1 deletion(-)

diff --git a/lms/djangoapps/courseware/tests/test_module_render.py b/lms/djangoapps/courseware/tests/test_module_render.py
index 12507174a7..cf4b8a6235 100644
--- a/lms/djangoapps/courseware/tests/test_module_render.py
+++ b/lms/djangoapps/courseware/tests/test_module_render.py
@@ -39,7 +39,7 @@ from xmodule.modulestore import ModuleStoreEnum
 from xmodule.modulestore.django import modulestore
 from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
 from xmodule.modulestore.tests.factories import ItemFactory, CourseFactory, check_mongo_calls
-from xmodule.x_module import XModuleDescriptor, STUDENT_VIEW
+from xmodule.x_module import XModuleDescriptor, XModule, STUDENT_VIEW
 
 TEST_DATA_DIR = settings.COMMON_TEST_DATA_ROOT
 
@@ -51,6 +51,14 @@ class PureXBlock(XBlock):
     pass
 
 
+class EmptyXModule(XModule):
+    pass
+
+
+class EmptyXModuleDescriptor(XModuleDescriptor):
+    module_class = EmptyXModule
+
+
 @ddt.ddt
 @override_settings(MODULESTORE=TEST_DATA_MOCK_MODULESTORE)
 class ModuleRenderTestCase(ModuleStoreTestCase, LoginEnrollmentTestCase):
@@ -1105,3 +1113,43 @@ class TestRebindModule(TestSubmittingProblems):
         module = self.get_module_for_user(self.anon_user)
         module.system.rebind_noauth_module_to_user(module, self.anon_user)
         self.assertFalse(psycho_handler.called)
+
+
+@ddt.ddt
+@override_settings(MODULESTORE=TEST_DATA_MOCK_MODULESTORE)
+class TestEventPublishing(ModuleStoreTestCase, LoginEnrollmentTestCase):
+
+    def setUp(self):
+        """
+        Set up the course and user context
+        """
+        super(TestEventPublishing, self).setUp()
+
+        self.mock_user = UserFactory()
+        self.mock_user.id = 1
+        self.request_factory = RequestFactory()
+
+    @ddt.data('xblock', 'xmodule')
+    @XBlock.register_temp_plugin(PureXBlock, identifier='xblock')
+    @XBlock.register_temp_plugin(EmptyXModuleDescriptor, identifier='xmodule')
+    @patch.object(render, 'make_track_function')
+    def test_event_publishing(self, block_type, mock_track_function):
+        request = self.request_factory.get('')
+        request.user = self.mock_user
+        course = CourseFactory()
+        descriptor = ItemFactory(category=block_type, parent=course)
+        field_data_cache = FieldDataCache([course, descriptor], course.id, self.mock_user)
+        block = render.get_module(self.mock_user, request, descriptor.location, field_data_cache)
+
+        event_type = 'event_type'
+        event = {'event': 'data'}
+
+        block.runtime.publish(block, event_type, event)
+
+        mock_track_function.assert_called_once_with(request)
+
+        if block_type == 'xblock':
+            self.assertFalse(mock_track_function.return_value.called)
+        else:
+            mock_track_function.return_value.assert_called_once_with(event_type, event)
+