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 setUpClass(cls):
super(SavedExportSchemaDBTest, cls).setUpClass()
SavedExportSchema(domain='domain1', index=["domain1", "blah"]).save()
SavedExportSchema(domain='domain1', index=["domain2", "blah"]).save()
SavedExportSchema(domain='domain1', index=["domain2", "blah"]).save()
