Changeset 6544 for main/waeup.sirp/trunk/src/waeup/sirp

Timestamp:

23 Jul 2011, 10:23:30 (13 years ago)

Author:

uli

Message:

Add test for reimport of accesscodes.

File:

• main/waeup.sirp/trunk/src/waeup/sirp/accesscodes/tests/test_accesscodes.py (modified) (7 diffs)

main/waeup.sirp/trunk/src/waeup/sirp/accesscodes/tests/test_accesscodes.py

@@ -30 +30 @@
 from hurry.workflow.interfaces import InvalidTransitionError, IWorkflowState
 from zope.app.testing.functional import (
-    FunctionalTestCase, FunctionalTestSetup, getRootFolder)
+    FunctionalTestSetup, getRootFolder)
 from zope.component.hooks import setSite, clearSite
 from zope.interface.verify import verifyObject, verifyClass
@@ -36 +36 @@
 from waeup.sirp.app import University
 from waeup.sirp.interfaces import IObjectHistory
-from waeup.sirp.testing import FunctionalLayer
+from waeup.sirp.testing import FunctionalLayer, FunctionalTestCase
 from waeup.sirp.accesscodes.accesscodes import (
     AccessCodeBatch, get_access_code, invalidate_accesscode, AccessCode,
@@ -46 +46 @@
 
 
-
+optionflags = (
+    doctest.REPORT_NDIFF + doctest.ELLIPSIS + doctest.NORMALIZE_WHITESPACE)
 
 class AccessCodeHelpersTests(FunctionalTestCase):
@@ -257 +258 @@
         self.app = self.getRootFolder()['app']
 
+        batch = AccessCodeBatch(    # create batch with zero entries
+            datetime.now(), 'testuser', 'FOO', 9.99, 0)
+        self.app['accesscodes'].addBatch(batch)
+
+        self.ac1 = AccessCode(0, '11111111')
+        self.ac2 = AccessCode(1, '22222222')
+        self.ac3 = AccessCode(2, '33333333')
+        batch['FOO-1-11111111'] = self.ac1
+        batch['FOO-1-22222222'] = self.ac2
+        batch['FOO-1-33333333'] = self.ac3
+        self.batch = batch
+
         setSite(self.app)
         return
@@ -271 +284 @@
         assert verifyClass(IAccessCodeBatch, AccessCodeBatch)
 
+    def test_csv_export(self):
+        # Make sure CSV export of accesscodes works
+        batch = self.batch
+        invalidate_accesscode('FOO-1-11111111', comment='comment with "quotes"')
+        disable_accesscode('FOO-1-33333333')
+        basename = batch.archive()
+        result_path = os.path.join(batch._getStoragePath(), basename)
+        expected = '''
+"prefix","serial","ac","state","history"
+"FOO","9.99","1","0"
+"FOO","0","FOO-1-11111111","used","<YYYY-MM-DD hh:mm:ss> - ..."
+"FOO","1","FOO-1-22222222","initialized","<YYYY-MM-DD hh:mm:ss> - ..."
+"FOO","2","FOO-1-33333333","disabled","<YYYY-MM-DD hh:mm:ss> - ..."
+'''[1:]
+        contents = open(result_path, 'rb').read()
+        self.assertMatches(expected, contents)
+
 class AccessCodeBatchContainerTests(FunctionalTestCase):
     # Tests for AccessCodeContainer class
@@ -288 +318 @@
         self.app = self.getRootFolder()['app']
 
+        self.import_sample1_src = os.path.join(
+            os.path.dirname(__file__), 'sample_import.csv')
+
         setSite(self.app)
         return
@@ -300 +333 @@
         assert verifyObject(IAccessCodeBatchContainer, accesscodes)
         assert verifyClass(IAccessCodeBatchContainer, AccessCodeBatchContainer)
+
+    def test_csv_import(self):
+        # Make sure we can reimport sample data from local sample_import.csv
+        batchcontainer = self.app['accesscodes']
+        shutil.copyfile(        # Copy sample to import dir
+            os.path.join(os.path.dirname(__file__), 'sample_import.csv'),
+            os.path.join(batchcontainer._getStoragePath(), 'sample_import.csv')
+            )
+        batchcontainer.reimport('sample_import.csv')
+        batch = batchcontainer.get(u'FOO-1', None)
+        self.assertTrue(batch is not None)
+        keys = [x for x in batch.keys()]
+        self.assertEqual(
+            keys,
+            [u'FOO-1-11111111', u'FOO-1-22222222', u'FOO-1-33333333'])
+        self.fail('Continue writing tests here')