github
/
fail2ban
mirror of https://github.com/fail2ban/fail2ban
1
0
Fork 0
Code Issues Releases Wiki Activity

ENH: close files in _test_move_into_file

pull/218/head
Yaroslav Halchenko 2013-05-08 15:14:28 -04:00
parent 582d1c5ea5
commit 28794d842d
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
testcases/filtertestcase.py
View File

@ -477,8 +477,8 @@ def get_monitor_failures_testcase(Filter_):
def _test_move_into_file(self, interim_kill=False): def _test_move_into_file(self, interim_kill=False):
# if we move a new file into the location of an old (monitored) file # if we move a new file into the location of an old (monitored) file
self.file1 = _copy_lines_between_files(GetFailures.FILENAME_01, self.name, _copy_lines_between_files(GetFailures.FILENAME_01, self.name,
n=100) n=100).close()
# make sure that it is monitored first # make sure that it is monitored first
self.assert_correct_last_attempt(GetFailures.FAILURES_01) self.assert_correct_last_attempt(GetFailures.FAILURES_01)
self.assertEqual(self.filter.failManager.getFailTotal(), 3) self.assertEqual(self.filter.failManager.getFailTotal(), 3)
@ -488,14 +488,15 @@ def get_monitor_failures_testcase(Filter_):
time.sleep(0.2) # let them know time.sleep(0.2) # let them know
# now create a new one to override old one # now create a new one to override old one
self.file = _copy_lines_between_files(GetFailures.FILENAME_01, _copy_lines_between_files(GetFailures.FILENAME_01, self.name + '.new',
self.name + '.new', n=100) n=100).close()
os.rename(self.name + '.new', self.name) os.rename(self.name + '.new', self.name)
self.assert_correct_last_attempt(GetFailures.FAILURES_01) self.assert_correct_last_attempt(GetFailures.FAILURES_01)
self.assertEqual(self.filter.failManager.getFailTotal(), 6) self.assertEqual(self.filter.failManager.getFailTotal(), 6)
# and to make sure that it now monitored for changes # and to make sure that it now monitored for changes
_copy_lines_between_files(GetFailures.FILENAME_01, self.name, n=100) _copy_lines_between_files(GetFailures.FILENAME_01, self.name,
n=100).close()
self.assert_correct_last_attempt(GetFailures.FAILURES_01) self.assert_correct_last_attempt(GetFailures.FAILURES_01)
self.assertEqual(self.filter.failManager.getFailTotal(), 9) self.assertEqual(self.filter.failManager.getFailTotal(), 9)