def test_get_item_stream(self): Application.objects.all().delete() Application(name='test app', slug='test-app').save() tap = DocKitCMSDataTap(applications=['test-app']) items = list(tap) self.assertTrue(items) self.assertEqual(len(items), Application.objects.all().count()) assert len(items) tap.close()
def test_load_item(self): Application.objects.all().delete() source = MemoryDataTap([{ 'collection': Application._meta.collection, 'fields': { 'name': 'test app', 'slug': 'test-app', } }]) tap = DocKitCMSDataTap(instream=source) result = list(tap) self.assertTrue(len(result), 1) self.assertTrue(hasattr(result[0], 'save')) tap.commit() #this saves said objects tap.close() self.assertEqual(Application.objects.all().count(), 1)