From e9345ae5f05163adc84473ddaa1f819d62fe0927 Mon Sep 17 00:00:00 2001 From: Jianfei Bai Date: Sun, 3 Nov 2019 11:45:36 +0800 Subject: [PATCH] Move funcs of nodeutil to cloud/nodes.go. --- test/e2e/cloud/BUILD | 1 + test/e2e/cloud/nodes.go | 8 +++++- test/e2e/framework/nodes_util.go | 42 -------------------------------- 3 files changed, 8 insertions(+), 43 deletions(-) diff --git a/test/e2e/cloud/BUILD b/test/e2e/cloud/BUILD index c12a26927c3..7672be48b05 100644 --- a/test/e2e/cloud/BUILD +++ b/test/e2e/cloud/BUILD @@ -9,6 +9,7 @@ go_library( importpath = "k8s.io/kubernetes/test/e2e/cloud", visibility = ["//visibility:public"], deps = [ + "//staging/src/k8s.io/api/core/v1:go_default_library", "//staging/src/k8s.io/apimachinery/pkg/api/errors:go_default_library", "//staging/src/k8s.io/apimachinery/pkg/apis/meta/v1:go_default_library", "//staging/src/k8s.io/client-go/kubernetes:go_default_library", diff --git a/test/e2e/cloud/nodes.go b/test/e2e/cloud/nodes.go index a1be21a592e..9a4432746c1 100644 --- a/test/e2e/cloud/nodes.go +++ b/test/e2e/cloud/nodes.go @@ -19,6 +19,7 @@ package cloud import ( "time" + v1 "k8s.io/api/core/v1" apierrs "k8s.io/apimachinery/pkg/api/errors" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" clientset "k8s.io/client-go/kubernetes" @@ -55,7 +56,7 @@ var _ = SIGDescribe("[Feature:CloudProvider][Disruptive] Nodes", func() { framework.Logf("Original number of ready nodes: %d", len(origNodes.Items)) - err = framework.DeleteNodeOnCloudProvider(nodeToDelete) + err = deleteNodeOnCloudProvider(nodeToDelete) if err != nil { framework.Failf("failed to delete node %q, err: %q", nodeToDelete.Name, err) } @@ -73,3 +74,8 @@ var _ = SIGDescribe("[Feature:CloudProvider][Disruptive] Nodes", func() { }) }) + +// DeleteNodeOnCloudProvider deletes the specified node. +func deleteNodeOnCloudProvider(node *v1.Node) error { + return framework.TestContext.CloudConfig.Provider.DeleteNode(node) +} diff --git a/test/e2e/framework/nodes_util.go b/test/e2e/framework/nodes_util.go index cb38859f6f3..9b2665b2dbc 100644 --- a/test/e2e/framework/nodes_util.go +++ b/test/e2e/framework/nodes_util.go @@ -281,43 +281,6 @@ func nodePoolsGKE() ([]string, error) { return strings.Fields(stdout), nil } -// MigTemplate (GCE-only) returns the name of the MIG template that the -// nodes of the cluster use. -func MigTemplate() (string, error) { - var errLast error - var templ string - key := "instanceTemplate" - if wait.Poll(Poll, SingleCallTimeout, func() (bool, error) { - // TODO(mikedanese): make this hit the compute API directly instead of - // shelling out to gcloud. - // An `instance-groups managed describe` call outputs what we want to stdout. - output, _, err := retryCmd("gcloud", "compute", "instance-groups", "managed", - fmt.Sprintf("--project=%s", TestContext.CloudConfig.ProjectID), - "describe", - fmt.Sprintf("--zone=%s", TestContext.CloudConfig.Zone), - TestContext.CloudConfig.NodeInstanceGroup) - if err != nil { - errLast = fmt.Errorf("gcloud compute instance-groups managed describe call failed with err: %v", err) - return false, nil - } - - // The 'describe' call probably succeeded; parse the output and try to - // find the line that looks like "instanceTemplate: url/to/" and - // return . - if val := ParseKVLines(output, key); len(val) > 0 { - url := strings.Split(val, "/") - templ = url[len(url)-1] - Logf("MIG group %s using template: %s", TestContext.CloudConfig.NodeInstanceGroup, templ) - return true, nil - } - errLast = fmt.Errorf("couldn't find %s in output to get MIG template. Output: %s", key, output) - return false, nil - }) != nil { - return "", fmt.Errorf("MigTemplate() failed with last error: %v", errLast) - } - return templ, nil -} - func gceUpgradeScript() string { if len(TestContext.GCEUpgradeScript) == 0 { return path.Join(TestContext.RepoRoot, "cluster/gce/upgrade.sh") @@ -401,8 +364,3 @@ func (k *NodeKiller) kill(nodes []v1.Node) { } wg.Wait() } - -// DeleteNodeOnCloudProvider deletes the specified node. -func DeleteNodeOnCloudProvider(node *v1.Node) error { - return TestContext.CloudConfig.Provider.DeleteNode(node) -}