Skip to content

Commit c2f816a

Browse files
committed
Add PaginationToken to not required otuput attributes
1 parent 3dc5cd4 commit c2f816a

File tree

3 files changed

+72
-54
lines changed

3 files changed

+72
-54
lines changed

mypy_boto3_builder/type_maps/required_attribute_map.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
ServiceNameCatalog.all: {
2121
ALL: {
2222
"NextToken": False,
23+
"PaginationToken": False,
2324
"nextToken": False,
2425
"Contents": False,
2526
"Item": False,
Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
from mypy_boto3_builder.service_name import ServiceNameCatalog
2+
from mypy_boto3_builder.type_maps.required_attribute_map import is_required
3+
4+
5+
def test_is_required() -> None:
6+
assert is_required(ServiceNameCatalog.ec2, "RandomTypeDef", "PaginationToken") is False
7+
assert is_required(ServiceNameCatalog.dynamodb, "RandomTypeDef", "Key") is True
8+
assert is_required(ServiceNameCatalog.dynamodb, "RandomTypeDef", "LastEvaluatedKey") is False
9+
assert is_required(ServiceNameCatalog.stepfunctions, "RandomTypeDef", "stopDate") is True
10+
assert (
11+
is_required(ServiceNameCatalog.stepfunctions, "DescribeExecutionOutputTypeDef", "stopDate")
12+
is False
13+
)
14+
assert (
15+
is_required(ServiceNameCatalog.stepfunctions, "DescribeExecutionOutputTypeDef", "Date")
16+
is True
17+
)

0 commit comments

Comments
 (0)