def post(self, item_id):
min_value = None
max_value = None
if 'external_item_id' in request.form.keys():
external_item_id = request.form['external_item_id']
consumption_type = request.form['consumption_type']
consumption_amount = int(request.form['consumption_amount'])
address = request.form['address']
unit = request.form['unit']
if 'min_value' in request.form.keys():
min_value = request.form['min_value']
if 'max_value' in request.form.keys():
max_value = request.form['max_value']
else:
request_data = json.loads(request.data)
external_item_id = request_data['external_item_id']
consumption_type = request_data['consumption_type']
consumption_amount = int(request_data['consumption_amount'])
address = request_data['address']
unit = request_data['unit']
if 'min_value' in request_data.keys():
min_value = request_data['min_value']
if 'max_value' in request_data.keys():
max_value = request_data['max_value']
errors = validate(item_id=item_id,
usage_consumption_type=consumption_type,
usage_consumption_amount=consumption_amount,
usage_address=address,
usage_unit=unit,
usage_min_value=min_value,
usage_max_value=max_value)
if len(errors) > 0:
return {"errors": [error.to_json() for error in errors]}, 422
consumption_type = UsageTypeEnum(consumption_type)
unit = UnitEnum(unit)
usage = UsageModel(item_id, external_item_id, consumption_type,
consumption_amount, address, unit, min_value,
max_value)
usage.save_to_db()
return usage.to_json(), 201
def test_usage_resource():
print("#################### TESTING USAGE RESOURCE ####################")
# GETTING ALL USAGES
print("TEST_1 --- GETTING ALL USAGES")
uri = "usages"
expected_result = {
"usages": []
}
expected_status = 200
test_get(uri, expected_result, expected_status)
# POSTING ONE USAGE
print("TEST_2 --- POSTING ONE USAGE")
item_1 = send_get('items/1')
usage_1_item_id = item_1['id']
usage_1_external_item_id = 1
usage_1_consumption_type = UsageTypeEnum.KILOWATT
usage_1_consumption_amount = 5
usage_1_address = "http://127.0.0.1:5000/usages/1"
usage_1_unit = UnitEnum.TOGGLE
usage_1_min_value = 0
usage_1_max_value = 1
usage_1 = UsageModel(usage_1_item_id,
usage_1_external_item_id,
usage_1_consumption_type,
usage_1_consumption_amount,
usage_1_address,
usage_1_unit,
usage_1_min_value,
usage_1_max_value)
body = {
"item_id": usage_1_item_id,
"external_item_id": usage_1_external_item_id,
"consumption_type": usage_1_consumption_type.value,
"consumption_amount": usage_1_consumption_amount,
"address": usage_1_address,
"unit": "TOGGLE",
"min_value": usage_1_min_value,
"max_value": usage_1_max_value
}
usage_1_json = usage_1.to_json()
usage_1_json['id'] = 1
usage_1_json['url'] = "127.0.0.1:5000/api/v1/usages/1"
expected_result = usage_1_json
expected_status = 201
uri = "usages"
test_post(uri, body, expected_result, expected_status)
# GETTING ALL USAGES
print("TEST_3 --- GETTING ALL USAGES")
uri = "usages"
expected_result = {
"usages": [usage_1_json]
}
expected_status = 200
test_get(uri, expected_result, expected_status)
# GETTING ONE USAGE
print("TEST_4 --- GETTING ONE USAGE")
uri = "usages/1"
expected_result = usage_1_json
expected_status = 200
test_get(uri, expected_result, expected_status)
# UPDATING ONE USAGE
print("TEST_5 --- UPDATING ONE USAGE")
uri = 'usages/1'
expected_result = usage_1_json
expected_result['address'] = '127.0.0.1:5000/api/usages/7'
body = expected_result
expected_status = 200
test_put(uri, body, expected_result, expected_status)
usage_1_json['address'] = '127.0.0.1:5000/api/usages/7'
# GETTING ONE USAGE
print("TEST_6 --- GETTING ONE USAGE")
uri = "usages/1"
expected_result = usage_1_json
expected_status = 200
test_get(uri, expected_result, expected_status)
# POSTING ONE USAGE - BAD REQUEST
print("TEST_7 --- POSTING ONE USAGE - BAD REQUEST")
item_1 = send_get('items/1')
usage_1_item_id = item_1['id']
usage_1_external_item_id = item_1['id']
usage_1_consumption_type = UsageTypeEnum.KILOWATT
usage_1_consumption_amount = 5
usage_1_address = "http://127.0.0.1:5000/usage/1"
usage_1_min_value = 0
usage_1_max_value = 1
body = {
"item_id": usage_1_item_id,
"external_item_id": usage_1_external_item_id,
"consumption_type": usage_1_consumption_type.value,
"consumption_amount": usage_1_consumption_amount,
"address": usage_1_address,
"unit": "FOGGLE",
"min_value": usage_1_min_value,
"max_value": usage_1_max_value
}
expected_result = {"errors": [Error(
"{} is not a valid unit option.".format("FOGGLE"),
"UnitEnum.has_value({}) returned False".format("FOGGLE"),
422,
"https://en.wikipedia.org/wiki/HTTP_422").to_json()]}
expected_status = 422
uri = "usages"
test_post(uri, body, expected_result, expected_status)
# POSTING ONE USAGE - BAD REQUEST
print("TEST_8 --- POSTING ONE USAGE - BAD REQUEST")
body['unit'] = usage_1_unit.value
body['consumption_type'] = 'KILO WHAT?'
expected_result = {"errors": [Error(
"{} is not a valid consumption type.".format("KILO WHAT?"),
"UsageTypeEnum.has_value({}) returned False".format("KILO WHAT?"),
422,
"https://en.wikipedia.org/wiki/HTTP_422").to_json()]}
expected_status = 422
uri = "usages"
test_post(uri, body, expected_result, expected_status)
# POSTING ONE USAGE - BAD REQUEST
print("TEST_9 --- POSTING ONE USAGE - BAD REQUEST")
body['consumption_type'] = usage_1_consumption_type.value
body['address'] = '12'
expected_result = {"errors": [Error(
"Address must be at least 4 characters long.",
"address was {} characters long".format(len(body['address'])),
422,
"https://en.wikipedia.org/wiki/HTTP_422").to_json()]}
expected_status = 422
uri = "usages"
test_post(uri, body, expected_result, expected_status)
# POSTING ONE USAGE - BAD REQUEST
print("TEST_10 --- POSTING ONE USAGE - BAD REQUEST")
body['address'] = "________________________________________________________________________________________" \
"________________________________________________________________________________________" \
"________________________________________________________________________________________" \
"________________________________________________________________________________________" \
"________________________________________________________________________________________" \
"________________________________________________________________________________________" \
"________________________________________________________________________________________"
expected_result = {"errors": [Error(
"Address cannot be longer than 255 characters.",
"address was {} characters long".format(len(body['address'])),
422,
"https://en.wikipedia.org/wiki/HTTP_422").to_json()]}
expected_status = 422
uri = "usages"
test_post(uri, body, expected_result, expected_status)
# POSTING ONE USAGE - BAD REQUEST
print("TEST_11 --- POSTING ONE USAGE - BAD REQUEST")
body['address'] = usage_1_address
body['consumption_amount'] = -1
expected_result = {"errors": [Error(
"Consumption amount cannot be below 0.",
"{} is below 0".format(-1),
422,
"https://en.wikipedia.org/wiki/HTTP_422").to_json()]}
expected_status = 422
uri = "usages"
test_post(uri, body, expected_result, expected_status)
# POSTING ONE USAGE - BAD REQUEST
print("TEST_12 --- POSTING ONE USAGE - BAD REQUEST")
body['consumption_amount'] = usage_1_consumption_amount
body['min_value'] = None
expected_result = {"errors": [Error(
"If either min or max value is given, both should be given.",
"Either min or max was None while the other was not.",
422,
"https://en.wikipedia.org/wiki/HTTP_422").to_json()]}
expected_status = 422
uri = "usages"
test_post(uri, body, expected_result, expected_status)
# POSTING ONE USAGE - BAD REQUEST
print("TEST_13 --- POSTING ONE USAGE - BAD REQUEST")
body['min_value'] = usage_1_min_value
body['max_value'] = None
expected_result = {"errors": [Error(
"If either min or max value is given, both should be given.",
"Either min or max was None while the other was not.",
422,
"https://en.wikipedia.org/wiki/HTTP_422").to_json()]}
expected_status = 422
uri = "usages"
test_post(uri, body, expected_result, expected_status)
# EXECUTING COMMAND - BAD REQUEST
print("TEST 14 --- POSTING COMMAND - BAD REQUEST")
body = {}
uri = "usages/1/command/2"
expected_result = {"errors": Error(
"New value does not fall within the expected range. ({} - {})"
.format(usage_1_json["min_value"], usage_1_json['max_value']),
"{} is outside of ({} - {})".format(2, usage_1_json["min_value"], usage_1_json['max_value']),
422,
""
).to_json()}
expected_status = 422
test_patch(uri, body, expected_result, expected_status)
# POSTING ONE USAGE
print("TEST_15 --- POSTING ONE USAGE")
item_2 = send_get('items/2')
usage_2_item_id = item_2['id']
usage_2_external_item_id = 1
usage_2_consumption_type = UsageTypeEnum.KILOWATT
usage_2_consumption_amount = 5
usage_2_address = "http://127.0.0.1:5000/usages/2"
usage_2_unit = UnitEnum.TOGGLE
usage_2_min_value = 0
usage_2_max_value = 1
usage_2 = UsageModel(usage_2_item_id,
usage_2_external_item_id,
usage_2_consumption_type,
usage_2_consumption_amount,
usage_2_address,
usage_2_unit,
usage_2_min_value,
usage_2_max_value)
body = {
"item_id": usage_2_item_id,
"external_item_id": usage_2_external_item_id,
"consumption_type": usage_2_consumption_type.value,
"consumption_amount": usage_2_consumption_amount,
"address": usage_2_address,
"unit": "TOGGLE",
"min_value": usage_2_min_value,
"max_value": usage_2_max_value
}
usage_2_json = usage_2.to_json()
usage_2_json['id'] = 2
usage_2_json['url'] = "127.0.0.1:5000/api/v1/usages/2"
expected_result = usage_2_json
expected_status = 201
uri = "usages"
test_post(uri, body, expected_result, expected_status)