feat: kinesis stream and functional test harness

Author: kapilt

poetry.lock

@@ -18,6 +18,7 @@ jsonschema = "^3.2.0"
 [tool.poetry.dev-dependencies]
 pytest = "^6.2.4"
 black = "^21.7b0"
+pytest-terraform = { git = "https://github.com/cloud-custodian/pytest-terraform.git", branch = "work-dir-test-api" }
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]

pyproject.toml

@@ -14,3 +14,9 @@ def test_app_lb(validate):
     resource = load_data("app_lb.json")
     props = translator.get_properties(resource)
     validate(translator, props)
+
+
+def test_kinesis(validate):
+    translator = Translator.get_translator("kinesis_stream")()
+    resource = load_data("kinesis_stream.json")
+    validate(translator, translator.get_properties(resource))

tests/unit/conftest.py

@@ -597,6 +597,9 @@ def _camel_str(self, k):
         parts = [p.capitalize() for p in k.split("_")]
         return "".join(parts)
 
+    def get_tags(self, tag_map):
+        return [{"Key": k, "Value": v} for k, v in tag_map.items()]
+
     def camel(self, d):
         r = {}
 
@@ -765,6 +768,20 @@ def get_identity(self, r):
         return {self.id: r["values"]["arn"]}
 
 
+class KinesisStream(Translator):
+
+    tf_type = "kinesis_stream"
+    cfn_type = "AWS::Kinesis::Stream"
+    id = "Name"
+    strip = ("shard_level_metrics", "encryption_type")
+    rename = {"retention_period": "RetentionPeriodHours"}
+
+    def get_properties(self, tfr):
+        cfr = super().get_properties(tfr)
+        cfr["Tags"] = self.get_tags(cfr.get("Tags", {}))
+        return cfr
+
+
 class Lambda(Translator):
 
     tf_type = "lambda_function"
@@ -796,9 +813,7 @@ def get_properties(self, tfr):
                 "variables"
             ]
         cfr["Code"] = {"ZipFile": tfr["values"]["filename"]}
-        cfr["Tags"] = [
-            {"Key": k, "Value": v} for k, v in tfr["values"].get("Tags", {}).items()
-        ]
+        cfr["Tags"] = self.get_tags(tfr["values"].get("Tags", {}))
         if "VpcConfig" in cfr:
             cfr["VpcConfig"].pop("VpcId")
         return cfr