replace ktesting with explicit calls

Signed-off-by: Etai Lev Ran <elevran@gmail.com>

Author: elevran

pkg/sidecar/proxy/connector_test.go

@@ -30,7 +30,6 @@ import (
 	"github.com/llm-d/llm-d-inference-scheduler/test/sidecar/mock"
 	. "github.com/onsi/ginkgo/v2" // nolint:revive
 	. "github.com/onsi/gomega"    // nolint:revive
-	"k8s.io/klog/v2/ktesting"
 )
 
 type sidecarTestInfo struct {
@@ -180,7 +179,7 @@ var _ = Describe("Common Connector tests", func() {
 func sidecarConnectionTestSetup(connector string) *sidecarTestInfo {
 	testInfo := sidecarTestInfo{}
 
-	_, testInfo.ctx = ktesting.NewTestContext(GinkgoT())
+	testInfo.ctx = newTestContext()
 	testInfo.ctx, testInfo.cancelFn = context.WithCancel(testInfo.ctx)
 	testInfo.stoppedCh = make(chan struct{})
 

pkg/sidecar/proxy/data_parallel_test.go

@@ -10,7 +10,6 @@ import (
 	. "github.com/onsi/ginkgo/v2" // nolint:revive
 	. "github.com/onsi/gomega"    // nolint:revive
 	"golang.org/x/sync/errgroup"
-	"k8s.io/klog/v2/ktesting"
 
 	"github.com/llm-d/llm-d-inference-scheduler/pkg/common"
 	sidecarmock "github.com/llm-d/llm-d-inference-scheduler/test/sidecar/mock"
@@ -24,7 +23,7 @@ const (
 var _ = Describe("Data Parallel support", func() {
 	When("configured with --data-parallel-size > 1", func() {
 		It("should create an extra proxy", func() {
-			_, ctx := ktesting.NewTestContext(GinkgoT())
+			ctx := newTestContext()
 			ctx, cancel := context.WithCancel(ctx)
 			grp, ctx := errgroup.WithContext(ctx)
 

pkg/sidecar/proxy/proxy_test.go

@@ -26,20 +26,34 @@ import (
 	"strings"
 	"time"
 
+	"sigs.k8s.io/controller-runtime/pkg/log"
+	ctrllog "sigs.k8s.io/controller-runtime/pkg/log"
+	"sigs.k8s.io/controller-runtime/pkg/log/zap"
+
 	"github.com/llm-d/llm-d-inference-scheduler/pkg/common"
 	"github.com/llm-d/llm-d-inference-scheduler/test/sidecar/mock"
 	. "github.com/onsi/ginkgo/v2" // nolint:revive
 	. "github.com/onsi/gomega"    // nolint:revive
-	"k8s.io/klog/v2/ktesting"
 )
 
+func newTestContext() context.Context {
+	logger := zap.New(
+		zap.WriteTo(GinkgoWriter),
+		zap.UseDevMode(true),
+	)
+	ctrllog.SetLogger(logger)
+	ctx := context.Background()
+	log.IntoContext(ctx, logger) // not strictly needed since we called SetLogger to set default
+	return ctx
+}
+
 var _ = Describe("Reverse Proxy", func() {
 	When("x-prefiller-url is not present", func() {
 		DescribeTable("should forward requests to decode server",
 
 			func(path string, secureProxy bool) {
-				_, ctx := ktesting.NewTestContext(GinkgoT())
 
+				ctx := newTestContext()
 				var cert *tls.Certificate
 				if secureProxy {
 					tempCert, err := CreateSelfSignedTLSCertificate()
@@ -159,7 +173,7 @@ var _ = Describe("Reverse Proxy", func() {
 			})
 
 			It("should successfully send request to 1. prefill 2. decode with the right fields (backward compatible behavior)", func() {
-				_, ctx := ktesting.NewTestContext(GinkgoT())
+				ctx := newTestContext()
 				ctx, cancelFn := context.WithCancel(ctx)
 				stoppedCh := make(chan struct{})
 
@@ -233,7 +247,7 @@ var _ = Describe("Reverse Proxy", func() {
 			})
 
 			It("should successfully send request to 1. prefill 2. decode with the right fields", func() {
-				_, ctx := ktesting.NewTestContext(GinkgoT())
+				ctx := newTestContext()
 				ctx, cancelFn := context.WithCancel(ctx)
 				stoppedCh := make(chan struct{})