1 files changed, 72 insertions, 0 deletions
diff --git a/plugin/azure/setup_test.go b/plugin/azure/setup_test.go
new file mode 100644
index 000000000..c0b22d581
--- /dev/null
+++ b/plugin/azure/setup_test.go
@@ -0,0 +1,72 @@
+package azure
+
+import (
+	"testing"
+
+	"github.com/caddyserver/caddy"
+)
+
+func TestSetup(t *testing.T) {
+	tests := []struct {
+		body          string
+		expectedError bool
+	}{
+		{`azure`, false},
+		{`azure :`, true},
+		{`azure resource_set:zone`, false},
+		{`azure resource_set:zone {
+    tenant
+}`, true},
+		{`azure resource_set:zone {
+    tenant
+}`, true},
+		{`azure resource_set:zone {
+    client
+}`, true},
+		{`azure resource_set:zone {
+    secret
+}`, true},
+		{`azure resource_set:zone {
+    subscription
+}`, true},
+		{`azure resource_set:zone {
+    upstream 10.0.0.1
+}`, true},
+
+		{`azure resource_set:zone {
+    upstream
+}`, true},
+		{`azure resource_set:zone {
+    foobar
+}`, true},
+		{`azure resource_set:zone {
+    tenant tenant_id
+    client client_id
+    secret client_secret
+    subscription subscription_id
+}`, false},
+
+		{`azure resource_set:zone {
+    fallthrough
+}`, false},
+		{`azure resource_set:zone {
+		environment AZUREPUBLICCLOUD
+	}`, false},
+		{`azure resource_set:zone resource_set:zone {
+			fallthrough
+		}`, true},
+		{`azure resource_set:zone,zone2 {
+			fallthrough
+		}`, false},
+		{`azure resource-set {
+			fallthrough
+		}`, true},
+	}
+
+	for i, test := range tests {
+		c := caddy.NewTestController("dns", test.body)
+		if _, _, _, err := parse(c); (err == nil) == test.expectedError {
+			t.Fatalf("Unexpected errors: %v in test: %d\n\t%s", err, i, test.body)
+		}
+	}
+}