From ed844cdedffc1fa6c53823a4d38f9d7c6270f399 Mon Sep 17 00:00:00 2001
From: Lorenzo Caggioni <lorenzo.caggioni@gmail.com>
Date: Fri, 18 Feb 2022 00:58:14 +0100
Subject: [PATCH] Add tests

---
 fast/stages/03-data-platform/dev/variables.tf |  1 -
 .../s03_data_platform/fixture/__init__.py     | 13 ++++++
 .../stages/s03_data_platform/fixture/main.tf  | 44 +++++++++++++++++++
 .../stages/s03_data_platform/test_plan.py     | 20 +++++++++
 4 files changed, 77 insertions(+), 1 deletion(-)
 create mode 100644 tests/fast/stages/s03_data_platform/fixture/__init__.py
 create mode 100644 tests/fast/stages/s03_data_platform/fixture/main.tf
 create mode 100644 tests/fast/stages/s03_data_platform/test_plan.py

diff --git a/fast/stages/03-data-platform/dev/variables.tf b/fast/stages/03-data-platform/dev/variables.tf
index 903b9cf35..70cf428e9 100644
--- a/fast/stages/03-data-platform/dev/variables.tf
+++ b/fast/stages/03-data-platform/dev/variables.tf
@@ -48,7 +48,6 @@ variable "folder_ids" {
   type = object({
     data-platform = string
   })
-  default = null
 }
 
 variable "groups" {
diff --git a/tests/fast/stages/s03_data_platform/fixture/__init__.py b/tests/fast/stages/s03_data_platform/fixture/__init__.py
new file mode 100644
index 000000000..6d6d1266c
--- /dev/null
+++ b/tests/fast/stages/s03_data_platform/fixture/__init__.py
@@ -0,0 +1,13 @@
+# Copyright 2022 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.
diff --git a/tests/fast/stages/s03_data_platform/fixture/main.tf b/tests/fast/stages/s03_data_platform/fixture/main.tf
new file mode 100644
index 000000000..e7fd8d4d4
--- /dev/null
+++ b/tests/fast/stages/s03_data_platform/fixture/main.tf
@@ -0,0 +1,44 @@
+/**
+ * Copyright 2022 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.
+ */
+
+# tfdoc: Data platform stage test
+
+module "stage" {
+  source = "../../../../../fast/stages/03-data-platform/dev/"
+  billing_account = {
+    id              = "012345-67890A-BCDEF0",
+    organization_id = 123456
+  }
+  folder_ids = {
+    data-platform = "folders/12345678"
+  }
+  host_project_ids = {
+    dev-spoke-0 = "fast-dev-net-spoke-0"
+  }
+  organization = {
+    domain      = "example.com"
+    id          = 123456789012
+    customer_id = "A11aaaaa1"
+  }
+  prefix = "fast"
+  subnet_self_links = {
+    dev-spoke-0 = {
+      "europe-west1/dev-dataplatform-ew1" : "https://www.googleapis.com/compute/v1/projects/fast-dev-net-spoke-0/regions/europe-west1/subnetworks/dev-dataplatform-ew1",
+      "europe-west1/dev-default-ew1" : "https://www.googleapis.com/compute/v1/projects/fast-dev-net-spoke-0/regions/europe-west1/subnetworks/dev-default-ew1"
+    }
+  }
+  vpc_self_links = { dev-spoke-0 = "https://www.googleapis.com/compute/v1/projects/fast-dev-net-spoke-0/global/networks/dev-spoke-0" }
+}
diff --git a/tests/fast/stages/s03_data_platform/test_plan.py b/tests/fast/stages/s03_data_platform/test_plan.py
new file mode 100644
index 000000000..6189f62e3
--- /dev/null
+++ b/tests/fast/stages/s03_data_platform/test_plan.py
@@ -0,0 +1,20 @@
+# Copyright 2022 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.
+
+
+def test_counts(fast_e2e_plan_runner):
+  "Test stage."
+  num_modules, num_resources, _ = fast_e2e_plan_runner()
+  # TODO: to re-enable per-module resource count check print _, then test
+  assert num_modules > 0 and num_resources > 0