github/kubernetes
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow update of onXPN field in fake GCE clients.

Browse Source
This commit is contained in:
Pavithra Ramesh 2020-01-29 12:37:35 -08:00
parent edbd51bc69
commit a2399064a7
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
staging/src/k8s.io/legacy-cloud-providers/gce/gce_fake.go
View File

@ -35,6 +35,7 @@ type TestClusterValues struct {
SecondaryZoneName string SecondaryZoneName string
ClusterID string ClusterID string
ClusterName string ClusterName string
OnXPN bool
} }
// DefaultTestClusterValues Creates a reasonable set of default cluster values // DefaultTestClusterValues Creates a reasonable set of default cluster values
@ -74,8 +75,14 @@ func NewFakeGCECloud(vals TestClusterValues) *Cloud {
projectID: vals.ProjectID, projectID: vals.ProjectID,
networkProjectID: vals.ProjectID, networkProjectID: vals.ProjectID,
ClusterID: fakeClusterID(vals.ClusterID), ClusterID: fakeClusterID(vals.ClusterID),
onXPN: vals.OnXPN,
} }
c := cloud.NewMockGCE(&gceProjectRouter{gce}) c := cloud.NewMockGCE(&gceProjectRouter{gce})
gce.c = c gce.c = c
return gce return gce
} }
// UpdateFakeGCECloud updates the fake GCE cloud with the specified values. Currently only the onXPN value is updated.
func UpdateFakeGCECloud(g *Cloud, vals TestClusterValues) {
g.onXPN = vals.OnXPN
}