Extend testAccAzureRMIotHub_customRoutes test

Also check whether EventHub endpoint works
hashicorp · Mar 8, 2019 · 010901f · 010901f
1 parent a4e479e
commit 010901f
Showing 1 changed file with 42 additions and 1 deletion.
diff --git a/azurerm/resource_arm_iothub_test.go b/azurerm/resource_arm_iothub_test.go
@@ -102,6 +102,10 @@ func TestAccAzureRMIotHub_customRoutes(t *testing.T) {
 				Config: testAccAzureRMIotHub_customRoutes(rInt, rStr, testLocation()),
 				Check: resource.ComposeTestCheckFunc(
 					testCheckAzureRMIotHubExists(resourceName),
+					resource.TestCheckResourceAttr(resourceName, "endpoint.#", "2"),
+					resource.TestCheckResourceAttr(resourceName, "endpoint.0.type", "AzureIotHub.StorageContainer"),
+					resource.TestCheckResourceAttr(resourceName, "endpoint.1.type", "AzureIotHub.IoTHub"),
+					resource.TestCheckResourceAttr(resourceName, "route.#", "2"),
 				),
 			},
 			{
@@ -289,6 +293,29 @@ resource "azurerm_storage_container" "test" {
   container_access_type = "private"
 }
 
+resource "azurerm_eventhub_namespace" "test" {
+  resource_group_name = "${azurerm_resource_group.test.name}"
+  location            = "${azurerm_resource_group.test.location}"
+  name                = "acctest-%d"
+  sku                 = "Basic"
+}
+
+resource "azurerm_eventhub" "test" {
+  name                = "acctest"
+  resource_group_name = "${azurerm_resource_group.test.name}"
+  namespace_name      = "${azurerm_eventhub_namespace.test.name}"
+  partition_count     = 2
+  message_retention   = 1
+}
+
+resource "azurerm_eventhub_authorization_rule" "test" {
+  resource_group_name = "${azurerm_resource_group.test.name}"
+  namespace_name      = "${azurerm_eventhub_namespace.test.name}"
+  eventhub_name       = "${azurerm_eventhub.test.name}"
+  name                = "acctest"
+  send                = true  
+}
+
 resource "azurerm_iothub" "test" {
   name                = "acctestIoTHub-%d"
   resource_group_name = "${azurerm_resource_group.test.name}"
@@ -311,6 +338,12 @@ resource "azurerm_iothub" "test" {
     file_name_format           = "{iothub}/{partition}_{YYYY}_{MM}_{DD}_{HH}_{mm}"
   }
 
+  endpoint {
+    type              = "AzureIotHub.EventHub"
+    connection_string = "${azurerm_eventhub_authorization_rule.test.primary_connection_string}"
+    name              = "export2"
+  }
+
   route {
     name           = "export"
     source         = "DeviceMessages"
@@ -319,11 +352,19 @@ resource "azurerm_iothub" "test" {
     enabled        = true
   }
 
+  route {
+    name           = "export2"
+    source         = "DeviceMessages"
+    condition      = "true"
+    endpoint_names = ["export2"]
+    enabled        = true
+  }
+
   tags {
     "purpose" = "testing"
   }
 }
-`, rInt, location, rStr, rInt)
+`, rInt, location, rStr, rInt, rInt)
 }
 
 func testAccAzureRMIotHub_fallbackRoute(rInt int, location string) string {