def test_process_source(self, mock_dump, mock_read):
test_data_dict = [
{
'date': '01/01/10',
'fips': '12081',
'open': '268',
'current': '260',
'thirty': '4',
'sixty': '1',
'ninety': '0',
'other': '3'
},
{
'date': '01/02/10',
'fips': '12081',
'open': '280',
'current': '290',
'thirty': '20',
'sixty': '10',
'ninety': '4',
'other': '3'
},
]
mock_reader = (row for row in test_data_dict)
mock_read.return_value = mock_reader
process_source(
self.start_date, self.through_date, dump_slug='mock_csv_slug')
self.assertEqual(CountyMortgageData.objects.count(), 2)
self.assertEqual(mock_read.call_count, 1)
self.assertEqual(mock_dump.call_count, 1)
def test_process_source(self, mock_dump, mock_read):
test_data_dict = [
{
'date': '01/01/10',
'fips': '12081',
'open': '268',
'current': '260',
'thirty': '4',
'sixty': '1',
'ninety': '0',
'other': '3'
},
{
'date': '01/02/10',
'fips': '12081',
'open': '280',
'current': '290',
'thirty': '20',
'sixty': '10',
'ninety': '4',
'other': '3'
},
]
mock_reader = (row for row in test_data_dict)
mock_read.return_value = mock_reader
process_source(
self.start_date, self.through_date, dump_slug='mock_csv_slug')
self.assertEqual(CountyMortgageData.objects.count(), 2)
self.assertEqual(mock_read.call_count, 1)
self.assertEqual(mock_dump.call_count, 1)