Skip to content

Commit 474f955

Browse files
committed
Use more appropriate test function
1 parent d1573b8 commit 474f955

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

reader_test.go

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -11,12 +11,12 @@ import (
1111

1212
func TestReader(t *testing.T) {
1313
reader, err := Open("test-data/test-data/GeoIP2-City-Test.mmdb")
14-
assert.Nil(t, err)
14+
require.NoError(t, err)
1515

1616
defer reader.Close()
1717

1818
record, err := reader.City(net.ParseIP("81.2.69.160"))
19-
assert.Nil(t, err)
19+
require.NoError(t, err)
2020

2121
m := reader.Metadata()
2222
assert.Equal(t, uint(2), m.BinaryFormatMajorVersion)
@@ -120,22 +120,22 @@ func TestReader(t *testing.T) {
120120

121121
func TestMetroCode(t *testing.T) {
122122
reader, err := Open("test-data/test-data/GeoIP2-City-Test.mmdb")
123-
assert.Nil(t, err)
123+
require.NoError(t, err)
124124
defer reader.Close()
125125

126126
record, err := reader.City(net.ParseIP("216.160.83.56"))
127-
assert.Nil(t, err)
127+
require.NoError(t, err)
128128

129129
assert.Equal(t, uint(819), record.Location.MetroCode)
130130
}
131131

132132
func TestAnonymousIP(t *testing.T) {
133133
reader, err := Open("test-data/test-data/GeoIP2-Anonymous-IP-Test.mmdb")
134-
assert.Nil(t, err)
134+
require.NoError(t, err)
135135
defer reader.Close()
136136

137137
record, err := reader.AnonymousIP(net.ParseIP("1.2.0.0"))
138-
assert.Nil(t, err)
138+
require.NoError(t, err)
139139

140140
assert.True(t, record.IsAnonymous)
141141

@@ -148,11 +148,11 @@ func TestAnonymousIP(t *testing.T) {
148148

149149
func TestASN(t *testing.T) {
150150
reader, err := Open("test-data/test-data/GeoLite2-ASN-Test.mmdb")
151-
assert.Nil(t, err)
151+
require.NoError(t, err)
152152
defer reader.Close()
153153

154154
record, err := reader.ASN(net.ParseIP("1.128.0.0"))
155-
assert.Nil(t, err)
155+
require.NoError(t, err)
156156

157157
assert.Equal(t, uint(1221), record.AutonomousSystemNumber)
158158

@@ -161,24 +161,24 @@ func TestASN(t *testing.T) {
161161

162162
func TestConnectionType(t *testing.T) {
163163
reader, err := Open("test-data/test-data/GeoIP2-Connection-Type-Test.mmdb")
164-
assert.Nil(t, err)
164+
require.NoError(t, err)
165165

166166
defer reader.Close()
167167

168168
record, err := reader.ConnectionType(net.ParseIP("1.0.1.0"))
169-
assert.Nil(t, err)
169+
require.NoError(t, err)
170170

171171
assert.Equal(t, "Cable/DSL", record.ConnectionType)
172172
}
173173

174174
func TestCountry(t *testing.T) {
175175
reader, err := Open("test-data/test-data/GeoIP2-Country-Test.mmdb")
176-
assert.Nil(t, err)
176+
require.NoError(t, err)
177177

178178
defer reader.Close()
179179

180180
record, err := reader.Country(net.ParseIP("81.2.69.160"))
181-
assert.Nil(t, err)
181+
require.NoError(t, err)
182182

183183
assert.False(t, record.Country.IsInEuropeanUnion)
184184
assert.False(t, record.RegisteredCountry.IsInEuropeanUnion)
@@ -187,11 +187,11 @@ func TestCountry(t *testing.T) {
187187

188188
func TestDomain(t *testing.T) {
189189
reader, err := Open("test-data/test-data/GeoIP2-Domain-Test.mmdb")
190-
assert.Nil(t, err)
190+
require.NoError(t, err)
191191
defer reader.Close()
192192

193193
record, err := reader.Domain(net.ParseIP("1.2.0.0"))
194-
assert.Nil(t, err)
194+
require.NoError(t, err)
195195
assert.Equal(t, "maxmind.com", record.Domain)
196196
}
197197

@@ -225,11 +225,11 @@ func TestEnterprise(t *testing.T) {
225225

226226
func TestISP(t *testing.T) {
227227
reader, err := Open("test-data/test-data/GeoIP2-ISP-Test.mmdb")
228-
assert.Nil(t, err)
228+
require.NoError(t, err)
229229
defer reader.Close()
230230

231231
record, err := reader.ISP(net.ParseIP("149.101.100.0"))
232-
assert.Nil(t, err)
232+
require.NoError(t, err)
233233

234234
assert.Equal(t, uint(6167), record.AutonomousSystemNumber)
235235

0 commit comments

Comments
 (0)