def test_exists(db, session): # pylint: disable=unused-argument """Verify that the exists function works correctly.""" # if there is no imei in the table imei_norm = '64728204390652' model_exists = ApprovedImeis.exists(imei_norm) query_exists = session.execute(text("""SELECT EXISTS( SELECT 1 FROM public.approvedimeis WHERE imei='{0}' AND removed IS NOT TRUE )""".format(imei_norm))).fetchone() assert query_exists[0] == model_exists # add a record and than check imei_norm = '64728204390652' request_id = 237654998 status = 'status' delta_status = 'delta status' approved_imei = ApprovedImeis(imei_norm, request_id, status, delta_status) approved_imei.add() assert ApprovedImeis.exists(imei_norm) # set existing record's removed=True than check res = session.execute(text("""UPDATE public.approvedimeis SET removed=True WHERE imei='{0}' AND request_id='{1}'""".format(imei_norm, request_id))) assert res assert not ApprovedImeis.exists(imei_norm)
def test_bulk_insert_imeis(db, session): # pylint: disable=unused-argument """Verify that the bulk_insert_imeis() works as expected.""" imei_norm1 = '67895678901234' imei_norm2 = '76545678906548' imei_norm3 = '54375699900000' request_id = 2376322 status = 'status' delta_status = 'delta status' imeis = [ ApprovedImeis(imei_norm1, request_id, status, delta_status), ApprovedImeis(imei_norm2, request_id, status, delta_status), ApprovedImeis(imei_norm3, request_id, status, delta_status) ] ApprovedImeis.bulk_insert_imeis(imeis) assert ApprovedImeis.exists(imei_norm1) assert ApprovedImeis.exists(imei_norm2) assert ApprovedImeis.exists(imei_norm3)