Changeset 5800 for main/waeup.sirp/trunk/src/waeup/sirp/applicants/jambtables/tests

Timestamp:

7 Mar 2011, 02:14:02 (14 years ago)

Author:

uli

Message:

Add tests for new function.

File:

• main/waeup.sirp/trunk/src/waeup/sirp/applicants/jambtables/tests/test_jambtables.py (modified) (2 diffs)

main/waeup.sirp/trunk/src/waeup/sirp/applicants/jambtables/tests/test_jambtables.py

@@ -38 +38 @@
 from zope.site import LocalSiteManager
 from zope.site.hooks import setSite, clearSite
+from zope.site.testing import createSiteManager
 
 from waeup.sirp.app import University
 from waeup.sirp.datacenter import DataCenter
+from waeup.sirp.testing import WAeUPSIRPUnitTestLayer
 from waeup.sirp.applicants.jambtables import (
-    JAMBDataPlugin, JAMBDataRoot,
+    JAMBDataPlugin, JAMBDataRoot, get_jambdata,
     )
 from waeup.sirp.applicants.jambtables.interfaces import (
     IJAMBDataRoot,
     )
+from waeup.sirp.applicants import Applicant
+from waeup.sirp.applicants.container import ApplicantsContainer
+
 
 class FakeSite(dict):
@@ -175 +180 @@
         return
 
-        
+class GetJAMBDataTestCase(unittest.TestCase):
+
+    layer = WAeUPSIRPUnitTestLayer
+    
+    def setUp(self):
+        self.app = University()
+        # Fill up some JAMB data...
+        self.appl1 = Applicant()
+        self.appl2 = Applicant()
+        self.app['jambdata']['91100546DD'] = self.appl1
+        self.app['jambdata']['91100546DE'] = self.appl2
+        createSiteManager(self.app, setsite=True)
+        return
+
+    def tearDown(self):
+        pass
+
+    def test_contained(self):
+        self.assertEqual(get_jambdata('91100546DD'), self.appl1)
+        return
+
+    def test_not_contained(self):
+        self.assertEqual(get_jambdata('not-existing'), None)
+        return
+
 def test_suite():
     suite = unittest.TestSuite()
     for testcase in [
-        JAMBDataRootTestCase, JAMBDataPluginTestCase,
+        JAMBDataRootTestCase, JAMBDataPluginTestCase, GetJAMBDataTestCase,
         ]:
         suite.addTest(unittest.TestLoader().loadTestsFromTestCase(