def test_run(self, mock_os, mock_tempfile, mock_storage):
"""Test run()."""
fake_tmpname = 'tmp_name'
fake_output_name = 'abc'
gvp = gcs_violations.GcsViolations(
'abcd',
1514764800123456,
[],
self.fake_global_conf,
{},
self.fake_notifier_conf)
gvp._get_output_filename = mock.MagicMock(return_value=fake_output_name)
gcs_path = '{}/{}'.format(
gvp.notification_config['gcs_path'], fake_output_name)
mock_tmp_csv = mock.MagicMock()
mock_tempfile.return_value = mock_tmp_csv
mock_tmp_csv.name = fake_tmpname
mock_tmp_csv.write = mock.MagicMock()
gvp.run()
mock_tmp_csv.write.assert_called()
mock_storage.return_value.put_text_file.assert_called_once_with(
fake_tmpname, gcs_path)
def test_run_with_invalid_data_format(self, mock_write_csv,
mock_json_stringify, mock_storage):
"""Test run() with json file format."""
notifier_config = (
fake_violations.NOTIFIER_CONFIGS_GCS_INVALID_DATA_FORMAT)
notification_config = notifier_config['resources'][0]['notifiers'][0]['configuration']
resource = 'policy_violations'
cycle_timestamp = '2018-03-24T00:49:02.891287'
mock_json_stringify.return_value = 'test123'
gvp = gcs_violations.GcsViolations(
resource,
cycle_timestamp,
fake_violations.VIOLATIONS,
fake_violations.GLOBAL_CONFIGS,
notifier_config,
notification_config)
gvp._get_output_filename = mock.MagicMock()
with self.assertRaises(base_notification.InvalidDataFormatError):
gvp.run()
self.assertFalse(gvp._get_output_filename.called)
self.assertFalse(mock_write_csv.called)
self.assertFalse(mock_json_stringify.called)
def test_run_with_json(self, mock_write_csv, mock_json_stringify,
mock_storage):
"""Test run() with json file format."""
notifier_config = fake_violations.NOTIFIER_CONFIGS_GCS_JSON
notification_config = notifier_config['resources'][0]['notifiers'][0]['configuration']
resource = 'policy_violations'
cycle_timestamp = '2018-03-24T00:49:02.891287'
mock_json_stringify.return_value = 'test123'
gvp = gcs_violations.GcsViolations(
resource,
cycle_timestamp,
fake_violations.VIOLATIONS,
fake_violations.GLOBAL_CONFIGS,
notifier_config,
notification_config)
gvp._get_output_filename = mock.MagicMock()
gvp.run()
self.assertTrue(gvp._get_output_filename.called)
self.assertEqual(
string_formats.VIOLATION_JSON_FMT,
gvp._get_output_filename.call_args[0][0])
self.assertFalse(mock_write_csv.called)
self.assertTrue(mock_json_stringify.called)
def test_get_output_filename_with_json(self, mock_date_time):
"""Test _get_output_filename()."""
mock_date_time.get_utc_now_datetime = mock.MagicMock()
mock_date_time.get_utc_now_datetime.return_value = self.fake_utcnow
expected_timestamp = self.fake_utcnow.strftime(
string_formats.TIMESTAMP_TIMEZONE_FILES)
gvp = gcs_violations.GcsViolations(
'abcd',
1514764800123456,
[],
self.fake_global_conf,
{},
self.fake_notifier_conf)
actual_filename = gvp._get_output_filename(
string_formats.VIOLATION_JSON_FMT)
self.assertEqual(
string_formats.VIOLATION_JSON_FMT.format(
gvp.resource, gvp.inventory_index_id, expected_timestamp),
actual_filename)
def test_run_with_csv(self, mock_csv_writer, mock_parser, mock_storage):
"""Test run() with default file format (CSV)."""
notifier_config = fake_violations.NOTIFIER_CONFIGS_GCS_DEFAULT
notification_config = notifier_config['resources'][0]['notifiers'][0]['configuration']
resource = 'policy_violations'
cycle_timestamp = '2018-03-24T00:49:02.891287'
gvp = gcs_violations.GcsViolations(
resource,
cycle_timestamp,
fake_violations.VIOLATIONS,
fake_violations.GLOBAL_CONFIGS,
notifier_config,
notification_config)
gvp._get_output_filename = mock.MagicMock()
gvp.run()
self.assertTrue(gvp._get_output_filename.called)
self.assertEqual(
string_formats.VIOLATION_CSV_FMT,
gvp._get_output_filename.call_args[0][0])
self.assertTrue(mock_csv_writer.called)
self.assertFalse(mock_parser.called)
