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