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)
