fix test logging

PeerDB-io · Nov 15, 2023 · dcd9bea · dcd9bea
1 parent 7013eb1
commit dcd9bea
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/flow/e2e/snowflake/peer_flow_sf_test.go b/flow/e2e/snowflake/peer_flow_sf_test.go
@@ -97,6 +97,13 @@ func (s *PeerFlowE2ETestSuiteSF) setupTemporalLogger() {
 	s.SetLogger(tlogger)
 }
 
+type logWriterType struct{ t *testing.T }
+
+func (l logWriterType) Write(p []byte) (n int, err error) {
+	l.t.Logf(string(p))
+	return len(p), nil
+}
+
 func (s *PeerFlowE2ETestSuiteSF) SetupSuite() {
 	err := godotenv.Load()
 	if err != nil {
@@ -107,6 +114,7 @@ func (s *PeerFlowE2ETestSuiteSF) SetupSuite() {
 
 	log.SetReportCaller(true)
 	log.SetLevel(log.WarnLevel)
+	log.SetOutput(logWriterType{t: s.T()})
 
 	s.setupTemporalLogger()