Add tests for logging sinks

tests/modules/organization/fixture/main.tf

@@ -26,4 +26,6 @@ module "test" {
   policy_list                 = var.policy_list
   firewall_policies           = var.firewall_policies
   firewall_policy_attachments = var.firewall_policy_attachments
+  logging_sinks               = var.logging_sinks
+  logging_exclusions          = var.logging_exclusions
 }

tests/modules/organization/fixture/variables.tf

@@ -73,3 +73,18 @@ variable "firewall_policy_attachments" {
   type    = map(string)
   default = {}
 }
+
+variable "logging_sinks" {
+  type = map(object({
+    destination = string
+    type        = string
+    filter      = string
+    iam         = bool
+  }))
+  default = {}
+}
+
+variable "logging_exclusions" {
+  type    = map(string)
+  default = {}
+}

tests/modules/organization/test_plan_logging.py

@@ -0,0 +1,108 @@
+# Copyright 2020 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+import os
+import pytest
+
+from collections import Counter
+
+FIXTURES_DIR = os.path.join(os.path.dirname(__file__), 'fixture')
+
+
+def test_sinks(plan_runner):
+  "Test folder-level sinks."
+  logging_sinks = """ {
+    warning = {
+      type        = "gcs"
+      destination = "mybucket"
+      filter      = "severity=WARNING"
+      iam         = true
+    }
+    info = {
+      type        = "bigquery"
+      destination = "projects/myproject/datasets/mydataset"
+      filter      = "severity=INFO"
+      iam         = true
+    }
+    notice = {
+      type        = "pubsub"
+      destination = "projects/myproject/topics/mytopic"
+      filter      = "severity=NOTICE"
+      iam         = true
+    }
+  }
+  """
+  _, resources = plan_runner(FIXTURES_DIR, logging_sinks=logging_sinks)
+  assert len(resources) == 6
+
+  resource_types = Counter([r['type'] for r in resources])
+  assert resource_types == {
+    'google_bigquery_dataset_iam_binding': 1,
+    'google_logging_organization_sink': 3,
+    'google_pubsub_topic_iam_binding': 1,
+    'google_storage_bucket_iam_binding': 1
+  }
+  
+  sinks = [r for r in resources
+           if r['type'] == 'google_logging_organization_sink']
+  assert sorted([r['index'] for r in sinks]) == [
+      'info',
+      'notice',
+      'warning',
+  ]
+  values = [(r['index'], r['values']['filter'], r['values']['destination'])
+            for r in sinks]
+  assert sorted(values) == [
+    ('info',
+     'severity=INFO',
+     'bigquery.googleapis.com/projects/myproject/datasets/mydataset'),
+    ('notice',
+     'severity=NOTICE',
+     'pubsub.googleapis.com/projects/myproject/topics/mytopic'),
+    ('warning', 'severity=WARNING', 'storage.googleapis.com/mybucket')]
+
+  bindings = [r for r in resources
+              if 'binding' in r['type']]
+  values = [(r['index'], r['type'], r['values']['role'])
+            for r in bindings]
+  assert sorted(values) == [
+    ('info', 'google_bigquery_dataset_iam_binding', 'roles/bigquery.dataEditor'),
+    ('notice', 'google_pubsub_topic_iam_binding', 'roles/pubsub.publisher'),
+    ('warning', 'google_storage_bucket_iam_binding', 'roles/storage.objectCreator')
+  ]
+
+  
+def test_exclusions(plan_runner):
+  "Test folder-level logging exclusions."
+  logging_exclusions = (
+      '{'
+      'exclusion1 = "resource.type=gce_instance", '
+      'exclusion2 = "severity=NOTICE", '
+      '}'
+  )
+  _, resources = plan_runner(FIXTURES_DIR,
+                             logging_exclusions=logging_exclusions)
+  assert len(resources) == 2
+  exclusions = [r for r in resources
+                if r['type'] == 'google_logging_organization_exclusion']
+  assert sorted([r['index'] for r in exclusions]) == [
+      'exclusion1',
+      'exclusion2',
+  ]
+  values = [(r['index'], r['values']['filter']) for r in exclusions]
+  assert sorted(values) == [
+    ('exclusion1', 'resource.type=gce_instance'),
+    ('exclusion2', 'severity=NOTICE')
+  ]