added tests for Gcp Memorystore Redis

forseti-security · Mar 31, 2021 · d465edc · d465edc
1 parent e919843
commit d465edc
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/tests/test_resources.py b/tests/test_resources.py
@@ -40,6 +40,7 @@
 from rpe.resources.gcp import GcpComputeFirewall
 from rpe.resources.gcp import GcpComputeSubnetwork
 from rpe.resources.gcp import GcpDataflowJob
+from rpe.resources.gcp import GcpMemorystoreRedis
 
 test_project = "my_project"
 test_resource_name = "my_resource"
@@ -244,6 +245,16 @@
         resource_type='dataflow.googleapis.com/Job',
         name='//dataflow.googleapis.com/projects/my_project/locations/us-central1/jobs/my_resource'
     ),
+    ResourceTestCase(
+        resource_data={
+            'name': test_resource_name,
+            'location': 'us-central1',
+            'project_id': test_project
+        },
+        cls=GcpMemorystoreRedis,
+        resource_type='redis.googleapis.com/Instance',
+        name='//redis.googleapis.com/projects/my_project/locations/us-central1/instances/my_resource'
+    ),
 ]
 
 

diff --git a/tests/test_resources_cai.py b/tests/test_resources_cai.py
@@ -43,6 +43,7 @@
 from rpe.resources.gcp import GcpIamServiceAccount
 from rpe.resources.gcp import GcpIamServiceAccountKey
 from rpe.resources.gcp import GcpDataflowJob
+from rpe.resources.gcp import GcpMemorystoreRedis
 
 client_kwargs = {
     'credentials': Credentials(token='')
@@ -212,6 +213,13 @@
         },
         resource_cls=GcpDataflowJob
     ),
+    CaiTestCase(
+        data={
+            "name": "//redis.googleapis.com/projects/test-project/locations/us-central1/instances/test-resource",
+            "asset_type": "redis.googleapis.com/Instance",
+        },
+        resource_cls=GcpMemorystoreRedis
+    ),
 ]