1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
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')
|