Add DynamoDB to tests and CloudFormation

Author: igorborgest

testing/template.yaml

@@ -376,6 +376,22 @@ Resources:
           }
         Name: "aws-data-wrangler-mysql"
 
+  DynamoDBTable:
+    Type: AWS::DynamoDB::Table
+    Properties:
+      AttributeDefinitions:
+        - AttributeName: "attr_hash"
+          AttributeType: "S"
+        - AttributeName: "attr_range"
+          AttributeType: "S"
+      KeySchema:
+        - AttributeName: "attr_hash"
+          KeyType: "HASH"
+        - AttributeName: "attr_range"
+          KeyType: "RANGE"
+      BillingMode: "PAY_PER_REQUEST"
+      TableName: "aws-data-wrangler-test"
+
 Outputs:
   BucketName:
     Value: !Ref Bucket
@@ -415,4 +431,7 @@ Outputs:
     Description: Postgres Address
   MysqlAddress:
     Value: !GetAtt AuroraInstanceMysql.Endpoint.Address
-    Description: Mysql Address
+    Description: Mysql Address
+  DynamoDbTableARN:
+    Value: !GetAtt DynamoDBTable.Arn
+    Description: DynamoDB table name

testing/test_awswrangler/test_dynamodb.py

@@ -0,0 +1,27 @@
+import logging
+
+import boto3
+import pytest
+
+# import awswrangler as wr
+
+logging.basicConfig(level=logging.INFO, format="[%(asctime)s][%(levelname)s][%(name)s][%(funcName)s] %(message)s")
+logging.getLogger("awswrangler").setLevel(logging.DEBUG)
+
+
+@pytest.fixture(scope="module")
+def cloudformation_outputs():
+    response = boto3.client("cloudformation").describe_stacks(StackName="aws-data-wrangler-test")
+    outputs = {}
+    for output in response.get("Stacks")[0].get("Outputs"):
+        outputs[output.get("OutputKey")] = output.get("OutputValue")
+    yield outputs
+
+
+@pytest.fixture(scope="module")
+def ddb_table_arn(session, cloudformation_outputs):
+    if "DynamoDbTableARN" in cloudformation_outputs:
+        arn = cloudformation_outputs["DynamoDbTableARN"]
+    else:
+        raise Exception("You must deploy the test infrastructure using Cloudformation!")
+    yield arn