Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

move to httpretty fork instead of bundling it.

Browse Source
This commit is contained in:
Steve Pulec 2013-03-17 18:13:21 -04:00
parent 5bec85c6c0
commit cd518d060e
7 changed files with 36 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
moto/core/models.py
View File

@ -1,7 +1,7 @@
import functools
import re
from moto.packages.httpretty import HTTPretty
from httpretty import HTTPretty
from .responses import metadata_response
from .utils import convert_regex_to_flask_path

8
moto/core/responses.py
View File

@ -7,10 +7,10 @@ from moto.core.utils import headers_to_dict, camelcase_to_underscores, method_na
class BaseResponse(object):
def dispatch2(self, uri, body, headers):
return self.dispatch(uri, body, headers)
def dispatch2(self, uri, method, body, headers):
return self.dispatch(uri, method, body, headers)
def dispatch(self, uri, body, headers):
def dispatch(self, uri, method, body, headers):
if body:
querystring = parse_qs(body)
else:
@ -29,7 +29,7 @@ class BaseResponse(object):
raise NotImplementedError("The {} action has not been implemented".format(action))
def metadata_response(uri, body, headers):
def metadata_response(uri, method, body, headers):
"""
Mock response for localhost metadata

2
moto/core/utils.py
View File

@ -102,7 +102,7 @@ class convert_flask_to_httpretty_response(object):
body = request.data or query
headers = dict(request.headers)
result = self.callback(uri, body, headers)
result = self.callback(uri, method, body, headers)
if isinstance(result, basestring):
# result is just the response
return result

45
moto/dynamodb/responses.py
View File

@ -27,18 +27,19 @@ GET_SESSION_TOKEN_RESULT = """
</GetSessionTokenResponse>"""
def sts_handler(uri, body, headers):
def sts_handler(uri, method, body, headers):
return GET_SESSION_TOKEN_RESULT
class DynamoHandler(object):
def __init__(self, uri, body, headers):
def __init__(self, uri, method, body, headers):
self.uri = uri
self.method = method
self.body = body
self.headers = headers
def get_method_name(self, headers):
def get_endpoint_name(self, headers):
"""Parses request headers and extracts part od the X-Amz-Target
that corresponds to a method of DynamoHandler
@ -52,14 +53,14 @@ class DynamoHandler(object):
return dynamo_json_dump({'__type': type_}), dict(status=400)
def dispatch(self):
method = self.get_method_name(self.headers)
if method:
method = camelcase_to_underscores(method)
return getattr(self, method)(self.uri, self.body, self.headers)
endpoint = self.get_endpoint_name(self.headers)
if endpoint:
endpoint = camelcase_to_underscores(endpoint)
return getattr(self, endpoint)(self.uri, self.method, self.body, self.headers)
else:
return "", dict(status=404)
def list_tables(self, uri, body, headers):
def list_tables(self, uri, method, body, headers):
limit = body.get('Limit')
if body.get("ExclusiveStartTableName"):
last = body.get("ExclusiveStartTableName")
@ -76,7 +77,7 @@ class DynamoHandler(object):
response["LastEvaluatedTableName"] = tables[-1]
return dynamo_json_dump(response)
def create_table(self, uri, body, headers):
def create_table(self, uri, method, body, headers):
name = body['TableName']
key_schema = body['KeySchema']
@ -103,7 +104,7 @@ class DynamoHandler(object):
)
return dynamo_json_dump(table.describe)
def delete_table(self, uri, body, headers):
def delete_table(self, uri, method, body, headers):
name = body['TableName']
table = dynamodb_backend.delete_table(name)
if table:
@ -112,7 +113,7 @@ class DynamoHandler(object):
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
return self.error(er)
def update_table(self, uri, body, headers):
def update_table(self, uri, method, body, headers):
name = body['TableName']
throughput = body["ProvisionedThroughput"]
new_read_units = throughput["ReadCapacityUnits"]
@ -120,7 +121,7 @@ class DynamoHandler(object):
table = dynamodb_backend.update_table_throughput(name, new_read_units, new_write_units)
return dynamo_json_dump(table.describe)
def describe_table(self, uri, body, headers):
def describe_table(self, uri, method, body, headers):
name = body['TableName']
try:
table = dynamodb_backend.tables[name]
@ -129,7 +130,7 @@ class DynamoHandler(object):
return self.error(er)
return dynamo_json_dump(table.describe)
def put_item(self, uri, body, headers):
def put_item(self, uri, method, body, headers):
name = body['TableName']
item = body['Item']
result = dynamodb_backend.put_item(name, item)
@ -141,7 +142,7 @@ class DynamoHandler(object):
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
return self.error(er)
def batch_write_item(self, uri, body, headers):
def batch_write_item(self, uri, method, body, headers):
table_batches = body['RequestItems']
for table_name, table_requests in table_batches.iteritems():
@ -172,7 +173,7 @@ class DynamoHandler(object):
return dynamo_json_dump(response)
def get_item(self, uri, body, headers):
def get_item(self, uri, method, body, headers):
name = body['TableName']
key = body['Key']
hash_key = key['HashKeyElement']
@ -187,7 +188,7 @@ class DynamoHandler(object):
er = 'com.amazonaws.dynamodb.v20111205#ResourceNotFoundException'
return self.error(er)
def batch_get_item(self, uri, body, headers):
def batch_get_item(self, uri, method, body, headers):
table_batches = body['RequestItems']
results = {
@ -210,7 +211,7 @@ class DynamoHandler(object):
results["Responses"][table_name] = {"Items": items, "ConsumedCapacityUnits": 1}
return dynamo_json_dump(results)
def query(self, uri, body, headers):
def query(self, uri, method, body, headers):
name = body['TableName']
hash_key = body['HashKeyValue']
range_condition = body.get('RangeKeyCondition')
@ -241,7 +242,7 @@ class DynamoHandler(object):
# }
return dynamo_json_dump(result)
def scan(self, uri, body, headers):
def scan(self, uri, method, body, headers):
name = body['TableName']
filters = {}
@ -273,7 +274,7 @@ class DynamoHandler(object):
# }
return dynamo_json_dump(result)
def delete_item(self, uri, body, headers):
def delete_item(self, uri, method, body, headers):
name = body['TableName']
key = body['Key']
hash_key = key['HashKeyElement']
@ -292,8 +293,8 @@ class DynamoHandler(object):
return self.error(er)
def handler(uri, body, headers):
def handler(uri, method, body, headers):
if 'GetSessionToken' in body:
return sts_handler(uri, body, headers)
return sts_handler(uri, method, body, headers)
body = json.loads(body or '{}')
return DynamoHandler(uri, body, headers_to_dict(headers)).dispatch()
return DynamoHandler(uri, method, body, headers_to_dict(headers)).dispatch()

2
moto/ec2/responses/__init__.py
View File

@ -65,7 +65,7 @@ class EC2Response(object):
Windows,
]
def dispatch(self, uri, body, headers):
def dispatch(self, uri, method, body, headers):
if body:
querystring = parse_qs(body)
else:

6
moto/s3/responses.py
View File

@ -14,9 +14,8 @@ def all_buckets():
return template.render(buckets=all_buckets)
def bucket_response(uri, body, headers):
def bucket_response(uri, method, body, headers):
hostname = uri.hostname
method = uri.method
querystring = parse_qs(uri.query)
bucket_name = bucket_name_from_hostname(hostname)
@ -60,11 +59,10 @@ def bucket_response(uri, body, headers):
raise NotImplementedError("Method {} has not been impelemented in the S3 backend yet".format(method))
def key_response(uri_info, body, headers):
def key_response(uri_info, method, body, headers):
key_name = uri_info.path.lstrip('/')
hostname = uri_info.hostname
method = uri_info.method
headers = headers_to_dict(headers)
bucket_name = bucket_name_from_hostname(hostname)

4
setup.py
View File

@ -20,5 +20,9 @@ setup(
"boto",
"Jinja2",
"flask",
"spulec-httpretty"
],
dependency_links=[
'http://github.com/spulec/HTTPretty/tarball/master#egg=spulec-httpretty',
]
)