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

153
tests/test_xray/test_xray_boto3.py
View file

@ -3,7 +3,7 @@ from __future__ import unicode_literals
import boto3
import json
import botocore.exceptions
import sure # noqa
import sure # noqa
from moto import mock_xray
@ -12,128 +12,137 @@ import datetime
@mock_xray
def test_put_telemetry():
client = boto3.client('xray', region_name='us-east-1')
client = boto3.client("xray", region_name="us-east-1")
client.put_telemetry_records(
TelemetryRecords=[
{
'Timestamp': datetime.datetime(2015, 1, 1),
'SegmentsReceivedCount': 123,
'SegmentsSentCount': 123,
'SegmentsSpilloverCount': 123,
'SegmentsRejectedCount': 123,
'BackendConnectionErrors': {
'TimeoutCount': 123,
'ConnectionRefusedCount': 123,
'HTTPCode4XXCount': 123,
'HTTPCode5XXCount': 123,
'UnknownHostCount': 123,
'OtherCount': 123
}
},
"Timestamp": datetime.datetime(2015, 1, 1),
"SegmentsReceivedCount": 123,
"SegmentsSentCount": 123,
"SegmentsSpilloverCount": 123,
"SegmentsRejectedCount": 123,
"BackendConnectionErrors": {
"TimeoutCount": 123,
"ConnectionRefusedCount": 123,
"HTTPCode4XXCount": 123,
"HTTPCode5XXCount": 123,
"UnknownHostCount": 123,
"OtherCount": 123,
},
}
],
EC2InstanceId='string',
Hostname='string',
ResourceARN='string'
EC2InstanceId="string",
Hostname="string",
ResourceARN="string",
)
@mock_xray
def test_put_trace_segments():
client = boto3.client('xray', region_name='us-east-1')
client = boto3.client("xray", region_name="us-east-1")
client.put_trace_segments(
TraceSegmentDocuments=[
json.dumps({
'name': 'example.com',
'id': '70de5b6f19ff9a0a',
'start_time': 1.478293361271E9,
'trace_id': '1-581cf771-a006649127e371903a2de979',
'end_time': 1.478293361449E9
})
json.dumps(
{
"name": "example.com",
"id": "70de5b6f19ff9a0a",
"start_time": 1.478293361271e9,
"trace_id": "1-581cf771-a006649127e371903a2de979",
"end_time": 1.478293361449e9,
}
)
]
)
@mock_xray
def test_trace_summary():
client = boto3.client('xray', region_name='us-east-1')
client = boto3.client("xray", region_name="us-east-1")
client.put_trace_segments(
TraceSegmentDocuments=[
json.dumps({
'name': 'example.com',
'id': '70de5b6f19ff9a0a',
'start_time': 1.478293361271E9,
'trace_id': '1-581cf771-a006649127e371903a2de979',
'in_progress': True
}),
json.dumps({
'name': 'example.com',
'id': '70de5b6f19ff9a0b',
'start_time': 1478293365,
'trace_id': '1-581cf771-a006649127e371903a2de979',
'end_time': 1478293385
})
json.dumps(
{
"name": "example.com",
"id": "70de5b6f19ff9a0a",
"start_time": 1.478293361271e9,
"trace_id": "1-581cf771-a006649127e371903a2de979",
"in_progress": True,
}
),
json.dumps(
{
"name": "example.com",
"id": "70de5b6f19ff9a0b",
"start_time": 1478293365,
"trace_id": "1-581cf771-a006649127e371903a2de979",
"end_time": 1478293385,
}
),
]
)
client.get_trace_summaries(
StartTime=datetime.datetime(2014, 1, 1),
EndTime=datetime.datetime(2017, 1, 1)
StartTime=datetime.datetime(2014, 1, 1), EndTime=datetime.datetime(2017, 1, 1)
)
@mock_xray
def test_batch_get_trace():
client = boto3.client('xray', region_name='us-east-1')
client = boto3.client("xray", region_name="us-east-1")
client.put_trace_segments(
TraceSegmentDocuments=[
json.dumps({
'name': 'example.com',
'id': '70de5b6f19ff9a0a',
'start_time': 1.478293361271E9,
'trace_id': '1-581cf771-a006649127e371903a2de979',
'in_progress': True
}),
json.dumps({
'name': 'example.com',
'id': '70de5b6f19ff9a0b',
'start_time': 1478293365,
'trace_id': '1-581cf771-a006649127e371903a2de979',
'end_time': 1478293385
})
json.dumps(
{
"name": "example.com",
"id": "70de5b6f19ff9a0a",
"start_time": 1.478293361271e9,
"trace_id": "1-581cf771-a006649127e371903a2de979",
"in_progress": True,
}
),
json.dumps(
{
"name": "example.com",
"id": "70de5b6f19ff9a0b",
"start_time": 1478293365,
"trace_id": "1-581cf771-a006649127e371903a2de979",
"end_time": 1478293385,
}
),
]
)
resp = client.batch_get_traces(
TraceIds=['1-581cf771-a006649127e371903a2de979', '1-581cf772-b006649127e371903a2de979']
TraceIds=[
"1-581cf771-a006649127e371903a2de979",
"1-581cf772-b006649127e371903a2de979",
]
)
len(resp['UnprocessedTraceIds']).should.equal(1)
len(resp['Traces']).should.equal(1)
len(resp["UnprocessedTraceIds"]).should.equal(1)
len(resp["Traces"]).should.equal(1)
# Following are not implemented, just testing it returns what boto expects
@mock_xray
def test_batch_get_service_graph():
client = boto3.client('xray', region_name='us-east-1')
client = boto3.client("xray", region_name="us-east-1")
client.get_service_graph(
StartTime=datetime.datetime(2014, 1, 1),
EndTime=datetime.datetime(2017, 1, 1)
StartTime=datetime.datetime(2014, 1, 1), EndTime=datetime.datetime(2017, 1, 1)
)
@mock_xray
def test_batch_get_trace_graph():
client = boto3.client('xray', region_name='us-east-1')
client = boto3.client("xray", region_name="us-east-1")
client.batch_get_traces(
TraceIds=['1-581cf771-a006649127e371903a2de979', '1-581cf772-b006649127e371903a2de979']
TraceIds=[
"1-581cf771-a006649127e371903a2de979",
"1-581cf772-b006649127e371903a2de979",
]
)

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)