Skip to content

Commit 2d5402f

Browse files
committed
Update lint rules, use testify/assert
1 parent a1a340c commit 2d5402f

File tree

4 files changed

+71
-79
lines changed

4 files changed

+71
-79
lines changed

.golangci.yml

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,16 @@ linters-settings:
1919
recommendations:
2020
- errors
2121
forbidigo:
22+
analyze-types: true
2223
forbid:
2324
- ^fmt.Print(f|ln)?$
2425
- ^log.(Panic|Fatal|Print)(f|ln)?$
2526
- ^os.Exit$
2627
- ^panic$
2728
- ^print(ln)?$
29+
- p: ^testing.T.(Error|Errorf|Fatal|Fatalf|Fail|FailNow)$
30+
pkg: ^testing$
31+
msg: "use testify/assert instead"
2832
varnamelen:
2933
max-distance: 12
3034
min-name-length: 2
@@ -127,9 +131,12 @@ issues:
127131
exclude-dirs-use-default: false
128132
exclude-rules:
129133
# Allow complex tests and examples, better to be self contained
130-
- path: (examples|main\.go|_test\.go)
134+
- path: (examples|main\.go)
131135
linters:
136+
- gocognit
132137
- forbidigo
138+
- path: _test\.go
139+
linters:
133140
- gocognit
134141

135142
# Allow forbidden identifiers in CLI commands

go.mod

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,11 @@
11
module github.com/pion/logging
22

33
go 1.20
4+
5+
require github.com/stretchr/testify v1.10.0
6+
7+
require (
8+
github.com/davecgh/go-spew v1.1.1 // indirect
9+
github.com/pmezard/go-difflib v1.0.0 // indirect
10+
gopkg.in/yaml.v3 v3.0.1 // indirect
11+
)

go.sum

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
2+
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
3+
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
4+
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
5+
github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
6+
github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
7+
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
8+
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
9+
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
10+
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

logging_test.go

Lines changed: 45 additions & 78 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ import (
1010
"testing"
1111

1212
"github.com/pion/logging"
13+
"github.com/stretchr/testify/assert"
1314
)
1415

1516
func 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

3129
func 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

4847
func 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

6563
func 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

8279
func 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

9995
func 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

116111
func 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

270239
func 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

Comments
 (0)