Ensures node becomes schedulable at the end of tests that delete nodes

This commit is contained in:
Cheng Xing 2017-06-26 18:41:14 -07:00
parent 35efde9e6d
commit 78d111b75f

View File

@ -429,8 +429,9 @@ var _ = framework.KubeDescribe("Pod Disks", func() {
podClient.Delete(host0Pod.Name, metav1.NewDeleteOptions(0)) podClient.Delete(host0Pod.Name, metav1.NewDeleteOptions(0))
detachAndDeletePDs(diskName, []types.NodeName{host0Name}) detachAndDeletePDs(diskName, []types.NodeName{host0Name})
framework.WaitForNodeToBeReady(f.ClientSet, string(host0Name), nodeStatusTimeout) framework.WaitForNodeToBeReady(f.ClientSet, string(host0Name), nodeStatusTimeout)
framework.WaitForAllNodesSchedulable(f.ClientSet, nodeStatusTimeout)
nodes = framework.GetReadySchedulableNodesOrDie(f.ClientSet) nodes = framework.GetReadySchedulableNodesOrDie(f.ClientSet)
Expect(len(nodes.Items)).To(Equal(initialGroupSize)) Expect(len(nodes.Items)).To(Equal(initialGroupSize), "Requires node count to return to initial group size.")
}() }()
By("submitting host0Pod to kubernetes") By("submitting host0Pod to kubernetes")
@ -480,7 +481,7 @@ var _ = framework.KubeDescribe("Pod Disks", func() {
originalCount := len(nodes.Items) originalCount := len(nodes.Items)
containerName := "mycontainer" containerName := "mycontainer"
nodeToDelete := &nodes.Items[0] nodeToDelete := &nodes.Items[0]
defer func() error { defer func() {
By("Cleaning up PD-RW test env") By("Cleaning up PD-RW test env")
detachAndDeletePDs(diskName, []types.NodeName{host0Name}) detachAndDeletePDs(diskName, []types.NodeName{host0Name})
nodeToDelete.ObjectMeta.SetResourceVersion("0") nodeToDelete.ObjectMeta.SetResourceVersion("0")
@ -489,11 +490,9 @@ var _ = framework.KubeDescribe("Pod Disks", func() {
framework.ExpectNoError(err, "Unable to re-create the deleted node") framework.ExpectNoError(err, "Unable to re-create the deleted node")
framework.ExpectNoError(framework.WaitForGroupSize(framework.TestContext.CloudConfig.NodeInstanceGroup, int32(initialGroupSize)), "Unable to get the node group back to the original size") framework.ExpectNoError(framework.WaitForGroupSize(framework.TestContext.CloudConfig.NodeInstanceGroup, int32(initialGroupSize)), "Unable to get the node group back to the original size")
framework.WaitForNodeToBeReady(f.ClientSet, nodeToDelete.Name, nodeStatusTimeout) framework.WaitForNodeToBeReady(f.ClientSet, nodeToDelete.Name, nodeStatusTimeout)
framework.WaitForAllNodesSchedulable(f.ClientSet, nodeStatusTimeout)
nodes = framework.GetReadySchedulableNodesOrDie(f.ClientSet) nodes = framework.GetReadySchedulableNodesOrDie(f.ClientSet)
if len(nodes.Items) != originalCount { Expect(len(nodes.Items)).To(Equal(originalCount), "Requires node count to return to original node count.")
return fmt.Errorf("The node count is not back to original count")
}
return nil
}() }()
By("submitting host0Pod to kubernetes") By("submitting host0Pod to kubernetes")