def testCsvSource(self):
source = CsvSource.from_buffer(StringIO.StringIO(self.INPUT),
self.MODEL)
models = list(source)
self.assertEqual(models[0]._get_content(),
{'description': 'some description',
'subject': 'some subject'})
self.assertEqual(models[1]._get_content(),
{'description': 'another description',
'subject': 'another subject'})
def testImportFromSource(self):
urlopen_mock = self.mocker.replace('urllib2.urlopen')
for id_ in self.EXPECTED_IDS:
urlopen_mock(mocker.MATCH(self._match_request))
self.mocker.result(StringIO.StringIO(
'{"issue": {"id": %s}}' % id_))
self.mocker.replay()
source = CsvSource.from_buffer(StringIO.StringIO(self.INPUT),
self.MODEL)
results = self.importer.import_from_source(source)
self.assertEqual(
list(results),
self.EXPECTED_IDS)
