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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
#!/usr/bin/env python
#
# Copyright (C) 2013 Martin Owens
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 3.0 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library.
#
"""
Test cron item removal
"""
import os
import sys
sys.path.insert(0, '../')
import unittest
from crontab import CronTab, PY3
try:
from test import test_support
except ImportError:
from test import support as test_support
if PY3:
unicode = str
START_TAB = """
3 * * * * command1 # CommentID C
2 * * * * command2 # CommentID AAB
1 * * * * command3 # CommentID B3
"""
class RemovalTestCase(unittest.TestCase):
"""Test basic functionality of crontab."""
def setUp(self):
self.filenames = []
self.crontab = CronTab(tab=START_TAB.strip())
def test_01_remove(self):
"""Remove Item"""
self.assertEqual(len(self.crontab), 3)
self.crontab.remove(self.crontab.crons[0])
self.assertEqual(len(self.crontab), 2)
self.assertEqual(len(self.crontab.render()), 69)
def test_02_remove_all(self):
"""Remove All"""
self.crontab.remove_all()
self.assertEqual(len(self.crontab), 0)
self.assertEqual(unicode(self.crontab), '')
def test_03_remove_cmd(self):
"""Remove all with Command"""
self.crontab.remove_all(command='command2')
self.assertEqual(len(self.crontab), 2)
self.assertEqual(len(self.crontab.render()), 67)
self.crontab.remove_all(command='command3')
self.assertEqual(len(self.crontab), 1)
self.assertEqual(len(self.crontab.render()), 33)
def test_04_remove_id(self):
"""Remove all with Comment/ID"""
self.crontab.remove_all(comment='CommentID B3')
self.assertEqual(len(self.crontab), 2)
self.assertEqual(len(self.crontab.render()), 68)
def test_05_remove_date(self):
"""Remove all with Time Code"""
self.crontab.remove_all(time='2 * * * *')
self.assertEqual(len(self.crontab), 2)
self.assertEqual(len(self.crontab.render()), 67)
def test_05_remove_all_error(self):
"""Remove all with old arg"""
with self.assertRaises(AttributeError):
self.crontab.remove_all('command')
def test_06_removal_of_none(self):
"""Remove all respects None as a possible value"""
self.crontab[1].set_comment(None)
self.crontab.remove_all(comment=None)
self.assertEqual(len(self.crontab), 2)
self.assertEqual(len(self.crontab.render()), 67)
def test_07_removal_in_loop(self):
"""Remove items in a loop"""
for job in self.crontab:
self.crontab.remove(job)
self.assertEqual(len(self.crontab), 0)
def test_08_removal_write_loop(self):
"""Remove items in a loop and save each time"""
filename = self.get_new_file('removal')
crontab = CronTab()
for i in range(10):
crontab.new('test', comment=str(i))
crontab.write(filename)
crontab = CronTab(tabfile=filename)
self.assertEqual(len(crontab), 1)
self.assertNotEqual(str(crontab), '')
crontab.remove_all(comment=str(i))
crontab.write(filename)
crontab = CronTab(tabfile=filename)
self.assertEqual(len(crontab), 0)
self.assertEqual(str(crontab), '')
def test_09_removal_during_iter(self):
crontab = CronTab()
for x in range(0, 5, 1):
job = crontab.new(command="cmd", comment="SAME_ID")
job.setall("%d * * * *" % (x + 1))
for item in crontab.find_comment("SAME_ID"):
crontab.remove(item)
self.assertEqual(len(crontab), 0)
def test_11_remove_generator(self):
"""Remove jobs from the find generator"""
tabs = self.crontab.find_command('command2')
self.crontab.remove(tabs)
self.assertEqual(len(self.crontab), 2)
def test_12_remove_nonsense(self):
"""Fail to remove bad type"""
self.assertRaises(TypeError, self.crontab.remove, 5)
self.assertRaises(TypeError, self.crontab.remove, "foo")
def get_new_file(self, name):
"""Gets a filename and records it for deletion"""
this_dir = os.path.dirname(__file__)
filename = os.path.join(this_dir, 'data', 'spool', name)
self.filenames.append(filename)
return filename
def tearDown(self):
for filename in self.filenames:
if os.path.isfile(filename):
os.unlink(filename)
if __name__ == '__main__':
test_support.run_unittest(
RemovalTestCase,
)
|
