From a29fd607baa0ad97e1f0937261dc11aea8a4fda8 Mon Sep 17 00:00:00 2001
From: Menghan Li <menghanl@google.com>
Date: Wed, 15 Sep 2021 13:51:20 -0700
Subject: [PATCH] [tlogger_error] tlogger: print log type

Error logs cause tests to fail. This makes it easier (possible) to find
the error log
---
 internal/grpctest/tlogger.go | 35 +++++++++++++++++++++++++----------
 1 file changed, 25 insertions(+), 10 deletions(-)

diff --git a/internal/grpctest/tlogger.go b/internal/grpctest/tlogger.go
index f1712245ec4a..19def8a49a83 100644
--- a/internal/grpctest/tlogger.go
+++ b/internal/grpctest/tlogger.go
@@ -41,8 +41,23 @@ const callingFrame = 4
 
 type logType int
 
+func (l logType) String() string {
+	switch l {
+	case infoLog:
+		return "INFO"
+	case warningLog:
+		return "WARN"
+	case errorLog:
+		return "ERRO"
+	case fatalLog:
+		return "FATA"
+	}
+	return "UNKNOWN"
+}
+
 const (
-	logLog logType = iota
+	infoLog logType = iota
+	warningLog
 	errorLog
 	fatalLog
 )
@@ -83,7 +98,7 @@ func (g *tLogger) log(ltype logType, depth int, format string, args ...interface
 		g.t.Error(err)
 		return
 	}
-	args = append([]interface{}{prefix}, args...)
+	args = append([]interface{}{ltype.String(), prefix}, args...)
 	args = append(args, fmt.Sprintf(" (t=+%s)", time.Since(g.start)))
 
 	if format == "" {
@@ -180,35 +195,35 @@ func (g *tLogger) expected(s string) bool {
 }
 
 func (g *tLogger) Info(args ...interface{}) {
-	g.log(logLog, 0, "", args...)
+	g.log(infoLog, 0, "", args...)
 }
 
 func (g *tLogger) Infoln(args ...interface{}) {
-	g.log(logLog, 0, "", args...)
+	g.log(infoLog, 0, "", args...)
 }
 
 func (g *tLogger) Infof(format string, args ...interface{}) {
-	g.log(logLog, 0, format, args...)
+	g.log(infoLog, 0, format, args...)
 }
 
 func (g *tLogger) InfoDepth(depth int, args ...interface{}) {
-	g.log(logLog, depth, "", args...)
+	g.log(infoLog, depth, "", args...)
 }
 
 func (g *tLogger) Warning(args ...interface{}) {
-	g.log(logLog, 0, "", args...)
+	g.log(warningLog, 0, "", args...)
 }
 
 func (g *tLogger) Warningln(args ...interface{}) {
-	g.log(logLog, 0, "", args...)
+	g.log(warningLog, 0, "", args...)
 }
 
 func (g *tLogger) Warningf(format string, args ...interface{}) {
-	g.log(logLog, 0, format, args...)
+	g.log(warningLog, 0, format, args...)
 }
 
 func (g *tLogger) WarningDepth(depth int, args ...interface{}) {
-	g.log(logLog, depth, "", args...)
+	g.log(warningLog, depth, "", args...)
 }
 
 func (g *tLogger) Error(args ...interface{}) {