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)