81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
import boto3
|
|
|
|
from moto import mock_aws
|
|
|
|
|
|
@mock_aws
|
|
def test_update_table__billing_mode():
|
|
client = boto3.client("dynamodb", region_name="us-east-1")
|
|
client.create_table(
|
|
TableName="test",
|
|
AttributeDefinitions=[
|
|
{"AttributeName": "client", "AttributeType": "S"},
|
|
{"AttributeName": "app", "AttributeType": "S"},
|
|
],
|
|
KeySchema=[
|
|
{"AttributeName": "client", "KeyType": "HASH"},
|
|
{"AttributeName": "app", "KeyType": "RANGE"},
|
|
],
|
|
BillingMode="PAY_PER_REQUEST",
|
|
)
|
|
|
|
client.update_table(
|
|
TableName="test",
|
|
BillingMode="PROVISIONED",
|
|
ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1},
|
|
)
|
|
|
|
actual = client.describe_table(TableName="test")["Table"]
|
|
assert actual["BillingModeSummary"] == {"BillingMode": "PROVISIONED"}
|
|
assert actual["ProvisionedThroughput"] == {
|
|
"ReadCapacityUnits": 1,
|
|
"WriteCapacityUnits": 1,
|
|
}
|
|
|
|
|
|
@mock_aws
|
|
def test_update_table_throughput():
|
|
conn = boto3.resource("dynamodb", region_name="us-west-2")
|
|
table = conn.create_table(
|
|
TableName="messages",
|
|
KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}],
|
|
AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}],
|
|
ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5},
|
|
)
|
|
assert table.provisioned_throughput["ReadCapacityUnits"] == 5
|
|
assert table.provisioned_throughput["WriteCapacityUnits"] == 5
|
|
|
|
table.update(
|
|
ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 6}
|
|
)
|
|
|
|
assert table.provisioned_throughput["ReadCapacityUnits"] == 5
|
|
assert table.provisioned_throughput["WriteCapacityUnits"] == 6
|
|
|
|
|
|
@mock_aws
|
|
def test_update_table__enable_stream():
|
|
conn = boto3.client("dynamodb", region_name="us-east-1")
|
|
|
|
resp = conn.create_table(
|
|
TableName="test-stream-update",
|
|
KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}],
|
|
AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}],
|
|
ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1},
|
|
)
|
|
|
|
assert "StreamSpecification" not in resp["TableDescription"]
|
|
|
|
resp = conn.update_table(
|
|
TableName="test-stream-update",
|
|
StreamSpecification={"StreamEnabled": True, "StreamViewType": "NEW_IMAGE"},
|
|
)
|
|
|
|
assert "StreamSpecification" in resp["TableDescription"]
|
|
assert resp["TableDescription"]["StreamSpecification"] == {
|
|
"StreamEnabled": True,
|
|
"StreamViewType": "NEW_IMAGE",
|
|
}
|
|
assert "LatestStreamLabel" in resp["TableDescription"]
|
|
assert "LatestStreamArn" in resp["TableDescription"]
|