def add_service(self, http_request, service_name):
"""
Adds Service in database
:param http_request: Incoming Request
:param service_name: Name of the Service to be added.
:return: bottle.HTTPResponse
"""
session = self.Session()
success = False
try:
# return number of rows, as primary, should be 1 or 0.
q = session.query(db_model.Service).filter(
db_model.Service.service_name == service_name).first()
try:
if q is None:
session.add(db_model.Service(service_name=service_name))
session.commit()
success = True
except IntegrityError as e:
ret = "ERROR: " + e.detail
print(ret)
raise
except Exception as e1:
ret = "ERROR: " + str(e1.__dict__)
print(ret)
session.rollback()
finally:
session.close()
return add_service_response(http_request, service_name, success)
def test_update_complain(self):
"""
/api/complain/<complain_id>
"""
"""
TEST1 : NEGATIVE : invalid request.
"""
user_obj = db_model.User(
auth_code=None,
phone_number=self.USER_1_PHONE_NUMBER,
handset_serial_number=self.USER_1_HANDSET_SERIAL_NUMBER,
name=None,
email=None)
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj)
response_obj = _test_util.get_response(
requests.put('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] +
'/api/complain/' + str(1),
json=Request.get_request_json(request_obj)))
self.assertTrue(response_obj.return_code == Response.INVALID_REQUEST,
"Invalid request received.")
"""
TEST2 : NEGATIVE : valid request but invalid <complain_id> triggered for update.
"""
# add user.
user_obj = db_model.User(
auth_code=self.USER_1_AUTH_CODE,
phone_number=self.USER_1_PHONE_NUMBER,
handset_serial_number=self.USER_1_HANDSET_SERIAL_NUMBER,
name=self.USER_1_NAME,
email=self.USER_1_EMAIL)
# add a user in database.
_test_util.add_row_in_database(user_obj)
# trigger update request.
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj)
response_obj = _test_util.get_response(
requests.put('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] +
'/api/complain/' + str(100),
json=Request.get_request_json(request_obj)))
self.assertTrue(response_obj.return_code == Response.COMPLAIN_ERROR,
"Invalid request received.")
"""
TEST3 : POSITIVE : valid request.
"""
# add CATEGORY_1 in database
category_obj_for_db = db_model.Category(category_name=self.CATEGORY_1)
_test_util.add_row_in_database(category_obj_for_db)
# add SERVICE_1 in database
service_obj_for_db = db_model.Service(service_name=self.SERVICE_1)
_test_util.add_row_in_database(service_obj_for_db)
# add complain in database
complain_obj = db_model.Complain(
user_id=user_obj.user_id,
category_id=category_obj_for_db.category_id,
service_id=service_obj_for_db.service_id,
address=self.ADDRESS_2,
message=self.MESSAGE_2,
service_status=0,
complain_phone_number=self.USER_2_PHONE_NUMBER)
# add a complain in database.
_test_util.add_row_in_database(complain_obj)
updated_complain_obj = db_model.Complain(
user_id=user_obj.user_id,
category_id=category_obj_for_db.category_id,
service_id=service_obj_for_db.service_id,
address=self.ADDRESS_1,
message=self.MESSAGE_1,
service_status=2,
complain_phone_number=self.USER_1_PHONE_NUMBER)
# trigger update request.
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj,
complain=updated_complain_obj)
response_obj = _test_util.get_response(
requests.put('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] +
'/api/complain/' + str(complain_obj.complain_id),
json=Request.get_request_json(request_obj)))
self.assertTrue(response_obj.return_code == Response.COMPLAIN_UPDATED,
"Valid request received.")
session = self.Session()
try:
complain_from_database = session.query(db_model.Complain).filter(
db_model.Complain.user_id == user_obj.user_id).one()
self.assertTrue(True, "Query database.")
session.commit()
except sqlalchemy.orm.exc.NoResultFound:
session.rollback()
self.fail("No row found")
self.assertTrue(
response_obj.complain == [
AlchemyEncoder.get_pickle_dict(complain_from_database)
], "Valid request received.")
def test_get_all_services(self):
"""
/api/services
"""
"""
TEST1 : NEGATIVE : Invalid request due to unknown user
"""
user_obj = db_model.User(
auth_code=None,
phone_number=self.USER_1_PHONE_NUMBER,
handset_serial_number=self.USER_1_HANDSET_SERIAL_NUMBER,
name=None,
email=None)
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj)
response_obj = _test_util.get_response(
requests.get('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] + '/api/services',
json=Request.get_request_json(request_obj)))
self.assertTrue(response_obj.return_code == Response.INVALID_REQUEST,
"Invalid request received.")
"""
TEST2 : POSITIVE : no service in database.
"""
# get all services via api call
user_obj = db_model.User(
auth_code=self.USER_1_AUTH_CODE,
phone_number=self.USER_1_PHONE_NUMBER,
handset_serial_number=self.USER_1_HANDSET_SERIAL_NUMBER,
name=self.USER_1_NAME,
email=self.USER_1_EMAIL)
# add user object in database. We must have a valid user in database to get any business info from database.
_test_util.add_row_in_database(user_obj)
# get service list (in this case empty), via api call
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj)
response_obj = _test_util.get_response(
requests.get('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] + '/api/services',
json=Request.get_request_json(request_obj)))
self.assertTrue(
response_obj.return_code == Response.SERVICE_LIST_EMPTY,
"Category list empty")
"""
TEST3 : POSITIVE : get all services in database.
"""
# add SERVICE_1
service_obj_for_db = db_model.Service(service_name=self.SERVICE_1)
_test_util.add_row_in_database(service_obj_for_db)
# add SERVICE_2
service_obj_for_db = db_model.Service(service_name=self.SERVICE_2)
_test_util.add_row_in_database(service_obj_for_db)
ser_list = [self.SERVICE_1, self.SERVICE_2]
# get all services via api call
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj)
response_obj = _test_util.get_response(
requests.get('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] + '/api/services',
json=Request.get_request_json(request_obj)))
self.assertTrue(response_obj.return_code == Response.ALL_SERVICES,
"Return code for service list in database")
self.assertTrue(
len([
x for x in response_obj.service
if x.get('service_name') in ser_list
]) == 2, "Validate returned list")
def test_get_complains_for_user(self):
"""
/api/complain/<user_id>
"""
"""
TEST1: NEGATIVE : Invalid request due to unknown user in request json.
"""
user_obj = db_model.User(
auth_code=None,
phone_number=self.USER_1_PHONE_NUMBER,
handset_serial_number=self.USER_1_HANDSET_SERIAL_NUMBER,
name=None,
email=None)
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj)
response_obj = _test_util.get_response(
requests.get('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] +
'/api/complain/' + str(1),
json=Request.get_request_json(request_obj)))
self.assertTrue(response_obj.return_code == Response.INVALID_REQUEST,
"Invalid request received.")
"""
TEST2: NEGATIVE : Invalid request due to non-matching user in request json and api
"""
user_obj_1 = db_model.User(
auth_code=self.USER_1_AUTH_CODE,
phone_number=self.USER_1_PHONE_NUMBER,
handset_serial_number=self.USER_1_HANDSET_SERIAL_NUMBER,
name=self.USER_1_NAME,
email=self.USER_1_EMAIL)
# add a user in database.
_test_util.add_row_in_database(user_obj_1)
# get the user id from database.
session = self.Session()
try:
user_from_database = session.query(db_model.User).filter(
db_model.User.phone_number == self.USER_1_PHONE_NUMBER and \
db_model.User.handset_serial_number == self.USER_1_HANDSET_SERIAL_NUMBER).one()
session.commit()
self.assertTrue(True, "User from database")
except sqlalchemy.orm.exc.NoResultFound:
session.rollback()
self.fail("Intended user not found")
except sqlalchemy.orm.exc.MultipleResultsFound:
session.rollback()
self.fail(
"More than 1 phone number and serial number combination found in database"
)
invalid_user_id = user_from_database.user_id + 1
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj_1)
response_obj = _test_util.get_response(
requests.get('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] +
'/api/complain/' + str(invalid_user_id),
json=Request.get_request_json(request_obj)))
self.assertTrue(response_obj.return_code == Response.INVALID_REQUEST,
"Invalid request received.")
"""
TEST3: POSITIVE : No complains registered for querying user.
"""
valid_user_id = user_from_database.user_id
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj_1)
response_obj = _test_util.get_response(
requests.get('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] +
'/api/complain/' + str(valid_user_id),
json=Request.get_request_json(request_obj)))
self.assertTrue(
response_obj.return_code == Response.COMPLAIN_LIST_EMPTY,
"No complains for this user.")
"""
TEST4: POSITIVE : List of complains registered for querying user.
"""
# add CATEGORY_1
category_obj_for_db = db_model.Category(category_name=self.CATEGORY_1)
_test_util.add_row_in_database(category_obj_for_db)
# add CATEGORY_2
category_obj_for_db = db_model.Category(category_name=self.CATEGORY_2)
_test_util.add_row_in_database(category_obj_for_db)
# add SERVICE_1
service_obj_for_db = db_model.Service(service_name=self.SERVICE_1)
_test_util.add_row_in_database(service_obj_for_db)
# add SERVICE_2
service_obj_for_db = db_model.Service(service_name=self.SERVICE_2)
_test_util.add_row_in_database(service_obj_for_db)
# add user 2
user_obj_2 = db_model.User(
auth_code=self.USER_2_AUTH_CODE,
phone_number=self.USER_2_PHONE_NUMBER,
handset_serial_number=self.USER_2_HANDSET_SERIAL_NUMBER,
name=self.USER_2_NAME,
email=self.USER_2_EMAIL)
# add a user in database.
_test_util.add_row_in_database(user_obj_2)
# get all ids from database , required for creating new complains.
session = self.Session()
try:
category_1 = session.query(db_model.Category).filter(
db_model.Category.category_name == self.CATEGORY_1).one()
category_2 = session.query(db_model.Category).filter(
db_model.Category.category_name == self.CATEGORY_2).one()
service_1 = session.query(db_model.Service).filter(
db_model.Service.service_name == self.SERVICE_1).one()
service_2 = session.query(db_model.Service).filter(
db_model.Service.service_name == self.SERVICE_2).one()
user_1 = session.query(db_model.User).filter(
db_model.User.phone_number == self.USER_1_PHONE_NUMBER and \
db_model.User.handset_serial_number == self.USER_1_HANDSET_SERIAL_NUMBER).one()
user_2 = session.query(db_model.User).filter(
db_model.User.phone_number == self.USER_2_PHONE_NUMBER and \
db_model.User.handset_serial_number == self.USER_2_HANDSET_SERIAL_NUMBER).one()
self.assertTrue(True, "Query database.")
session.rollback()
except sqlalchemy.orm.exc.NoResultFound:
session.rollback()
self.fail("No row found")
# add complains via Complain api.
complain_1 = db_model.Complain(
user_id=user_1.user_id,
category_id=category_1.category_id,
service_id=service_1.service_id,
address=self.ADDRESS_1,
message=self.MESSAGE_1,
service_status=0,
complain_phone_number=self.USER_2_PHONE_NUMBER)
complain_2 = db_model.Complain(
user_id=user_1.user_id,
category_id=category_2.category_id,
service_id=service_2.service_id,
address=self.ADDRESS_2,
message=self.MESSAGE_2,
service_status=0,
complain_phone_number=self.USER_2_PHONE_NUMBER)
complain_3 = db_model.Complain(
user_id=user_2.user_id,
category_id=category_1.category_id,
service_id=service_1.service_id,
address=self.ADDRESS_1,
message=self.MESSAGE_1,
service_status=0,
complain_phone_number=self.USER_2_PHONE_NUMBER)
complain_4 = db_model.Complain(
user_id=user_2.user_id,
category_id=category_2.category_id,
service_id=service_2.service_id,
address=self.ADDRESS_2,
message=self.MESSAGE_2,
service_status=0,
complain_phone_number=self.USER_2_PHONE_NUMBER)
_test_util.add_row_in_database(complain_1)
_test_util.add_row_in_database(complain_2)
_test_util.add_row_in_database(complain_3)
_test_util.add_row_in_database(complain_4)
# request for user_1 complains
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj_1)
response_obj = _test_util.get_response(
requests.get('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] +
'/api/complain/' + str(user_obj_1.user_id),
json=Request.get_request_json(request_obj)))
self.assertTrue(response_obj.return_code == Response.ALL_COMPLAINS,
"All complains for this user.")
self.assertTrue(
response_obj.complain == [
AlchemyEncoder.get_pickle_dict(complain_1),
AlchemyEncoder.get_pickle_dict(complain_2)
], "All complains for this user.")
# request for user_2 complains
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj_2)
response_obj = _test_util.get_response(
requests.get('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] +
'/api/complain/' + str(user_obj_2.user_id),
json=Request.get_request_json(request_obj)))
self.assertTrue(response_obj.return_code == Response.ALL_COMPLAINS,
"All complains for this user.")
self.assertTrue(
response_obj.complain == [
AlchemyEncoder.get_pickle_dict(complain_3),
AlchemyEncoder.get_pickle_dict(complain_4)
], "All complains for this user.")
def test_add_complain(self):
"""
/api/complains
"""
"""
TEST1 : NEGATIVE : Invalid request due to unknown user
"""
user_obj = db_model.User(
auth_code=None,
phone_number=self.USER_1_PHONE_NUMBER,
handset_serial_number=self.USER_1_HANDSET_SERIAL_NUMBER,
name=None,
email=None)
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj)
response_obj = _test_util.get_response(
requests.post('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] +
'/api/complains',
json=Request.get_request_json(request_obj)))
self.assertTrue(response_obj.return_code == Response.INVALID_REQUEST,
"Invalid request received.")
"""
TEST2 : NEGATIVE : Empty complain object received.
"""
# add a user in database.
user_obj = db_model.User(
auth_code=self.USER_1_AUTH_CODE,
phone_number=self.USER_1_PHONE_NUMBER,
handset_serial_number=self.USER_1_HANDSET_SERIAL_NUMBER,
name=self.USER_1_NAME,
email=self.USER_1_EMAIL)
# add user in database. We must have a valid user in database to get any business info from database.
_test_util.add_row_in_database(user_obj)
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj)
response_obj = _test_util.get_response(
requests.post('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] +
'/api/complains',
json=Request.get_request_json(request_obj)))
self.assertTrue(response_obj.return_code == Response.EMPTY_COMPLAIN,
"Empty complain received.")
"""
TEST3 : NEGATIVE : Unable to add complain due to invalid category id or/and service id.
"""
# get the user id from database.
session = self.Session()
try:
user_from_database = session.query(db_model.User).filter(
db_model.User.phone_number == self.USER_1_PHONE_NUMBER and \
db_model.User.handset_serial_number == self.USER_1_HANDSET_SERIAL_NUMBER).one()
session.commit()
self.assertTrue(True, "User from database")
except sqlalchemy.orm.exc.NoResultFound:
session.rollback()
self.fail("Intended user not found")
except sqlalchemy.orm.exc.MultipleResultsFound:
session.rollback()
self.fail(
"More than 1 phone number and serial number combination found in database"
)
complain_obj = db_model.Complain(user_id=user_from_database.user_id,
category_id=1,
service_id=4,
address='Panchkula',
message='FixShit',
service_status=0,
complain_phone_number=9415549481)
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj,
complain=complain_obj)
response_obj = _test_util.get_response(
requests.post('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] +
'/api/complains',
json=Request.get_request_json(request_obj)))
# print(response_obj.return_code)
self.assertTrue(response_obj.return_code == Response.COMPLAIN_ERROR)
"""
TEST4 : POSITIVE : Successful addition of complain.
"""
# add service in database.
service_obj = db_model.Service(service_name=self.SERVICE_1)
_test_util.add_row_in_database(service_obj)
# add category in database.
_test_util.add_row_in_database(
db_model.Category(category_name=self.CATEGORY_1))
# get the user id from database.
session = self.Session()
try:
service_from_database = session.query(db_model.Service).filter(
db_model.Service.service_name == self.SERVICE_1).one()
self.assertTrue(True, "Service from database")
category_from_database = session.query(db_model.Category).filter(
db_model.Category.category_name == self.CATEGORY_1).one()
self.assertTrue(True, "Category from database")
session.commit()
except sqlalchemy.orm.exc.NoResultFound:
session.rollback()
self.fail("Intended row not found")
except sqlalchemy.orm.exc.MultipleResultsFound:
session.rollback()
self.fail("More than one row found in database")
complain_obj = db_model.Complain(
user_id=user_from_database.user_id,
category_id=category_from_database.category_id,
service_id=service_from_database.service_id,
address='Panchkula',
message='FixShit',
service_status=0,
complain_phone_number=9415549481)
request_obj = Request(timestamp=_test_util.get_current_time(),
user=user_obj,
complain=complain_obj)
response_obj = _test_util.get_response(
requests.post('http://' + get_server_config()['server_ip'] + ':' +
get_server_config()['server_port'] +
'/api/complains',
json=Request.get_request_json(request_obj)))
self.assertTrue(response_obj.return_code == Response.COMPLAIN_ADDED)
# check if returned complain id is added in server database.
session = self.Session()
try:
session.query(db_model.Complain).filter(
db_model.Complain.complain_id == response_obj.complain[0]
['complain_id']).one()
self.assertTrue(True, "Query database.")
session.commit()
except sqlalchemy.orm.exc.NoResultFound:
session.rollback()
self.fail("No row found")