def test_generate_explicit_output(self): with tempdir() as temp_dir: with mock_system_call() as mock_system: generate(self.db_rw, 's3_bucket', self.raven_client, self.stats_client, upload=False, concurrency=1, datamaps='', output=temp_dir) mock_calls = mock_system.mock_calls self.assertEqual(len(mock_calls), 3)
def test_generate_explicit_output(self): with tempdir() as temp_dir: with mock_system_call() as mock_system: generate(self.db_master, 's3_bucket', self.heka_client, self.stats_client, upload=False, concurrency=1, datamaps='', output=temp_dir) mock_calls = mock_system.mock_calls self.assertEqual(len(mock_calls), 3)
def test_generate(self): with mock_system_call() as mock_system: generate(self.db_rw, 's3_bucket', self.raven_client, self.stats_client, upload=False, concurrency=1, datamaps='') mock_calls = mock_system.mock_calls self.assertEqual(len(mock_calls), 3) self.assertTrue(mock_calls[0][1][0].startswith('encode')) self.assertTrue(mock_calls[1][1][0].startswith('enumerate')) self.assertTrue(mock_calls[2][1][0].startswith('enumerate')) self.check_stats( timer=['datamaps.export_to_csv', 'datamaps.csv_rows', 'datamaps.encode', 'datamaps.render'])
def test_generate(self): with mock_system_call() as mock_system: generate(self.db_master, 's3_bucket', self.heka_client, self.stats_client, upload=False, concurrency=1, datamaps='') mock_calls = mock_system.mock_calls self.assertEqual(len(mock_calls), 3) self.assertTrue(mock_calls[0][1][0].startswith('encode')) self.assertTrue(mock_calls[1][1][0].startswith('enumerate')) self.assertTrue(mock_calls[2][1][0].startswith('enumerate')) self.check_stats( timer=['datamaps.export_to_csv', 'datamaps.csv_rows', 'datamaps.encode', 'datamaps.render'])