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

Timestamp:

27 Jan 2015, 08:31:34 (10 years ago)

Author:

Henrik Bettermann

Message:

Export datasets of selected items specified by a list of identifiers. Will be used in students and applicants modules.

File:

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

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

@@ -342 +342 @@
         # 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')
+
+    def test_export_selected(self):
+        # we can pass in a list of identifiers
+        exporter = ExporterBase()
+        self.assertRaises(NotImplementedError, exporter.export_selected,
+                          'foo', selected=[])
         return
 
@@ -360 +365 @@
     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
+
+    def export_selected(self, site, filepath=None, selected=None):
+        if selected:
             open(filepath, 'wb').write(SAMPLE_FILTERED_DATA)
             return
@@ -399 +411 @@
         # 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
+
+    def test_export_job_selected(self):
+        # we export subsets
+        result_path = export_job(None, 'cave_exporter', selected=['foo'])
         contents = open(result_path, 'rb').read()
         shutil.rmtree(os.path.dirname(result_path))