Function DrainPrimary

tests/utils/nodes/nodes.go:40–84

DrainPrimary drains the node containing the primary pod. It returns the names of the pods that were running on that node

(
	ctx context.Context,
	crudClient client.Client,
	namespace,
	clusterName string,
	timeoutSeconds int,
)

// DrainPrimary drains the node containing the primary pod.
// It returns the names of the pods that were running on that node
func DrainPrimary(
	ctx context.Context,
	crudClient client.Client,
	namespace,
	clusterName string,
	timeoutSeconds int,
) []string {
	var primaryNode string
	var podNames []string
	By("identifying primary node and draining", func() {
		pod, err := clusterutils.GetPrimary(ctx, crudClient, namespace, clusterName)
		Expect(err).ToNot(HaveOccurred())
		primaryNode = pod.Spec.NodeName

		// Gather the pods running on this node
		podList, err := clusterutils.ListPods(ctx, crudClient, namespace, clusterName)
		Expect(err).ToNot(HaveOccurred())
		for _, pod := range podList.Items {
			if pod.Spec.NodeName == primaryNode {
				podNames = append(podNames, pod.Name)
			}
		}

		// Draining the primary pod's node
		var stdout, stderr string
		Eventually(func() error {
			cmd := fmt.Sprintf("kubectl drain %v --ignore-daemonsets --delete-emptydir-data --force --timeout=%ds",
				primaryNode, timeoutSeconds)
			stdout, stderr, err = run.Unchecked(cmd)
			return err
		}, timeoutSeconds).ShouldNot(HaveOccurred(), fmt.Sprintf("stdout: %s, stderr: %s", stdout, stderr))
	})
	By("ensuring no cluster pod is still running on the drained node", func() {
		Eventually(func() ([]string, error) {
			podList, err := clusterutils.ListPods(ctx, crudClient, namespace, clusterName)
			usedNodes := make([]string, 0, len(podList.Items))
			for _, pod := range podList.Items {
				usedNodes = append(usedNodes, pod.Spec.NodeName)
			}
			return usedNodes, err
		}, 60).ShouldNot(ContainElement(primaryNode))
	})

	return podNames
}

// UncordonAll executes the 'kubectl uncordon' command on each node of the list
func UncordonAll(