diff --git a/test/e2e/framework/node/init/init.go b/test/e2e/framework/node/init/init.go index 653eb6bdebc..a482d9d8da1 100644 --- a/test/e2e/framework/node/init/init.go +++ b/test/e2e/framework/node/init/init.go @@ -30,14 +30,10 @@ import ( func init() { framework.NewFrameworkExtensions = append(framework.NewFrameworkExtensions, func(f *framework.Framework) { - ginkgo.AfterEach(func(ctx context.Context) { - if f.ClientSet == nil { - // Test didn't reach f.BeforeEach, most - // likely because the test got - // skipped. Nothing to check... - return - } - e2enode.AllNodesReady(ctx, f.ClientSet, 7*time.Minute) + ginkgo.BeforeEach(func() { + ginkgo.DeferCleanup(func(ctx context.Context) { + framework.ExpectNoError(e2enode.AllNodesReady(ctx, f.ClientSet, 7*time.Minute)) + }) }) }, )