From a2399064a77170a610dbb597483aae7a75ab749c Mon Sep 17 00:00:00 2001 From: Pavithra Ramesh Date: Wed, 29 Jan 2020 12:37:35 -0800 Subject: [PATCH] Allow update of onXPN field in fake GCE clients. --- staging/src/k8s.io/legacy-cloud-providers/gce/gce_fake.go | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/staging/src/k8s.io/legacy-cloud-providers/gce/gce_fake.go b/staging/src/k8s.io/legacy-cloud-providers/gce/gce_fake.go index 09a516c6550..72936211fdc 100644 --- a/staging/src/k8s.io/legacy-cloud-providers/gce/gce_fake.go +++ b/staging/src/k8s.io/legacy-cloud-providers/gce/gce_fake.go @@ -35,6 +35,7 @@ type TestClusterValues struct { SecondaryZoneName string ClusterID string ClusterName string + OnXPN bool } // DefaultTestClusterValues Creates a reasonable set of default cluster values @@ -74,8 +75,14 @@ func NewFakeGCECloud(vals TestClusterValues) *Cloud { projectID: vals.ProjectID, networkProjectID: vals.ProjectID, ClusterID: fakeClusterID(vals.ClusterID), + onXPN: vals.OnXPN, } c := cloud.NewMockGCE(&gceProjectRouter{gce}) gce.c = c 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 +}