dynamodb_get_item_test.py

import pytest
import boto3

from botocore.exceptions import ClientError, ParamValidationError
from moto import mock_dynamodb2

from dynamodb_get_item import get_from_dynamodb

@mock_dynamodb2
def test_get_from_dynamodb():
    client = boto3.client("dynamodb", region_name="us-east-1")
    dynamodb = boto3.resource("dynamodb", region_name="us-east-1")
    table_name = 'SanAntonioSpurs-test'

    # Create the DynamoDB table.
    client.create_table(
        TableName=table_name,
        AttributeDefinitions=[
            {"AttributeName": "PlayerName", "AttributeType": "S"},
            {"AttributeName": "JerseyNumber", "AttributeType": "N"},
        ],
        KeySchema=[
            {"AttributeName": "PlayerName", "KeyType": "HASH"},
            {"AttributeName": "JerseyNumber", "KeyType": "RANGE"},
        ],
        ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123},
    )
    client.put_item(
        TableName=table_name,
        Item={"PlayerName": {"S": "Tony Parker"}, "JerseyNumber": {"N": "9"}, "Position": {"S": "Point Guard"}}
    )

    # Test that the code raises the relevant exceptions
    with pytest.raises(ClientError):
        get_from_dynamodb("SanAntonioSpurs-non-existing-table", {"PlayerName": "Bruce Bowen", "Position": "Shooting Guard", "JerseyNumber": 12})

    with pytest.raises(ParamValidationError):
        get_from_dynamodb(table_name, "intended type mismatch")

    # Test that the code works properly
    response = get_from_dynamodb(table_name, {"PlayerName": "Tony Parker", "JerseyNumber": 9})
    assert response['PlayerName'] == "Tony Parker"
    assert response['Position'] == "Point Guard"
    assert response['JerseyNumber'] == 9