test/utils/ktesting/tcontext_internal_test.go - master - githubmirror/kubernetes

test/utils/ktesting/tcontext_internal_test.go
99 строк3 KB
Nate Bracy
tests(utils/ktesting): make CleanupGracePeriod configurable in TContext
13 май 2026, 19:08
13 май 2026, 19:08af9e352
/*
Copyright The Kubernetes Authors.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

// This file is in package ktesting (not ktesting_test) so that it can access
// unexported fields of TContext.
package ktesting

import (
	"context"
	"testing"
	"testing/synctest"
	"time"

	"github.com/onsi/gomega"

	"k8s.io/kubernetes/test/utils/ktesting/initoption"
)

// deadlineT2 mirrors the deadlineT helper in ktesting_test.
type deadlineT2 struct {
	TB
	deadline *time.Time
}

func (t *deadlineT2) Deadline() (time.Time, bool) {
	if t.deadline == nil {
		return time.Time{}, false
	}
	return *t.deadline, true
}

// TestDefaultCleanupGracePeriod verifies that cleanupGracePeriod is set to
// DefaultCleanupGracePeriod when no override is given.
func TestDefaultCleanupGracePeriod(t *testing.T) {
	synctest.Test(t, func(t *testing.T) {
		mockDeadline := time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC)
		mockT := &deadlineT2{TB: t, deadline: &mockDeadline}
		tCtx := Init(mockT)
		if tCtx.cleanupGracePeriod != DefaultCleanupGracePeriod {
			t.Errorf("expected cleanupGracePeriod %v, got %v", DefaultCleanupGracePeriod, tCtx.cleanupGracePeriod)
		}
	})
}

// TestCustomCleanupGracePeriod verifies that WithCleanupGracePeriod stores the
// override in cleanupGracePeriod and shifts the effective deadline accordingly.
func TestCustomCleanupGracePeriod(t *testing.T) {
	synctest.Test(t, func(t *testing.T) {
		mockDeadline := time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC)
		mockT := &deadlineT2{TB: t, deadline: &mockDeadline}
		custom := 30 * time.Second
		tCtx := Init(mockT, initoption.WithCleanupGracePeriod(custom))

		if tCtx.cleanupGracePeriod != custom {
			t.Errorf("expected cleanupGracePeriod %v, got %v", custom, tCtx.cleanupGracePeriod)
		}

		actualDeadline, ok := tCtx.Deadline()
		if !ok {
			t.Fatal("expected a deadline but got none")
		}
		expect := mockDeadline.Add(-custom)
		tCtx.Expect(actualDeadline).To(
			gomega.BeTemporally("==", expect),
			"context deadline should be shifted by the custom grace period")
	})
}

// TestInitCtxCleanupGracePeriod verifies that InitCtx applies
// WithCleanupGracePeriod via the previously ignored opts parameter.
func TestInitCtxCleanupGracePeriod(t *testing.T) {
	custom := 42 * time.Second
	tCtx := InitCtx(context.Background(), t, initoption.WithCleanupGracePeriod(custom))
	if tCtx.cleanupGracePeriod != custom {
		t.Errorf("expected cleanupGracePeriod %v, got %v", custom, tCtx.cleanupGracePeriod)
	}
}

// TestInitCtxDefaultCleanupGracePeriod verifies that InitCtx falls back to
// DefaultCleanupGracePeriod when no option is given.
func TestInitCtxDefaultCleanupGracePeriod(t *testing.T) {
	tCtx := InitCtx(context.Background(), t)
	if tCtx.cleanupGracePeriod != DefaultCleanupGracePeriod {
		t.Errorf("expected cleanupGracePeriod %v, got %v", DefaultCleanupGracePeriod, tCtx.cleanupGracePeriod)
	}
}