Suppress logrus output during tests

canonical · Jul 9, 2024 · 0e77aa3 · 0e77aa3
1 parent 092e35e
commit 0e77aa3
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/pkg/kine/drivers/generic_test.go b/pkg/kine/drivers/generic_test.go
@@ -3,14 +3,21 @@ package drivers_test
 import (
 	"context"
 	"fmt"
+	"io"
 	"testing"
 
 	"github.com/canonical/k8s-dqlite/pkg/kine/drivers/generic"
 	"github.com/canonical/k8s-dqlite/pkg/kine/logstructured/sqllog"
 	"github.com/canonical/k8s-dqlite/pkg/kine/server"
 	. "github.com/onsi/gomega"
+	"github.com/sirupsen/logrus"
 )
 
+func init() {
+	logrus.SetOutput(io.Discard)
+	logrus.SetLevel(logrus.FatalLevel)
+}
+
 type makeBackendFunc func(ctx context.Context, tb testing.TB) (server.Backend, *generic.Generic, error)
 
 func testCompaction(t *testing.T, makeBackend makeBackendFunc) {

diff --git a/test/util_test.go b/test/util_test.go
@@ -3,15 +3,22 @@ package test
 import (
 	"context"
 	"fmt"
+	"io"
 	"strings"
 	"testing"
 	"time"
 
 	"github.com/canonical/go-dqlite/app"
 	"github.com/canonical/k8s-dqlite/pkg/kine/endpoint"
+	"github.com/sirupsen/logrus"
 	clientv3 "go.etcd.io/etcd/client/v3"
 )
 
+func init() {
+	logrus.SetOutput(io.Discard)
+	logrus.SetLevel(logrus.FatalLevel)
+}
+
 // newKine spins up a new instance of kine.
 //
 // newKine will create a sqlite or dqlite endpoint based on the provided backendType.