diff --git a/tests/test_awslambda/test_lambda.py b/tests/test_awslambda/test_lambda.py index 34fd9c9b3..479aaaa8a 100644 --- a/tests/test_awslambda/test_lambda.py +++ b/tests/test_awslambda/test_lambda.py @@ -4,7 +4,6 @@ import base64 import botocore.client import boto3 import hashlib -from http import HTTPStatus import io import json import time @@ -473,7 +472,7 @@ def test_publish(): latest_arn = function_list['Functions'][0]['FunctionArn'] res = conn.publish_version(FunctionName='testFunction') - assert res['ResponseMetadata']['HTTPStatusCode'] == HTTPStatus.CREATED + assert res['ResponseMetadata']['HTTPStatusCode'] == 201 function_list = conn.list_functions() function_list['Functions'].should.have.length_of(2) @@ -856,7 +855,7 @@ def test_list_versions_by_function(): ) res = conn.publish_version(FunctionName='testFunction') - assert res['ResponseMetadata']['HTTPStatusCode'] == HTTPStatus.CREATED + assert res['ResponseMetadata']['HTTPStatusCode'] == 201 versions = conn.list_versions_by_function(FunctionName='testFunction') assert versions['Versions'][0]['FunctionArn'] == 'arn:aws:lambda:us-west-2:123456789012:function:testFunction:$LATEST'