def test_recently_discharged_inmates_ids(self): inmate_class = Mock() j_ids = [j_id for j_id in range(1, 4)] input_values = [make_county_inmate(j_id) for j_id in j_ids] inmate_class.recently_discharged_inmates.return_value = input_values inmates = Inmates(inmate_class, Mock()) response_q = Queue(1) inmates.recently_discharged_inmates_ids(response_q) recently_discharged_inmates_ids = response_q.get() assert recently_discharged_inmates_ids == j_ids