Changeset 9797 for main/waeup.kofa/trunk/src/waeup/kofa/utils/tests

Timestamp:

13 Dec 2012, 15:39:31 (12 years ago)

Author:

uli

Message:

Local exports for departments.

File:

• main/waeup.kofa/trunk/src/waeup/kofa/utils/tests/test_batching.py (modified) (4 diffs)

main/waeup.kofa/trunk/src/waeup/kofa/utils/tests/test_batching.py

@@ -87 +87 @@
 """
 
+SAMPLE_FILTERED_DATA = """name,dinoports,owner,taxpayer
+Barneys Home,2,Barney,1
+Wilmas Asylum,1,Wilma,1
+"""
+
 class CaveProcessor(BatchProcessor):
     util_name = 'caveprocessor'
@@ -334 +339 @@
         return
 
+    def test_export_filtered(self):
+        # we can pass in positional and keyword args
+        exporter = ExporterBase()
+        writer, outfile = exporter.get_csv_writer(filepath=self.workfile)
+        self.assertRaises(NotImplementedError, exporter.export_filtered,
+                          'foo', bar='bar')
+        return
 
 class CaveExporter(ExporterBase):
@@ -344 +356 @@
             return SAMPLE_DATA
         open(filepath, 'wb').write(SAMPLE_DATA)
+        return
+
+    def export_filtered(self, site, filepath=None, foo=None, bar=None):
+        if foo or bar:
+            open(filepath, 'wb').write(SAMPLE_FILTERED_DATA)
+            return
+        self.export_all(site, filepath=filepath)
         return
 
@@ -375 +394 @@
         verify.verifyClass(IExportJob, AsyncExportJob)
         verify.verifyObject(IExportJob, job)
+        return
+
+    def test_export_job_with_args(self):
+        # we export filtered sets
+        result_path = export_job(None, 'cave_exporter', foo='foo')
+        contents = open(result_path, 'rb').read()
+        shutil.rmtree(os.path.dirname(result_path))
+        self.assertEqual(contents, SAMPLE_FILTERED_DATA)
         return