Merge pull request #223 from briantkennedy/location

Add location method to Key.

Author: google-oss-prow[bot]

pkg/cloud/meta/key.go

@@ -72,6 +72,18 @@ func (k *Key) Type() KeyType {
 	}
 }
 
+// Location returns the location for the key.
+func (k *Key) Location() string {
+	switch {
+	case k.Zone != "":
+		return k.Zone
+	case k.Region != "":
+		return k.Region
+	default:
+		return "global"
+	}
+}
+
 // String returns a string representation of the key.
 func (k Key) String() string {
 	switch k.Type() {

pkg/cloud/meta/key_test.go

@@ -37,6 +37,21 @@ func TestKeyType(t *testing.T) {
 	}
 }
 
+func TestKeyLocation(t *testing.T) {
+	for _, tc := range []struct {
+		key  *Key
+		want string
+	}{
+		{GlobalKey("abc"), "global"},
+		{ZonalKey("abc", "us-central1-b"), "us-central1-b"},
+		{RegionalKey("abc", "us-central1"), "us-central1"},
+	} {
+		if tc.key.Location() != tc.want {
+			t.Errorf("key.Location() == %v, want %v", tc.key.Type(), tc.want)
+		}
+	}
+}
+
 func TestKeyString(t *testing.T) {
 	t.Parallel()