def test_logfilename(self):
loader = Loader('data.csv', model_class=TestModel)
name = loader.logfile.name
self.assertTrue(re.match(r'^data.csv.\d{4}-\d{2}-\d{2}.log$', name))
os.remove(name)
options = {'logfilename': 'test.log'}
loader = Loader('data.csv', model_class=TestModel, options=options)
self.assertEqual(loader.logfile.name, 'test.log')
os.remove('test.log')
loader = Loader(StringIO('test'), model_class=TestModel)
self.assertFalse(loader.logfile)
def test_filelikeobject(self):
with open(self.filename) as fil:
content = StringIO(text_type(fil.read()))
loader = Loader(content, model_class=TestModel)
loader.load()
self.assertEqual(TestModel.objects.all().count(), 3)
def test_load_from_file(self):
loader = Loader(self.filename, model_class=TestModel)
loader.load()
self.assertEqual(TestModel.objects.all().count(), 3)
def test_load_from_file(self):
loader = Loader(self.filename, model_class=TestModel)
loader.load()
self.assertEqual(TestModel.objects.all().count(), 3)
def test_feedbacksize(self):
loader = Loader(None, model_class=TestModel)
self.assertEqual(loader.feedbacksize, 5000)
options = {'feedbacksize': 20}
loader = Loader(None, model_class=TestModel, options=options)
self.assertEqual(loader.feedbacksize, 20)
def test_optionpassing(self):
options = {'create': False, 'update': True}
ldr = Loader('test', model_class=TestModel, options=options)
self.assertEqual(ldr.extractor.options, options)
self.assertFalse(ldr.generator.create)
def test_filelikeobject(self):
with open(self.filename) as fil:
content = StringIO(text_type(fil.read()))
loader = Loader(content, model_class=TestModel)
loader.load()
self.assertEqual(TestModel.objects.all().count(), 3)
