def test_find_inmates_with_exclude_list(self):
number_to_fetch = 4
expected = expect_jail_id_calls(number_to_fetch)
inmate_scraper = Mock()
monitor = Mock()
search_commands = SearchCommands(inmate_scraper, monitor)
search_commands.find_inmates(gen_inmate_ids(yesterday(), number_to_fetch)[1:-1], number_to_fetch=number_to_fetch)
assert inmate_scraper.create_if_exists.call_args_list == [expected[0], expected[number_to_fetch - 1]]
def test_find_inmates(self):
number_to_fetch = 3
expected = expect_jail_id_calls(number_to_fetch)
inmate_scraper = Mock()
monitor = Mock()
search_commands = SearchCommands(inmate_scraper, monitor)
search_commands.find_inmates(number_to_fetch=number_to_fetch)
assert inmate_scraper.create_if_exists.call_args_list == expected
assert monitor.notify.call_args_list == [call(search_commands.__class__, search_commands.FINISHED_FIND_INMATES)]
def test_check_if_really_discharged(self):
number_to_fetch = 3
expected = expect_jail_id_calls(number_to_fetch)
inmate_scraper = Mock()
monitor = Mock()
search_commands = SearchCommands(inmate_scraper, monitor)
search_commands.check_if_really_discharged(gen_inmate_ids(yesterday(), number_to_fetch))
assert inmate_scraper.resurrect_if_found.call_args_list == expected
assert monitor.notify.call_args_list == [call(search_commands.__class__,
search_commands.FINISHED_CHECK_OF_RECENTLY_DISCHARGED_INMATES)]
def test_update_inmates_status(self):
number_to_fetch = 8
jail_ids = range(number_to_fetch)
expected = map(lambda x: call(x), jail_ids)
inmate_scraper = Mock()
monitor = Mock()
search_commands = SearchCommands(inmate_scraper, monitor)
search_commands.update_inmates_status(jail_ids)
assert inmate_scraper.update_inmate_status.call_args_list == expected
assert monitor.notify.call_args_list == [call(search_commands.__class__,
search_commands.FINISHED_UPDATE_INMATES_STATUS)]
def test_find_missing_inmates(self):
number_to_fetch = 3
number_days_to_fetch = 4
start_date = date.today() - ONE_DAY * number_days_to_fetch
jail_ids = []
excluded_jail_ids = []
for day_index in range(number_days_to_fetch):
daily_jail_ids = gen_inmate_ids(start_date + ONE_DAY * day_index, number_to_fetch)
excluded_jail_ids.append(daily_jail_ids.pop((day_index + 1) % number_to_fetch))
jail_ids.extend(daily_jail_ids)
expected = map(lambda x: call(x), jail_ids)
inmate_scraper = Mock()
monitor = Mock()
search_commands = SearchCommands(inmate_scraper, monitor)
search_commands.find_inmates(excluded_jail_ids, number_to_fetch=number_to_fetch, start_date=start_date)
assert inmate_scraper.create_if_exists.call_args_list == expected
assert monitor.notify.call_args_list == [call(search_commands.__class__, search_commands.FINISHED_FIND_INMATES)]