feat: allow controller-runtime log level in tests to be set via LOG_LEVEL

Author: kekcleader

cmd/listener.go

@@ -3,11 +3,13 @@ package cmd
 import (
 	"crypto/tls"
 	"os"
+	"strings"
 
 	kcpapis "github.com/kcp-dev/kcp/sdk/apis/apis/v1alpha1"
 	kcpcore "github.com/kcp-dev/kcp/sdk/apis/core/v1alpha1"
 	kcptenancy "github.com/kcp-dev/kcp/sdk/apis/tenancy/v1alpha1"
 	"github.com/spf13/cobra"
+	"go.uber.org/zap/zapcore"
 	apiextensionsv1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
 	"k8s.io/apimachinery/pkg/runtime"
 	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
@@ -41,10 +43,20 @@ var listenCmd = &cobra.Command{
 		utilruntime.Must(kcptenancy.AddToScheme(scheme))
 		utilruntime.Must(apiextensionsv1.AddToScheme(scheme))
 
-		opts := zap.Options{
-			Development: true,
+		var zapLevel zapcore.Level
+		switch strings.ToUpper(log.GetLevel().String()) {
+		case "ERROR":
+			zapLevel = zapcore.ErrorLevel
+		case "WARN":
+			zapLevel = zapcore.WarnLevel
+		case "INFO":
+			zapLevel = zapcore.InfoLevel
+		case "DEBUG", "TRACE":
+			zapLevel = zapcore.DebugLevel
+		default:
+			zapLevel = zapcore.InfoLevel
 		}
-		ctrl.SetLogger(zap.New(zap.UseFlagOptions(&opts)))
+		ctrl.SetLogger(zap.New(zap.UseDevMode(false), zap.Level(zapLevel)))
 
 		disableHTTP2 := func(c *tls.Config) {
 			log.Info().Msg("disabling http/2")

tests/gateway_test/suite_test.go

@@ -1,24 +1,29 @@
 package gateway_test
 
 import (
-	appsv1 "k8s.io/api/apps/v1"
-	corev1 "k8s.io/api/core/v1"
-	v1 "k8s.io/api/rbac/v1"
-	"k8s.io/apimachinery/pkg/runtime"
 	"net/http/httptest"
 	"os"
 	"path/filepath"
+	"strings"
 	"testing"
 
+	appsv1 "k8s.io/api/apps/v1"
+	corev1 "k8s.io/api/core/v1"
+	v1 "k8s.io/api/rbac/v1"
+	"k8s.io/apimachinery/pkg/runtime"
+
 	"github.com/graphql-go/graphql"
 	"github.com/openmfp/golang-commons/logger"
 	"github.com/stretchr/testify/require"
 	"github.com/stretchr/testify/suite"
+	"go.uber.org/zap/zapcore"
 	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
 	"k8s.io/client-go/rest"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 	"sigs.k8s.io/controller-runtime/pkg/envtest"
 	"sigs.k8s.io/controller-runtime/pkg/kcp"
+	"sigs.k8s.io/controller-runtime/pkg/log"
+	"sigs.k8s.io/controller-runtime/pkg/log/zap"
 
 	"github.com/openmfp/account-operator/api/v1alpha1"
 	appConfig "github.com/openmfp/kubernetes-graphql-gateway/common/config"
@@ -27,6 +32,25 @@ import (
 	"github.com/openmfp/kubernetes-graphql-gateway/gateway/schema"
 )
 
+func TestMain(m *testing.M) {
+	var zapLevel zapcore.Level
+	level := strings.ToUpper(os.Getenv("LOG_LEVEL"))
+	switch level {
+	case "ERROR":
+		zapLevel = zapcore.ErrorLevel
+	case "WARN":
+		zapLevel = zapcore.WarnLevel
+	case "INFO":
+		zapLevel = zapcore.InfoLevel
+	case "DEBUG", "TRACE":
+		zapLevel = zapcore.DebugLevel
+	default:
+		zapLevel = zapcore.ErrorLevel
+	}
+	log.SetLogger(zap.New(zap.UseDevMode(false), zap.Level(zapLevel)))
+	os.Exit(m.Run())
+}
+
 type CommonTestSuite struct {
 	suite.Suite
 	testEnv       *envtest.Environment