| // Copyright Istio Authors |
| // |
| // 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. |
| |
| package platform |
| |
| import ( |
| "errors" |
| "fmt" |
| "os" |
| "reflect" |
| "sync" |
| "testing" |
| ) |
| |
| import ( |
| "cloud.google.com/go/compute/metadata" |
| ) |
| |
| func TestGCPMetadata(t *testing.T) { |
| tests := []struct { |
| name string |
| shouldFill shouldFillFn |
| projectIDFn metadataFn |
| numericProjectIDFn metadataFn |
| locationFn metadataFn |
| clusterNameFn metadataFn |
| instanceNameFn metadataFn |
| instanceIDFn metadataFn |
| instanceTemplateFn metadataFn |
| instanceCreatedByFn metadataFn |
| env map[string]string |
| want map[string]string |
| }{ |
| { |
| "should not fill", |
| func() bool { return false }, |
| func() (string, error) { return "pid", nil }, |
| func() (string, error) { return "npid", nil }, |
| func() (string, error) { return "location", nil }, |
| func() (string, error) { return "cluster", nil }, |
| func() (string, error) { return "instanceName", nil }, |
| func() (string, error) { return "instance", nil }, |
| func() (string, error) { return "instanceTemplate", nil }, |
| func() (string, error) { return "createdBy", nil }, |
| map[string]string{}, |
| map[string]string{}, |
| }, |
| { |
| "should fill", |
| func() bool { return true }, |
| func() (string, error) { return "pid", nil }, |
| func() (string, error) { return "npid", nil }, |
| func() (string, error) { return "location", nil }, |
| func() (string, error) { return "cluster", nil }, |
| func() (string, error) { return "instanceName", nil }, |
| func() (string, error) { return "instance", nil }, |
| func() (string, error) { return "instanceTemplate", nil }, |
| func() (string, error) { return "createdBy", nil }, |
| map[string]string{}, |
| map[string]string{ |
| GCPProject: "pid", GCPProjectNumber: "npid", GCPLocation: "location", GCPCluster: "cluster", GCEInstance: "instanceName", |
| GCEInstanceID: "instance", GCEInstanceTemplate: "instanceTemplate", GCEInstanceCreatedBy: "createdBy", |
| GCPClusterURL: "https://container.googleapis.com/v1/projects/pid/locations/location/clusters/cluster", |
| }, |
| }, |
| { |
| "project id error", |
| func() bool { return true }, |
| func() (string, error) { return "", errors.New("error") }, |
| func() (string, error) { return "npid", nil }, |
| func() (string, error) { return "location", nil }, |
| func() (string, error) { return "cluster", nil }, |
| func() (string, error) { return "instanceName", nil }, |
| func() (string, error) { return "instance", nil }, |
| func() (string, error) { return "instanceTemplate", nil }, |
| func() (string, error) { return "createdBy", nil }, |
| map[string]string{}, |
| map[string]string{ |
| GCPLocation: "location", GCPProjectNumber: "npid", GCPCluster: "cluster", GCEInstance: "instanceName", GCEInstanceID: "instance", |
| GCEInstanceTemplate: "instanceTemplate", GCEInstanceCreatedBy: "createdBy", |
| }, |
| }, |
| { |
| "numeric project id error", |
| func() bool { return true }, |
| func() (string, error) { return "pid", nil }, |
| func() (string, error) { return "", errors.New("error") }, |
| func() (string, error) { return "location", nil }, |
| func() (string, error) { return "cluster", nil }, |
| func() (string, error) { return "instanceName", nil }, |
| func() (string, error) { return "instance", nil }, |
| func() (string, error) { return "instanceTemplate", nil }, |
| func() (string, error) { return "createdBy", nil }, |
| map[string]string{}, |
| map[string]string{ |
| GCPLocation: "location", GCPProject: "pid", GCPCluster: "cluster", GCEInstance: "instanceName", GCEInstanceID: "instance", |
| GCEInstanceTemplate: "instanceTemplate", GCEInstanceCreatedBy: "createdBy", |
| GCPClusterURL: "https://container.googleapis.com/v1/projects/pid/locations/location/clusters/cluster", |
| }, |
| }, |
| { |
| "location error", |
| func() bool { return true }, |
| func() (string, error) { return "pid", nil }, |
| func() (string, error) { return "npid", nil }, |
| func() (string, error) { return "location", errors.New("error") }, |
| func() (string, error) { return "cluster", nil }, |
| func() (string, error) { return "instanceName", nil }, |
| func() (string, error) { return "instance", nil }, |
| func() (string, error) { return "instanceTemplate", nil }, |
| func() (string, error) { return "createdBy", nil }, |
| map[string]string{}, |
| map[string]string{ |
| GCPProject: "pid", GCPProjectNumber: "npid", GCPCluster: "cluster", GCEInstance: "instanceName", GCEInstanceID: "instance", |
| GCEInstanceTemplate: "instanceTemplate", GCEInstanceCreatedBy: "createdBy", |
| }, |
| }, |
| { |
| "cluster name error", |
| func() bool { return true }, |
| func() (string, error) { return "pid", nil }, |
| func() (string, error) { return "npid", nil }, |
| func() (string, error) { return "location", nil }, |
| func() (string, error) { return "cluster", errors.New("error") }, |
| func() (string, error) { return "instanceName", nil }, |
| func() (string, error) { return "instance", nil }, |
| func() (string, error) { return "instanceTemplate", nil }, |
| func() (string, error) { return "createdBy", nil }, |
| map[string]string{}, |
| map[string]string{ |
| GCPProject: "pid", GCPProjectNumber: "npid", GCPLocation: "location", GCEInstance: "instanceName", GCEInstanceID: "instance", |
| GCEInstanceTemplate: "instanceTemplate", GCEInstanceCreatedBy: "createdBy", |
| }, |
| }, |
| { |
| "instance name error", |
| func() bool { return true }, |
| func() (string, error) { return "pid", nil }, |
| func() (string, error) { return "npid", nil }, |
| func() (string, error) { return "location", nil }, |
| func() (string, error) { return "cluster", nil }, |
| func() (string, error) { return "instanceName", errors.New("error") }, |
| func() (string, error) { return "instance", nil }, |
| func() (string, error) { return "instanceTemplate", nil }, |
| func() (string, error) { return "createdBy", nil }, |
| map[string]string{}, |
| map[string]string{ |
| GCPProject: "pid", GCPProjectNumber: "npid", GCPLocation: "location", GCPCluster: "cluster", GCEInstanceID: "instance", |
| GCEInstanceTemplate: "instanceTemplate", GCEInstanceCreatedBy: "createdBy", |
| GCPClusterURL: "https://container.googleapis.com/v1/projects/pid/locations/location/clusters/cluster", |
| }, |
| }, |
| { |
| "instance id error", |
| func() bool { return true }, |
| func() (string, error) { return "pid", nil }, |
| func() (string, error) { return "npid", nil }, |
| func() (string, error) { return "location", nil }, |
| func() (string, error) { return "cluster", nil }, |
| func() (string, error) { return "instanceName", nil }, |
| func() (string, error) { return "", errors.New("error") }, |
| func() (string, error) { return "instanceTemplate", nil }, |
| func() (string, error) { return "createdBy", nil }, |
| map[string]string{}, |
| map[string]string{ |
| GCPProject: "pid", GCPProjectNumber: "npid", GCPLocation: "location", GCPCluster: "cluster", GCEInstance: "instanceName", |
| GCEInstanceTemplate: "instanceTemplate", GCEInstanceCreatedBy: "createdBy", |
| GCPClusterURL: "https://container.googleapis.com/v1/projects/pid/locations/location/clusters/cluster", |
| }, |
| }, |
| { |
| "instance template error", |
| func() bool { return true }, |
| func() (string, error) { return "pid", nil }, |
| func() (string, error) { return "npid", nil }, |
| func() (string, error) { return "location", nil }, |
| func() (string, error) { return "cluster", nil }, |
| func() (string, error) { return "instanceName", nil }, |
| func() (string, error) { return "instance", nil }, |
| func() (string, error) { return "", errors.New("error") }, |
| func() (string, error) { return "createdBy", nil }, |
| map[string]string{}, |
| map[string]string{ |
| GCPProject: "pid", GCPProjectNumber: "npid", GCPLocation: "location", GCPCluster: "cluster", GCEInstance: "instanceName", |
| GCEInstanceID: "instance", GCEInstanceCreatedBy: "createdBy", |
| GCPClusterURL: "https://container.googleapis.com/v1/projects/pid/locations/location/clusters/cluster", |
| }, |
| }, |
| { |
| "instance created by error", |
| func() bool { return true }, |
| func() (string, error) { return "pid", nil }, |
| func() (string, error) { return "npid", nil }, |
| func() (string, error) { return "location", nil }, |
| func() (string, error) { return "cluster", nil }, |
| func() (string, error) { return "instanceName", nil }, |
| func() (string, error) { return "instance", nil }, |
| func() (string, error) { return "instanceTemplate", nil }, |
| func() (string, error) { return "", errors.New("error") }, |
| map[string]string{}, |
| map[string]string{ |
| GCPProject: "pid", GCPProjectNumber: "npid", GCPLocation: "location", GCPCluster: "cluster", GCEInstance: "instanceName", |
| GCEInstanceID: "instance", GCEInstanceTemplate: "instanceTemplate", |
| GCPClusterURL: "https://container.googleapis.com/v1/projects/pid/locations/location/clusters/cluster", |
| }, |
| }, |
| { |
| "use env variable", |
| func() bool { return true }, |
| func() (string, error) { return "pid", nil }, |
| func() (string, error) { return "npid", nil }, |
| func() (string, error) { return "location", nil }, |
| func() (string, error) { return "cluster", nil }, |
| func() (string, error) { return "instanceName", nil }, |
| func() (string, error) { return "instance", nil }, |
| func() (string, error) { return "instanceTemplate", nil }, |
| func() (string, error) { return "createdBy", nil }, |
| map[string]string{"GCP_METADATA": "env_pid|env_pn|env_cluster|env_location"}, |
| map[string]string{ |
| GCPProject: "env_pid", GCPProjectNumber: "env_pn", GCPLocation: "env_location", GCPCluster: "env_cluster", |
| GCEInstance: "instanceName", GCEInstanceID: "instance", GCEInstanceTemplate: "instanceTemplate", GCEInstanceCreatedBy: "createdBy", |
| GCPClusterURL: "https://container.googleapis.com/v1/projects/env_pid/locations/env_location/clusters/env_cluster", |
| }, |
| }, |
| } |
| |
| for idx, tt := range tests { |
| t.Run(fmt.Sprintf("[%d] %s", idx, tt.name), func(t *testing.T) { |
| for e, v := range tt.env { |
| os.Setenv(e, v) |
| if e == "GCP_METADATA" { |
| GCPMetadata = v |
| } |
| } |
| shouldFillMetadata, projectIDFn, numericProjectIDFn, clusterLocationFn, clusterNameFn, |
| instanceNameFn, instanceIDFn, instanceTemplateFn, createdByFn = tt.shouldFill, tt.projectIDFn, tt.numericProjectIDFn, tt.locationFn, tt.clusterNameFn, |
| tt.instanceNameFn, tt.instanceIDFn, tt.instanceTemplateFn, tt.instanceCreatedByFn |
| e := NewGCP() |
| got := e.Metadata() |
| if !reflect.DeepEqual(got, tt.want) { |
| t.Errorf("gcpEnv.Metadata() => '%v'; want '%v'", got, tt.want) |
| } |
| for e := range tt.env { |
| os.Unsetenv(e) |
| if e == "GCP_METADATA" { |
| GCPMetadata = "" |
| } |
| } |
| envOnce, envPid, envNpid, envCluster, envLocation = sync.Once{}, "", "", "", "" |
| }) |
| } |
| } |
| |
| func TestGCPQuotaProject(t *testing.T) { |
| cases := []struct { |
| name string |
| quotaProject string |
| wantFound bool |
| wantProject string |
| }{ |
| {"no value set", "", false, ""}, |
| {"value set", "234234323", true, "234234323"}, |
| } |
| for _, v := range cases { |
| t.Run(v.name, func(tt *testing.T) { |
| shouldFillMetadata = func() bool { return true } |
| tmpQuotaProject := GCPQuotaProjectVar |
| GCPQuotaProjectVar = v.quotaProject |
| defer func() { |
| GCPQuotaProjectVar = tmpQuotaProject |
| shouldFillMetadata = metadata.OnGCE |
| }() |
| meta := NewGCP().Metadata() |
| val, found := meta[GCPQuotaProject] |
| if got, want := found, v.wantFound; got != want { |
| tt.Errorf("Metadata() returned an unexpected value for GCPQuotaProject; found value: %t, want %t", got, want) |
| } |
| if got, want := val, v.wantProject; got != want { |
| tt.Errorf("Incorrect value for GCPQuotaProject; got = %q, want = %q", got, want) |
| } |
| }) |
| } |
| } |
| |
| func TestMetadataCache(t *testing.T) { |
| tests := []struct { |
| name string |
| shouldFill shouldFillFn |
| projectIDFn metadataFn |
| numericProjectIDFn metadataFn |
| locationFn metadataFn |
| clusterNameFn metadataFn |
| instanceNameFn metadataFn |
| instanceIDFn metadataFn |
| instanceTemplateFn metadataFn |
| instanceCreatedByFn metadataFn |
| env map[string]string |
| want map[string]string |
| }{ |
| { |
| "should cache", |
| func() bool { return true }, |
| func() (string, error) { return "pid", nil }, |
| func() (string, error) { return "npid", nil }, |
| func() (string, error) { return "location", nil }, |
| func() (string, error) { return "cluster", nil }, |
| func() (string, error) { return "instanceName", nil }, |
| func() (string, error) { return "instance", nil }, |
| func() (string, error) { return "instanceTemplate", nil }, |
| func() (string, error) { return "createdBy", nil }, |
| map[string]string{}, |
| map[string]string{ |
| GCPProject: "pid", GCPProjectNumber: "npid", GCPLocation: "location", GCPCluster: "cluster", GCEInstance: "instanceName", |
| GCEInstanceID: "instance", GCEInstanceTemplate: "instanceTemplate", GCEInstanceCreatedBy: "createdBy", |
| GCPClusterURL: "https://container.googleapis.com/v1/projects/pid/locations/location/clusters/cluster", |
| }, |
| }, { |
| "should ignore", |
| func() bool { return true }, |
| func() (string, error) { return "newPid", nil }, |
| func() (string, error) { return "newNpid", nil }, |
| func() (string, error) { return "newLocation", nil }, |
| func() (string, error) { return "newCluster", nil }, |
| func() (string, error) { return "newInstanceName", nil }, |
| func() (string, error) { return "newInstance", nil }, |
| func() (string, error) { return "newInstanceTemplate", nil }, |
| func() (string, error) { return "newCreatedBy", nil }, |
| map[string]string{}, |
| map[string]string{ |
| GCPProject: "pid", GCPProjectNumber: "npid", GCPLocation: "location", GCPCluster: "cluster", GCEInstance: "instanceName", |
| GCEInstanceID: "instance", GCEInstanceTemplate: "instanceTemplate", GCEInstanceCreatedBy: "createdBy", |
| GCPClusterURL: "https://container.googleapis.com/v1/projects/pid/locations/location/clusters/cluster", |
| }, |
| }, |
| } |
| gcpEnv := NewGCP() |
| for idx, tt := range tests { |
| t.Run(fmt.Sprintf("[%d] %s", idx, tt.name), func(t *testing.T) { |
| shouldFillMetadata, projectIDFn, numericProjectIDFn, clusterLocationFn, clusterNameFn, |
| instanceNameFn, instanceIDFn, instanceTemplateFn, createdByFn = tt.shouldFill, tt.projectIDFn, tt.numericProjectIDFn, tt.locationFn, tt.clusterNameFn, |
| tt.instanceNameFn, tt.instanceIDFn, tt.instanceTemplateFn, tt.instanceCreatedByFn |
| got := gcpEnv.Metadata() |
| if !reflect.DeepEqual(got, tt.want) { |
| t.Errorf("gcpEnv.Metadata() => '%v'; want '%v'", got, tt.want) |
| } |
| envOnce, envPid, envNpid, envCluster, envLocation = sync.Once{}, "", "", "", "" |
| }) |
| } |
| } |