def test_04_check_pid(self):
lf = LockFile(self.LOCKPATH)
pid = lf.get_pid()
if sys.version_info <= (3, 0):
# py2.x coverage
regex_match = re.match(r'[0-9]+$', pid)
self.assertIsNotNone(regex_match)
else:
self.assertRegex(pid, r'^[0-9]+$')
def setUp(self):
self.directory = ospj(dirname(__file__), 'data/deny_hosts')
self.work_dir = ospj(self.directory, 'work')
self.logfile = ospj(self.work_dir, 'logfile')
self.prefs = Prefs()
self.lock_file = LockFile(ospj(self.directory, 'lockfile'))
self.lock_file.remove(die_=False)
self.lock_file.create()
self.prefs._Prefs__data['ETC_DIR'] = ospj(self.directory, 'etc')
self.prefs._Prefs__data['WORK_DIR'] = self.work_dir
def setup_denyhosts(self):
self.prefs = Prefs()
self.lock_file = LockFile(ospj(self.DIRECTORY, 'lockfile'))
self.lock_file.remove(die_=False)
self.lock_file.create()
self.prefs._Prefs__data['ETC_DIR'] = ospj(self.DIRECTORY, 'etc')
self.prefs._Prefs__data['WORK_DIR'] = self.WORK_DIR
self.prefs._Prefs__data['HOSTS_DENY'] = ospj(self.prefs._Prefs__data['ETC_DIR'], 'hosts.deny')
self.prefs._Prefs__data['DENY_THRESHOLD_INVALID'] = 5
self.prefs._Prefs__data['DENY_THRESHOLD_VALID'] = 5
self.prefs._Prefs__data['DENY_THRESHOLD_ROOT'] = 0
self.prefs._Prefs__data['DENY_THRESHOLD_RESTRICTED'] = 5
first_time = 1
except Exception, e:
if e[0] != 17:
print e
sys.exit(1)
setup_logging(prefs, enable_debug, verbose, daemon)
if not logfiles or daemon:
logfiles = [prefs.get('SECURE_LOG')]
elif len(logfiles) > 1:
ignore_offset = 1
if not prefs.get('ADMIN_EMAIL'): noemail = 1
lock_file = LockFile(prefs.get('LOCK_FILE'))
lock_file.create()
if upgrade099 and not daemon:
if not prefs.get('PURGE_DENY'):
lock_file.remove()
die("You have supplied the --upgrade099 flag, however you have not set PURGE_DENY in your configuration file")
else:
u = UpgradeTo099(prefs.get("HOSTS_DENY"))
if migrate and not daemon:
if not prefs.get('PURGE_DENY'):
lock_file.remove()
die("You have supplied the --migrate flag however you have not set PURGE_DENY in your configuration file.")
else:
def test_09_create_remove(self):
lf = LockFile(self.LOCKPATH)
self.assertIsNone(lf.create())
self.assertIsNone(lf.remove())
def test_08_get_pid_empty(self):
lf = LockFile(self.LOCKPATH)
self.assertEqual(lf.get_pid(), '')
def test_07_check_lockfile_removal_not_exists(self):
lf = LockFile(self.LOCKPATH)
with self.assertRaises(SystemExit) as cm:
lf.remove()
self.assertEqual(cm.exception, 'Error')
def test_06_check_lockfile_removal(self):
lf = LockFile(self.LOCKPATH)
self.assertIsNone(lf.remove())
def test_05_check_lockfile_create_exists(self):
lf = LockFile(self.LOCKPATH)
with self.assertRaises(SystemExit) as cm:
lf.create()
self.assertEqual(cm.exception, 'Error')
def test_03_check_lockfile_exists(self):
lf = LockFile(self.LOCKPATH)
self.assertTrue(lf.exists())
def test_02_check_lockfile_create(self):
lf = LockFile(self.LOCKPATH)
self.assertIsNone(lf.create())
close(lf.fd)
def test_01_check_lockfile_not_exists(self):
lf = LockFile(self.LOCKPATH)
self.assertFalse(lf.exists())
