def test_get_item_stream(self): SimpleDocument.objects.all().delete() SimpleDocument(charfield='testchar').save() tap = DocumentDataTap(instream=[SimpleDocument]) items = list(tap) self.assertTrue(items) self.assertEqual(len(items), SimpleDocument.objects.all().count()) assert len(items) tap.close()
def test_load_item(self): SimpleDocument.objects.all().delete() source = MemoryDataTap([{ 'collection': SimpleDocument._meta.collection, 'fields': { 'charfield': 'testchar', 'published': True, 'featured': False, } }]) tap = DocumentDataTap(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(SimpleDocument.objects.all().count(), 1)