def test_revert_new_exports_restore_old(self): saved_export_schema = SavedExportSchema(index=['my-domain', 'xmlns']) saved_export_schema.doc_type += DELETED_SUFFIX saved_export_schema.save() self.new_exports[0].legacy_saved_export_schema_id = saved_export_schema._id reverted = revert_new_exports(self.new_exports) self.assertEqual(len(reverted), 1) self.assertFalse(reverted[0].doc_type.endswith(DELETED_SUFFIX)) saved_export_schema.delete()
def _get_saved_exports(domain, has_deid_permissions, old_exports_getter, new_exports_getter): exports = old_exports_getter(domain) new_exports = new_exports_getter(domain) if use_new_exports(domain): exports += new_exports else: from corehq.apps.export.utils import revert_new_exports exports += revert_new_exports(new_exports) if not has_deid_permissions: exports = [e for e in exports if not e.is_safe] return sorted(exports, key=lambda x: x.name)
def test_revert_new_exports(self): reverted = revert_new_exports(self.new_exports) self.assertListEqual(reverted, []) for export in self.new_exports: self.assertTrue(export.doc_type.endswith(DELETED_SUFFIX))