def test_get_service_calls(lob: lobotomy.Lobotomy):
"""Should correctly retrieve service calls made."""
lob.data = {
"clients": {
"sts": {
"get_caller_identity": {
"Account": "123"
}
}
}
}
session = lob()
client = session.client("sts")
assert client.get_caller_identity()["Account"] == "123"
assert client.get_caller_identity(
)["Account"] == "123", "Expected to work twice."
assert len(lob.service_calls) == 2
assert len(lob.get_service_calls("sts", "get_caller_identity")) == 2
assert lob.get_service_call("sts", "get_caller_identity", 0)
assert lob.get_service_call("sts", "get_caller_identity", 1)
with pytest.raises(IndexError):
lob.get_service_call("sts", "get_caller_identity", 2)
def test_dynamodb_get_item(lobotomized: lobotomy.Lobotomy):
"""Should handle dynamodb.get_item without recursion errors."""
lobotomized.data = {"clients": {"dynamodb": {"get_item": {}}}}
session = lobotomized()
client = session.client("dynamodb")
client.get_item(TableName="foo", Key={"pk": {"S": "spam"}, "sk": {"S": "ham"}})
call = lobotomized.get_service_calls("dynamodb", "get_item")[0]
assert call.request["TableName"] == "foo"
def test_s3_download_file(lobotomized: lobotomy.Lobotomy):
"""Should handle s3.download_file correctly despite it being an augmentation."""
lobotomized.data = {"clients": {"s3": {"download_file": {}}}}
session = lobotomized()
client = session.client("s3")
client.download_file(Filename="foo", Bucket="bar", Key="baz")
call = lobotomized.get_service_calls("s3", "download_file")[0]
assert call.request["Filename"] == "foo"
assert call.request["Bucket"] == "bar"
assert call.request["Key"] == "baz"
def test_creation_empty_patched(lob: lobotomy.Lobotomy):
"""Should patch an empty lobotomy and then work after setting data manually."""
lob.data = {
"clients": {
"sts": {
"get_caller_identity": {
"Account": "123"
}
}
}
}
session = lob()
client = session.client("sts")
assert client.get_caller_identity()["Account"] == "123"
assert len(lob.service_calls) == 1
assert len(lob.get_service_calls("sts", "get_caller_identity")) == 1
