catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Run black on moto & test directories.

Browse source
This commit is contained in:
Asher Foa 2019-10-31 08:44:26 -07:00
commit 96e5b1993d
507 changed files with 52541 additions and 47814 deletions
Download patch file Download diff file Expand all files Collapse all files

38
tests/test_xray/test_xray_client.py
View file

@ -1,6 +1,6 @@
from __future__ import unicode_literals
from moto import mock_xray_client, XRaySegment, mock_dynamodb2
import sure # noqa
import sure # noqa
import boto3
from moto.xray.mock_client import MockEmitter
@ -9,10 +9,12 @@ import aws_xray_sdk.core.patcher as xray_core_patcher
import botocore.client
import botocore.endpoint
original_make_api_call = botocore.client.BaseClient._make_api_call
original_encode_headers = botocore.endpoint.Endpoint._encode_headers
import requests
original_session_request = requests.Session.request
original_session_prep_request = requests.Session.prepare_request
@ -24,24 +26,24 @@ def test_xray_dynamo_request_id():
xray_core_patcher._PATCHED_MODULES = set()
xray_core.patch_all()
client = boto3.client('dynamodb', region_name='us-east-1')
client = boto3.client("dynamodb", region_name="us-east-1")
with XRaySegment():
resp = client.list_tables()
resp['ResponseMetadata'].should.contain('RequestId')
id1 = resp['ResponseMetadata']['RequestId']
resp["ResponseMetadata"].should.contain("RequestId")
id1 = resp["ResponseMetadata"]["RequestId"]
with XRaySegment():
client.list_tables()
resp = client.list_tables()
id2 = resp['ResponseMetadata']['RequestId']
id2 = resp["ResponseMetadata"]["RequestId"]
id1.should_not.equal(id2)
setattr(botocore.client.BaseClient, '_make_api_call', original_make_api_call)
setattr(botocore.endpoint.Endpoint, '_encode_headers', original_encode_headers)
setattr(requests.Session, 'request', original_session_request)
setattr(requests.Session, 'prepare_request', original_session_prep_request)
setattr(botocore.client.BaseClient, "_make_api_call", original_make_api_call)
setattr(botocore.endpoint.Endpoint, "_encode_headers", original_encode_headers)
setattr(requests.Session, "request", original_session_request)
setattr(requests.Session, "prepare_request", original_session_prep_request)
@mock_xray_client
@ -52,10 +54,10 @@ def test_xray_udp_emitter_patched():
assert isinstance(xray_core.xray_recorder._emitter, MockEmitter)
setattr(botocore.client.BaseClient, '_make_api_call', original_make_api_call)
setattr(botocore.endpoint.Endpoint, '_encode_headers', original_encode_headers)
setattr(requests.Session, 'request', original_session_request)
setattr(requests.Session, 'prepare_request', original_session_prep_request)
setattr(botocore.client.BaseClient, "_make_api_call", original_make_api_call)
setattr(botocore.endpoint.Endpoint, "_encode_headers", original_encode_headers)
setattr(requests.Session, "request", original_session_request)
setattr(requests.Session, "prepare_request", original_session_prep_request)
@mock_xray_client
@ -64,9 +66,9 @@ def test_xray_context_patched():
xray_core_patcher._PATCHED_MODULES = set()
xray_core.patch_all()
xray_core.xray_recorder._context.context_missing.should.equal('LOG_ERROR')
xray_core.xray_recorder._context.context_missing.should.equal("LOG_ERROR")
setattr(botocore.client.BaseClient, '_make_api_call', original_make_api_call)
setattr(botocore.endpoint.Endpoint, '_encode_headers', original_encode_headers)
setattr(requests.Session, 'request', original_session_request)
setattr(requests.Session, 'prepare_request', original_session_prep_request)
setattr(botocore.client.BaseClient, "_make_api_call", original_make_api_call)
setattr(botocore.endpoint.Endpoint, "_encode_headers", original_encode_headers)
setattr(requests.Session, "request", original_session_request)
setattr(requests.Session, "prepare_request", original_session_prep_request)