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 get_native_datatap(self, instream=None, **kwargs):
'''
Returns a ModelDataTap suited for this resource
'''
if instream is None:
instream = [self.resource_adaptor]
return DocumentDataTap(instream, **kwargs)
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)
