@@ -24,6 +24,7 @@ import (
24
24
"os"
25
25
"runtime"
26
26
"strconv"
27
+ "strings"
27
28
"sync"
28
29
"testing"
29
30
"time"
@@ -320,13 +321,13 @@ func TestCustomSettings(t *testing.T) {
320
321
require .NoError (t , row .Err ())
321
322
322
323
var setting string
323
- require .NoError (t , row .Scan (& setting ))
324
- require .Equal (t , "custom_value" , setting )
324
+ assert .NoError (t , row .Scan (& setting ))
325
+ assert .Equal (t , "custom_value" , setting )
325
326
})
326
327
327
328
t .Run ("get non-existing custom setting value" , func (t * testing.T ) {
328
329
row := conn .QueryRow (context .Background (), "SELECT getSetting('custom_non_existing_setting')" )
329
- require . ErrorContains (t , row .Err (), "Unknown setting custom_non_existing_setting" )
330
+ assert . Contains (t , strings . ReplaceAll ( row .Err (). Error (), "'" , "" ), "Unknown setting custom_non_existing_setting" )
330
331
})
331
332
332
333
t .Run ("get custom setting value from query context" , func (t * testing.T ) {
@@ -335,11 +336,11 @@ func TestCustomSettings(t *testing.T) {
335
336
}))
336
337
337
338
row := conn .QueryRow (ctx , "SELECT getSetting('custom_query_setting')" )
338
- require .NoError (t , row .Err ())
339
+ assert .NoError (t , row .Err ())
339
340
340
341
var setting string
341
- require .NoError (t , row .Scan (& setting ))
342
- require .Equal (t , "custom_query_value" , setting )
342
+ assert .NoError (t , row .Scan (& setting ))
343
+ assert .Equal (t , "custom_query_value" , setting )
343
344
})
344
345
}
345
346
0 commit comments