From 814b1d7489ae0bbfc6366b199017e29e36c28a8f Mon Sep 17 00:00:00 2001
From: Krzysztof Wilczynski <krzysztof.wilczynski@linux.com>
Date: Sat, 3 Sep 2016 14:04:35 +0100
Subject: [PATCH] provider/aws: Add missing unit test for validateStateType to
 data_source_availability_zones.

This commit adds missing unit test of a helper function.

Signed-off-by: Krzysztof Wilczynski <krzysztof.wilczynski@linux.com>
---
 .../data_source_availability_zones_test.go    | 28 +++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/builtin/providers/aws/data_source_availability_zones_test.go b/builtin/providers/aws/data_source_availability_zones_test.go
index 7dbae2398..4d2df5acb 100644
--- a/builtin/providers/aws/data_source_availability_zones_test.go
+++ b/builtin/providers/aws/data_source_availability_zones_test.go
@@ -32,6 +32,34 @@ func TestAccAWSAvailabilityZones_basic(t *testing.T) {
 	})
 }
 
+func TestResourceCheckAwsAvailabilityZones_validateStateType(t *testing.T) {
+	_, errors := validateStateType("incorrect", "state")
+	if len(errors) == 0 {
+		t.Fatalf("Expected to trigger a validation error")
+	}
+
+	var testCases = []struct {
+		Value    string
+		ErrCount int
+	}{
+		{
+			Value:    "available",
+			ErrCount: 0,
+		},
+		{
+			Value:    "unavailable",
+			ErrCount: 0,
+		},
+	}
+
+	for _, tc := range testCases {
+		_, errors := validateStateType(tc.Value, "state")
+		if len(errors) != tc.ErrCount {
+			t.Fatalf("Expected %q not to trigger a validation error.", tc.Value)
+		}
+	}
+}
+
 func testAccCheckAwsAvailabilityZonesMeta(n string) resource.TestCheckFunc {
 	return func(s *terraform.State) error {
 		rs, ok := s.RootModule().Resources[n]