def test_verbose_debug_mode(self):
expected = 'hi'
log = Mock()
monitor = Monitor(log)
monitor.debug(expected)
monitor.debug(expected, debug_level=MONITOR_VERBOSE_DMSG_LEVEL)
assert len(log.debug.call_args_list) == 1
log = Mock()
monitor = Monitor(log, verbose_debug_mode=True)
monitor.debug(expected)
monitor.debug(expected, debug_level=MONITOR_VERBOSE_DMSG_LEVEL)
assert len(log.debug.call_args_list) == 2
def test_debug_msg(self):
timestamp = '*now*'
msg = 'hi'
expected = '%s - %s' % (str(timestamp), msg)
log = Mock()
monitor = Monitor(log)
monitor._debug(timestamp, msg)
log.debug.assert_called_once_with(expected)
def test_that_it_generates_heartbeats(self):
monitor = Monitor(None)
heartbeat = Heartbeat(monitor)
number_heartbeats_to_collect = 2
starting_time = datetime.now()
heartbeats = [
monitor.notification() for _ in range(number_heartbeats_to_collect)
]
ending_time = datetime.now()
assert len(heartbeats) == number_heartbeats_to_collect
assert (ending_time -
starting_time).seconds == number_heartbeats_to_collect
for hb in heartbeats:
assert hb[0] == heartbeat.__class__
def ng_scraper():
parser = argparse.ArgumentParser(
description="Scrape inmate data from Cook County Sheriff's site.")
parser.add_argument(
'-d',
'--day',
action='store',
dest='start_date',
default=None,
help=(
'Specify day to search for missing inmates, format is YYYY-MM-DD. '
'If not specified, searches all days.'))
parser.add_argument('--verbose',
action="store_true",
dest='verbose',
default=False,
help='Turn on verbose mode.')
args = parser.parse_args()
try:
monitor = Monitor(log, verbose_debug_mode=args.verbose)
monitor.debug(
"%s - Started scraping inmates from Cook County Sheriff's site." %
datetime.now())
scraper = Scraper(monitor)
if args.start_date:
scraper.check_for_missing_inmates(
datetime.strptime(args.start_date, '%Y-%m-%d').date())
else:
scraper.run(date.today() - timedelta(1), feature_controls())
monitor.debug(
"%s - Finished scraping inmates from Cook County Sheriff's site." %
datetime.now())
except Exception, e:
log.exception(e)
def test_notify(self):
notifier = Mock(spec=Test_Monitor)
expected = (notifier, '')
monitor = Monitor(None)
monitor.notify(notifier)
assert monitor.notification() == expected
def test_debug_msgs_off(self):
expected = 'hi'
log = Mock()
monitor = Monitor(log, no_debug_msgs=True)
monitor.debug(expected)
assert not log.debug.called, 'log.debug should not have been called'
def setup_method(self, method):
self._monitor = Monitor(Mock())
self._search = Mock()
self._inmate_scraper = Mock()
