def test_get_imei(db, session): # pylint: disable=unused-argument
"""Verify that the get_imei() return the same imei which is not removed."""
imei_norm = '23456421264573'
request_id = 237654998
status = 'status'
delta_status = 'delta status'
approved_imei = ApprovedImeis(imei_norm, request_id, status, delta_status)
approved_imei.add()
# get imei
imei_data = ApprovedImeis.get_imei(imei_norm)
assert imei_data.imei == imei_norm
assert imei_data.request_id == request_id
assert imei_data.status == status
assert imei_data.delta_status == delta_status
assert not imei_data.removed
# update removed=True and check again
res = session.execute(
text("""UPDATE public.approvedimeis
SET removed=True
WHERE imei='{0}' AND request_id='{1}'""".
format(imei_norm, request_id)))
assert res
imei_data = ApprovedImeis.get_imei(imei_norm)
assert not imei_data
def test_registration_request_rejected_section(flask_app, db): # pylint: disable=unused-argument
"""Verify that the registration request gets rejected when anyone of the section
is marked as rejected, imeis are removed from approvedimeis and notification is
being generated.
"""
# only one section is reviewed and rejected
headers = {'Content-Type': 'application/json'}
data = {
'device_count': 1,
'imei_per_device': 1,
'imeis': "[['23010403010533']]",
'm_location': 'local',
'user_name': '23423423rev user 1',
'user_id': 'assign-rev23442342-user-1'
}
request = create_assigned_dummy_request(data, 'Registration', 'rev230987',
'rev 230987')
assert request
request_id = request.id
device_data = {
'brand': 'samsung',
'operating_system': 'android',
'model_name': 's9',
'model_num': '30jjd',
'device_type': 'Smartphone',
'technologies': '2G,3G,4G',
'reg_id': request.id
}
request = create_dummy_devices(device_data, 'Registration', request)
assert request
# add one section with rejected status
RegComments.add('device_quota', 'test comment on section', 'rev230987',
'rev 230987', 7, request_id)
body_data = {
'request_id': request_id,
'request_type': 'registration_request',
'reviewer_id': 'rev230987'
}
rv = flask_app.put(SUBMIT_REVIEW_API,
data=json.dumps(body_data),
headers=headers)
assert rv.status_code == 201
response = json.loads(rv.data.decode('utf-8'))
assert response['status'] == 7
assert response['request_id'] == request_id
assert response['message'] == 'case {0} updated successfully'.format(
request_id)
assert response['request_type'] == 'registration_request'
imei = ApprovedImeis.get_imei('23010403010533')
assert imei.status == 'removed'
assert imei.delta_status == 'remove'
assert Notification.exist_users('assign-rev23442342-user-1')
def de_register_imeis(imeis):
"""Method to De-Register imeis.
"""
# find imei devices and related imeis
if len(imeis) == 1:
query = """SELECT normalized_imei
FROM public.imeidevice
WHERE device_id IN
(SELECT device_id
FROM public.imeidevice
WHERE normalized_imei='{imei}')""".format(
imei=imeis[0])
else:
query = """SELECT normalized_imei
FROM public.imeidevice
WHERE device_id IN
(SELECT device_id
FROM public.imeidevice
WHERE normalized_imei IN {imeis})""".format(
imeis=tuple(imeis))
try:
res = db.engine.execute(query).fetchall()
matched_imeis = list(chain.from_iterable(res))
# de register imeis from approved imeis
updated_imeis = []
for imei in matched_imeis:
imei_ = ApprovedImeis.get_imei(imei)
if imei_.status != 'removed':
imei_.status = 'removed'
imei_.delta_status = 'remove'
updated_imeis.append(imei_)
if len(updated_imeis) > 0:
ApprovedImeis.bulk_insert_imeis(updated_imeis)
return True
else:
return False
except Exception as e:
app.logger.error(
'An exception occurred while De-Registering IMEIs see exception log:'
)
app.logger.exception(e)
return None
def test_de_registration_request_approval(flask_app, db, app):
"""Verify that the system approves the request when all the sections are approved
and de-register imeis successfully.
"""
headers = {'Content-Type': 'application/json'}
data = {
'device_count': 1,
'imei_per_device': 1,
'imeis': "[['121621090005119']]",
'm_location': 'local',
'user_name': 'reg req approve rev user 9',
'user_id': 'reg-req-approve-rev-user-9'
}
request = create_assigned_dummy_request(data, 'Registration',
'dereg-req-approve-rev-user-10', 'dereg req approve rev user 10')
assert request
request_id = request.id
user_id = request.user_id
device_data = {
'brand': 'samsung',
'operating_system': 'android',
'model_name': 's9',
'model_num': '30jjd',
'device_type': 'Smartphone',
'technologies': '2G,3G,4G',
'reg_id': request_id
}
request = create_dummy_devices(device_data, 'Registration', request)
assert request
# create user quota
quota = DeviceQuota.create(user_id, 'individual')
assert quota
de_registration_data = {
'file': 'de-reg-test-file.txt',
'device_count': 1,
'user_id': 'dereg-req-approve-rev-user-10',
'user_name': 'dereg req approve rev user 10',
'reason': 'because we have to run tests successfully'
}
request = create_assigned_dummy_request(de_registration_data, 'De_Registration', 'dereg-req-approve-rev-user-10',
'dereg req approve rev user 10')
device_data = {
'devices': """[
{
"tac": "12162109",
"model_name": "TA-1034",
"brand_name": "NOKIA",
"model_num": "TA-1034",
"technology": "NONE",
"device_type": "Mobile Phone/Feature phone",
"count": 2,
"operating_system": "N/A"
}
]""",
'dereg_id': request.id
}
request = create_dummy_devices(device_data, 'De_Registration', request, db, file_content=['121621090005119'],
file_path='{0}/{1}/{2}'.format(app.config['DRS_UPLOADS'], request.tracking_id,
de_registration_data.get('file')))
assert request
request_id = request.id
# add imei to approvedimeis as whitelist to de register
approved_imei = ApprovedImeis.get_imei('12162109000511')
approved_imei.status = 'whitelist'
approved_imei.delta_status = 'update'
ApprovedImeis.bulk_insert_imeis([approved_imei])
DeRegComments.add('device_description', 'test comment on section', 'dereg-req-approve-rev-user-10',
'dereg req approve rev user 10', 6, request_id)
DeRegComments.add('imei_classification', 'test comment on section', 'dereg-req-approve-rev-user-10',
'dereg req approve rev user 10', 6, request_id)
DeRegComments.add('imei_registration', 'test comment on section', 'dereg-req-approve-rev-user-10',
'dereg req approve rev user 10', 6, request_id)
DeRegComments.add('approval_documents', 'test comment on section', 'dereg-req-approve-rev-user-10',
'dereg req approve rev user 10', 6, request_id)
body_data = {
'request_id': request_id,
'request_type': 'de_registration_request',
'reviewer_id': 'dereg-req-approve-rev-user-10'
}
rv = flask_app.put(SUBMIT_REVIEW_API, data=json.dumps(body_data), headers=headers)
assert rv.status_code == 201
response = json.loads(rv.data.decode('utf-8'))
imei = ApprovedImeis.get_imei('12162109000511')
assert response['request_type'] == 'de_registration_request'
assert response['status'] == 6
assert response['request_id'] == request_id
assert imei.status == 'removed'
def test_registration_request_approval(flask_app, db): # pylint: disable=unused-argument
"""Verify that the api behave properly upon approval of the request."""
# only one section is reviewed and rejected
headers = {'Content-Type': 'application/json'}
data = {
'device_count': 1,
'imei_per_device': 1,
'imeis': "[['94310813016000']]",
'm_location': 'local',
'user_name': 'reg req approve rev user 5',
'user_id': 'reg-req-approve-rev-user-5'
}
request = create_assigned_dummy_request(data, 'Registration',
'reg-req-approve-rev-user-5', 'reg req approve rev user 5')
assert request
request_id = request.id
user_id = request.user_id
device_data = {
'brand': 'samsung',
'operating_system': 'android',
'model_name': 's9',
'model_num': '30jjd',
'device_type': 'Smartphone',
'technologies': '2G,3G,4G',
'reg_id': request.id
}
request = create_dummy_devices(device_data, 'Registration', request)
assert request
# create user quota
quota = DeviceQuota.create(user_id, 'individual')
assert quota
user_reg_quota = quota.reg_quota
# approve all sections manually
RegComments.add('device_quota', 'test comment on section', 'reg-req-approve-rev-user-5',
'reg req approve rev user 5', 6, request_id)
RegComments.add('device_description', 'test comment on section', 'reg-req-approve-rev-user-5',
'reg req approve rev user 5', 6, request_id)
RegComments.add('imei_classification', 'test comment on section', 'reg-req-approve-rev-user-5',
'reg req approve rev user 5', 6, request_id)
RegComments.add('imei_registration', 'test comment on section', 'reg-req-approve-rev-user-5',
'reg req approve rev user 5', 6, request_id)
RegComments.add('approval_documents', 'test comment on section', 'reg-req-approve-rev-user-5',
'reg req approve rev user 5', 6, request_id)
body_data = {
'request_id': request_id,
'request_type': 'registration_request',
'reviewer_id': 'reg-req-approve-rev-user-5'
}
rv = flask_app.put(SUBMIT_REVIEW_API, data=json.dumps(body_data), headers=headers)
assert rv.status_code == 201
data = json.loads(rv.data.decode('utf-8'))
assert data['status'] == 6
assert data['request_id'] == request_id
assert data['message'] == 'case {0} updated successfully'.format(request_id)
assert data['request_type'] == 'registration_request'
imei = ApprovedImeis.get_imei('94310813016000')
assert imei.status == 'whitelist'
assert imei.delta_status == 'add'
assert Notification.exist_users(user_id)
assert DeviceQuota.get(user_id).reg_quota == user_reg_quota - 1