From ebbeeb16a228d9001c3db6285529f7cfd42cabe6 Mon Sep 17 00:00:00 2001
From: Chris Dodge <cdodge@edx.org>
Date: Wed, 31 Jul 2013 15:53:04 -0400
Subject: [PATCH] add some unit tests for the new fields

---
 .../lib/xmodule/xmodule/tests/test_course_module.py  | 12 +++++++++++-
 common/test/data/toy/roots/2012_Fall.xml             |  2 +-
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/common/lib/xmodule/xmodule/tests/test_course_module.py b/common/lib/xmodule/xmodule/tests/test_course_module.py
index 53181b5a28..48690a8b25 100644
--- a/common/lib/xmodule/xmodule/tests/test_course_module.py
+++ b/common/lib/xmodule/xmodule/tests/test_course_module.py
@@ -53,7 +53,7 @@ def get_dummy_course(start, announcement=None, is_new=None, advertised_start=Non
     end = to_attrb('end', end)
 
     start_xml = '''
-         <course org="{org}" course="{course}"
+         <course org="{org}" course="{course}" display_organization="{org}_display" display_coursenumber="{course}_display"
                 graceperiod="1 day" url_name="test"
                 start="{start}"
                 {announcement}
@@ -141,6 +141,16 @@ class IsNewCourseTestCase(unittest.TestCase):
             print "Checking start=%s advertised=%s" % (s[0], s[1])
             self.assertEqual(d.start_date_text, s[2])
 
+    def test_display_organization(self):
+        descriptor = get_dummy_course(start='2012-12-02T12:00', is_new=True)
+        self.assertNotEqual(descriptor.location.org, descriptor.display_org_with_default)
+        self.assertEqual(descriptor.display_org_with_default, "{0}_display".format(ORG))
+
+    def test_display_coursenumber(self):
+        descriptor = get_dummy_course(start='2012-12-02T12:00', is_new=True)
+        self.assertNotEqual(descriptor.location.course, descriptor.display_number_with_default)
+        self.assertEqual(descriptor.display_number_with_default, "{0}_display".format(COURSE))
+
     def test_is_newish(self):
         descriptor = get_dummy_course(start='2012-12-02T12:00', is_new=True)
         assert(descriptor.is_newish is True)
diff --git a/common/test/data/toy/roots/2012_Fall.xml b/common/test/data/toy/roots/2012_Fall.xml
index b71528809b..2ed8c5458c 100644
--- a/common/test/data/toy/roots/2012_Fall.xml
+++ b/common/test/data/toy/roots/2012_Fall.xml
@@ -1 +1 @@
-<course org="edX" course="toy" url_name="2012_Fall"/>
\ No newline at end of file
+<course org="edX" course="toy" url_name="2012_Fall" display_organization="edX_display" display_coursenum="2012_Fall_Display" />
\ No newline at end of file