def test_rotate_then_upload(self, mock_check_output):
logfile = self.tempdir + '/eve.json'
with open(logfile, 'w'):
pass
after_rename = '{}.{}.archived'.format(logfile, '12345678')
mock_check_output.return_value = 0
mock_check_output.side_effect = rename(logfile, after_rename)
now = datetime(2015, 3, 12)
watcher = SuricataAlertWatcher(log_dir=self.tempdir)
watcher.api = MagicMock()
watcher.api.send_file.return_value = 'send_destination'
watcher._rotate_logs()
self.assertFalse(path.exists(logfile))
self.assertTrue(path.exists(after_rename))
watcher._upload(now)
self.assertFalse(path.exists(after_rename))
self.assertEquals(watcher.api.send_file.call_args_list, [
call('logs', after_rename, now, suffix='suricata'),
])
self.assertEquals(watcher.api.send_signal.call_args_list, [
call('logs', {
'path': 'send_destination',
'utcoffset': 0,
'log_type': 'suricata',
'ip': '10.1.1.1',
})
])
def test_rotate_logs(self, mock_check_output):
watcher = SuricataAlertWatcher()
# Success
mock_check_output.return_value = 0
watcher._rotate_logs()
mock_check_output.assert_called_once_with(
['sudo', '-u', 'suricata', MANAGE_SCRIPT, 'rotate-logs'])
# Failure - don't die
mock_check_output.side_effect = CalledProcessError(1, '')
watcher._rotate_logs()
def test_rotate_logs(self, mock_check_output):
watcher = SuricataAlertWatcher()
# Success
mock_check_output.return_value = 0
watcher._rotate_logs()
mock_check_output.assert_called_once_with(
['sudo', '-u', 'suricata', MANAGE_SCRIPT, 'rotate-logs']
)
# Failure - don't die
mock_check_output.side_effect = CalledProcessError(1, '')
watcher._rotate_logs()