Example #1
0
 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)
Example #2
0
 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)
Example #3
0
 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'])
Example #4
0
 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'])