@@ -10,6 +10,7 @@ import (
1010 "testing"
1111
1212 "github.com/pion/logging"
13+ "github.com/stretchr/testify/assert"
1314)
1415
1516func testNoDebugLevel (t * testing.T , logger * logging.DefaultLeveledLogger ) {
@@ -19,13 +20,10 @@ func testNoDebugLevel(t *testing.T, logger *logging.DefaultLeveledLogger) {
1920 logger .WithOutput (& outBuf )
2021
2122 logger .Debug ("this shouldn't be logged" )
22- if outBuf .Len () > 0 {
23- t .Error ("Debug was logged when it shouldn't have been" )
24- }
23+ assert .GreaterOrEqual (t , 0 , outBuf .Len (), "Debug was logged when it shouldn't have been" )
24+
2525 logger .Debugf ("this shouldn't be logged" )
26- if outBuf .Len () > 0 {
27- t .Error ("Debug was logged when it shouldn't have been" )
28- }
26+ assert .GreaterOrEqual (t , 0 , outBuf .Len (), "Debug was logged when it shouldn't have been" )
2927}
3028
3129func testDebugLevel (t * testing.T , logger * logging.DefaultLeveledLogger ) {
@@ -36,13 +34,14 @@ func testDebugLevel(t *testing.T, logger *logging.DefaultLeveledLogger) {
3634
3735 dbgMsg := "this is a debug message"
3836 logger .Debug (dbgMsg )
39- if ! strings .Contains (outBuf .String (), dbgMsg ) {
40- t .Errorf ("Expected to find %q in %q, but didn't" , dbgMsg , outBuf .String ())
41- }
37+ assert .Truef (t , strings .Contains (outBuf .String (), dbgMsg ),
38+ "Expected to find %q in %q, but didn't" , dbgMsg , outBuf .String ())
39+ assert .Truef (t , strings .Contains (outBuf .String (), dbgMsg ),
40+ "Expected to find %q in %q, but didn't" , dbgMsg , outBuf .String ())
41+
4242 logger .Debugf (dbgMsg ) // nolint: govet
43- if ! strings .Contains (outBuf .String (), dbgMsg ) {
44- t .Errorf ("Expected to find %q in %q, but didn't" , dbgMsg , outBuf .String ())
45- }
43+ assert .Truef (t , strings .Contains (outBuf .String (), dbgMsg ),
44+ "Expected to find %q in %q, but didn't" , dbgMsg , outBuf .String ())
4645}
4746
4847func testWarnLevel (t * testing.T , logger * logging.DefaultLeveledLogger ) {
@@ -53,13 +52,12 @@ func testWarnLevel(t *testing.T, logger *logging.DefaultLeveledLogger) {
5352
5453 warnMsg := "this is a warning message"
5554 logger .Warn (warnMsg )
56- if ! strings .Contains (outBuf .String (), warnMsg ) {
57- t . Errorf ( "Expected to find %q in %q, but didn't" , warnMsg , outBuf .String ())
58- }
55+ assert . Truef ( t , strings .Contains (outBuf .String (), warnMsg ),
56+ "Expected to find %q in %q, but didn't" , warnMsg , outBuf .String ())
57+
5958 logger .Warnf (warnMsg ) // nolint: govet
60- if ! strings .Contains (outBuf .String (), warnMsg ) {
61- t .Errorf ("Expected to find %q in %q, but didn't" , warnMsg , outBuf .String ())
62- }
59+ assert .Truef (t , strings .Contains (outBuf .String (), warnMsg ),
60+ "Expected to find %q in %q, but didn't" , warnMsg , outBuf .String ())
6361}
6462
6563func testErrorLevel (t * testing.T , logger * logging.DefaultLeveledLogger ) {
@@ -70,13 +68,12 @@ func testErrorLevel(t *testing.T, logger *logging.DefaultLeveledLogger) {
7068
7169 errMsg := "this is an error message"
7270 logger .Error (errMsg )
73- if ! strings .Contains (outBuf .String (), errMsg ) {
74- t . Errorf ( "Expected to find %q in %q, but didn't" , errMsg , outBuf .String ())
75- }
71+ assert . Truef ( t , strings .Contains (outBuf .String (), errMsg ),
72+ "Expected to find %q in %q but didn't" , errMsg , outBuf .String ())
73+
7674 logger .Errorf (errMsg ) // nolint: govet
77- if ! strings .Contains (outBuf .String (), errMsg ) {
78- t .Errorf ("Expected to find %q in %q, but didn't" , errMsg , outBuf .String ())
79- }
75+ assert .Truef (t , strings .Contains (outBuf .String (), errMsg ),
76+ "Expected to find %q in %q but didn't" , errMsg , outBuf .String ())
8077}
8178
8279func testTraceLevel (t * testing.T , logger * logging.DefaultLeveledLogger ) {
@@ -87,13 +84,12 @@ func testTraceLevel(t *testing.T, logger *logging.DefaultLeveledLogger) {
8784
8885 traceMsg := "trace message"
8986 logger .Trace (traceMsg )
90- if ! strings .Contains (outBuf .String (), traceMsg ) {
91- t . Errorf ( "Expected to find %q in %q, but didn't" , traceMsg , outBuf .String ())
92- }
87+ assert . Truef ( t , strings .Contains (outBuf .String (), traceMsg ),
88+ "Expected to find %q in %q but didn't" , traceMsg , outBuf .String ())
89+
9390 logger .Tracef (traceMsg ) // nolint: govet
94- if ! strings .Contains (outBuf .String (), traceMsg ) {
95- t .Errorf ("Expected to find %q in %q, but didn't" , traceMsg , outBuf .String ())
96- }
91+ assert .Truef (t , strings .Contains (outBuf .String (), traceMsg ),
92+ "Expected to find %q in %q but didn't" , traceMsg , outBuf .String ())
9793}
9894
9995func testInfoLevel (t * testing.T , logger * logging.DefaultLeveledLogger ) {
@@ -104,13 +100,12 @@ func testInfoLevel(t *testing.T, logger *logging.DefaultLeveledLogger) {
104100
105101 infoMsg := "info message"
106102 logger .Info (infoMsg )
107- if ! strings .Contains (outBuf .String (), infoMsg ) {
108- t . Errorf ( "Expected to find %q in %q, but didn't" , infoMsg , outBuf .String ())
109- }
103+ assert . Truef ( t , strings .Contains (outBuf .String (), infoMsg ),
104+ "Expected to find %q in %q but didn't" , infoMsg , outBuf .String ())
105+
110106 logger .Infof (infoMsg ) // nolint: govet
111- if ! strings .Contains (outBuf .String (), infoMsg ) {
112- t .Errorf ("Expected to find %q in %q, but didn't" , infoMsg , outBuf .String ())
113- }
107+ assert .Truef (t , strings .Contains (outBuf .String (), infoMsg ),
108+ "Expected to find %q in %q but didn't" , infoMsg , outBuf .String ())
114109}
115110
116111func testAllLevels (t * testing.T , logger * logging.DefaultLeveledLogger ) {
@@ -134,18 +129,14 @@ func TestDefaultLoggerFactory(t *testing.T) {
134129
135130 logger := factory .NewLogger ("baz" )
136131 bazLogger , ok := logger .(* logging.DefaultLeveledLogger )
137- if ! ok {
138- t .Error ("Invalid logger type" )
139- }
132+ assert .True (t , ok , "Invalid logger type" )
140133
141134 testNoDebugLevel (t , bazLogger )
142135 testWarnLevel (t , bazLogger )
143136
144137 logger = factory .NewLogger ("foo" )
145138 fooLogger , ok := logger .(* logging.DefaultLeveledLogger )
146- if ! ok {
147- t .Error ("Invalid logger type" )
148- }
139+ assert .True (t , ok , "Invalid logger type" )
149140
150141 testDebugLevel (t , fooLogger )
151142}
@@ -170,34 +161,22 @@ func TestNewDefaultLoggerFactory(t *testing.T) {
170161 traceLevel := factory .NewLogger ("TRACE" )
171162
172163 disabledLogger , ok := disabled .(* logging.DefaultLeveledLogger )
173- if ! ok {
174- t .Error ("Missing disabled logger" )
175- }
164+ assert .True (t , ok , "Missing disabled logger" )
176165
177166 errorLogger , ok := errorLevel .(* logging.DefaultLeveledLogger )
178- if ! ok {
179- t .Error ("Missing error logger" )
180- }
167+ assert .True (t , ok , "Missing error logger" )
181168
182169 warnLogger , ok := warnLevel .(* logging.DefaultLeveledLogger )
183- if ! ok {
184- t .Error ("Missing warn logger" )
185- }
170+ assert .True (t , ok , "Missing warn logger" )
186171
187172 infoLogger , ok := infoLevel .(* logging.DefaultLeveledLogger )
188- if ! ok {
189- t .Error ("Missing info logger" )
190- }
173+ assert .True (t , ok , "Missing info logger" )
191174
192175 debugLogger , ok := debugLevel .(* logging.DefaultLeveledLogger )
193- if ! ok {
194- t .Error ("Missing debug logger" )
195- }
176+ assert .True (t , ok , "Missing debug logger" )
196177
197178 traceLogger , ok := traceLevel .(* logging.DefaultLeveledLogger )
198- if ! ok {
199- t .Error ("Missing trace logger" )
200- }
179+ assert .True (t , ok , "Missing trace logger" )
201180
202181 testNoDebugLevel (t , disabledLogger )
203182 testNoDebugLevel (t , errorLogger )
@@ -217,9 +196,7 @@ func TestNewDefaultLoggerFactoryLogAll(t *testing.T) {
217196 factory := logging .NewDefaultLoggerFactory ()
218197
219198 testAPI , ok := factory .NewLogger ("test" ).(* logging.DefaultLeveledLogger )
220- if ! ok {
221- t .Error ("Invalid logger factory type" )
222- }
199+ assert .True (t , ok , "Invalid logger factory type" )
223200
224201 testAllLevels (t , testAPI )
225202}
@@ -230,19 +207,13 @@ func TestNewDefaultLoggerFactorySpecifcScopes(t *testing.T) {
230207 factory := logging .NewDefaultLoggerFactory ()
231208
232209 feature , ok := factory .NewLogger ("feature" ).(* logging.DefaultLeveledLogger )
233- if ! ok {
234- t .Error ("Invalid logger factory type" )
235- }
210+ assert .True (t , ok , "Invalid logger factory type" )
236211
237212 rtp , ok := factory .NewLogger ("rtp-logger" ).(* logging.DefaultLeveledLogger )
238- if ! ok {
239- t .Error ("Invalid logger factory type" )
240- }
213+ assert .True (t , ok , "Invalid logger factory type" )
241214
242215 noScope , ok := factory .NewLogger ("no-scope" ).(* logging.DefaultLeveledLogger )
243- if ! ok {
244- t .Error ("Invalid logger factory type" )
245- }
216+ assert .True (t , ok , "Invalid logger factory type" )
246217
247218 testDebugLevel (t , feature )
248219 testDebugLevel (t , rtp )
@@ -262,9 +233,7 @@ func TestLogLevel(t *testing.T) {
262233 logLevel := logging .LogLevelDisabled
263234
264235 logLevel .Set (logging .LogLevelError )
265- if logLevel .Get () != logging .LogLevelError {
266- t .Error ("LogLevel was not set to LogLevelError" )
267- }
236+ assert .Equal (t , logging .LogLevelError , logLevel .Get (), "LogLevel was not set to LogLevelError" )
268237}
269238
270239func TestLogLevelString (t * testing.T ) {
@@ -279,9 +248,7 @@ func TestLogLevelString(t *testing.T) {
279248 }
280249
281250 for level , expectedStr := range expected {
282- if level .String () != expectedStr {
283- t .Errorf ("Expected %q, got %q" , expectedStr , level .String ())
284- }
251+ assert .Equal (t , expectedStr , level .String ())
285252 }
286253}
287254
0 commit comments