def test_get_all_mine_verified_status(test_client, db_session, auth_headers): healthy_count = 2 MineFactory.create_batch(size=healthy_count, verified_status__healthy_ind=True) unhealthy_count = 2 MineFactory.create_batch(size=unhealthy_count, verified_status__healthy_ind=False) get_resp = test_client.get('/mines/verified-status', headers=auth_headers['full_auth_header']) get_data = json.loads(get_resp.data.decode()) assert get_resp.status_code == 200, str(get_resp.response) assert len(get_data) == healthy_count + unhealthy_count, str(get_data)
def test_get_incidents_major_filter(self, test_client, db_session, auth_headers): """Should respect incidents major mine indicator param""" batch_size = 3 small_batch_size = 3 major_mine = MineFactory.create_batch(size=small_batch_size, major_mine_ind=True) MineFactory.create_batch(size=batch_size, major_mine_ind=False) get_resp = test_client.get(f"/incidents?major=t", headers=auth_headers['full_auth_header']) get_data = json.loads(get_resp.data.decode()) assert get_resp.status_code == 200 assert len(get_data['records']) == small_batch_size assert all(map( lambda v: v['mine_name'] in [ mine.mine_name for mine in major_mine], get_data['records']))
def test_get_mine_verified_status_by_user(test_client, db_session, auth_headers): healthy_count = 2 MineFactory.create_batch(size=healthy_count, verified_status__verifying_user='******', verified_status__healthy_ind=True) unhealthy_count = 2 MineFactory.create_batch(size=unhealthy_count, verified_status__verifying_user='******', verified_status__healthy_ind=False) get_resp = test_client.get('/mines/verified-status?user_id=mds', headers=auth_headers['full_auth_header']) get_data = json.loads(get_resp.data.decode()) assert get_resp.status_code == 200, str(get_resp.response) assert len([x for x in get_data if x['healthy_ind'] == True]) == healthy_count, str(get_data) assert len([x for x in get_data if x['healthy_ind'] == False ]) == unhealthy_count, str(get_data) assert all(a['verifying_user'] == 'mds' for a in get_data)