94 lines
2.7 KiB
Python
94 lines
2.7 KiB
Python
|
import os
|
||
|
|
import sys
|
||
|
|
import mock
|
||
|
|
|
||
|
|
|
||
|
|
class LogPickerTest(mock.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def getlogs_1_test(self):
|
||
|
|
from log_picker import LogPicker
|
||
|
|
|
||
|
|
miner = mock.Mock()
|
||
|
|
|
||
|
|
obj = LogPicker(miners=[miner], use_one_file=True)
|
||
|
|
obj.getlogs()
|
||
|
|
|
||
|
|
self.assertTrue(miner.set_logfile.called)
|
||
|
|
self.assertTrue(miner.getlog.called)
|
||
|
|
|
||
|
|
def getlogs_2_test(self):
|
||
|
|
from log_picker import LogPicker
|
||
|
|
from log_picker.logmining import LogMinerError
|
||
|
|
|
||
|
|
miner = mock.Mock()
|
||
|
|
miner.getlog.side_effect = LogMinerError('Just a test')
|
||
|
|
|
||
|
|
obj = LogPicker(miners=[miner], use_one_file=True)
|
||
|
|
# Temporary redirect stderr output
|
||
|
|
sys.stderr, backup = mock.Mock(), sys.stderr
|
||
|
|
obj.getlogs()
|
||
|
|
sys.stderr = backup
|
||
|
|
|
||
|
|
self.assertTrue(True)
|
||
|
|
|
||
|
|
def send_1_test(self):
|
||
|
|
from log_picker import LogPicker
|
||
|
|
|
||
|
|
sender = mock.Mock()
|
||
|
|
|
||
|
|
obj = LogPicker(sender_obj=sender)
|
||
|
|
obj.files = ["file"]
|
||
|
|
obj.send()
|
||
|
|
|
||
|
|
self.assertTrue(sender.sendfile.called)
|
||
|
|
self.assertEqual(sender.sendfile.call_args,
|
||
|
|
(('file', 'text/plain'), {}))
|
||
|
|
|
||
|
|
def send_2_test(self):
|
||
|
|
from log_picker import LogPicker
|
||
|
|
|
||
|
|
sender = mock.Mock()
|
||
|
|
archiver = mock.Mock()
|
||
|
|
archiver.mimetype = "application/anything"
|
||
|
|
|
||
|
|
obj = LogPicker(sender_obj=sender, archive_obj=archiver)
|
||
|
|
obj.files = ["file"]
|
||
|
|
obj.archive = "file2"
|
||
|
|
obj.send()
|
||
|
|
|
||
|
|
self.assertTrue(sender.sendfile.called)
|
||
|
|
self.assertEqual(sender.sendfile.call_args,
|
||
|
|
(('file2', 'application/anything'), {}))
|
||
|
|
|
||
|
|
def create_archive_1_test(self):
|
||
|
|
from log_picker import LogPicker
|
||
|
|
|
||
|
|
archiver = mock.Mock()
|
||
|
|
archiver.file_ext = ".zip"
|
||
|
|
|
||
|
|
obj = LogPicker(archive_obj=archiver)
|
||
|
|
obj.create_archive()
|
||
|
|
|
||
|
|
self.assertTrue(archiver.create_archive.called)
|
||
|
|
|
||
|
|
def get_tmp_file_test(self):
|
||
|
|
from log_picker import LogPicker
|
||
|
|
|
||
|
|
obj = LogPicker()
|
||
|
|
ret = obj._get_tmp_file("name")
|
||
|
|
self.assertTrue(os.path.isfile(ret))
|
||
|
|
|
||
|
|
def errprint_test(self):
|
||
|
|
from log_picker import LogPicker
|
||
|
|
|
||
|
|
obj = LogPicker()
|
||
|
|
# Temporary redirect stderr output
|
||
|
|
sys.stderr, backup = mock.Mock(), sys.stderr
|
||
|
|
obj._errprint("message")
|
||
|
|
res, sys.stderr = sys.stderr, backup
|
||
|
|
self.assertTrue(res.write.called)
|
||
|
|
self.assertEqual(res.write.call_args[0][0], 'message\n')
|
||
|
|
|