@@ -41,6 +41,7 @@ import (
4141 "math/rand"
4242 "net"
4343 "os"
44+ "strconv"
4445 "strings"
4546 "sync"
4647 "sync/atomic"
@@ -184,7 +185,7 @@ func newTestSession(proto protoVersion, addresses ...string) (*Session, error) {
184185}
185186
186187func TestDNSLookupConnected (t * testing.T ) {
187- log := & testLogger {}
188+ log := newTestLogger ( LogLevelDebug )
188189
189190 // Override the defaul DNS resolver and restore at the end
190191 failDNS = true
@@ -195,7 +196,6 @@ func TestDNSLookupConnected(t *testing.T) {
195196
196197 cluster := NewCluster ("cassandra1.invalid" , srv .Address , "cassandra2.invalid" )
197198 cluster .Logger = log
198- cluster .LegacyLogLevel = LogLevelDebug
199199 cluster .ProtoVersion = int (defaultProto )
200200 cluster .disableControlConn = true
201201
@@ -212,15 +212,14 @@ func TestDNSLookupConnected(t *testing.T) {
212212}
213213
214214func TestDNSLookupError (t * testing.T ) {
215- log := & testLogger {}
215+ log := newTestLogger ( LogLevelDebug )
216216
217217 // Override the defaul DNS resolver and restore at the end
218218 failDNS = true
219219 defer func () { failDNS = false }()
220220
221221 cluster := NewCluster ("cassandra1.invalid" , "cassandra2.invalid" )
222222 cluster .Logger = log
223- cluster .LegacyLogLevel = LogLevelDebug
224223 cluster .ProtoVersion = int (defaultProto )
225224 cluster .disableControlConn = true
226225
@@ -242,7 +241,7 @@ func TestDNSLookupError(t *testing.T) {
242241
243242func TestStartupTimeout (t * testing.T ) {
244243 ctx , cancel := context .WithCancel (context .Background ())
245- log := & testLogger {}
244+ log := newTestLogger ( LogLevelDebug )
246245
247246 srv := NewTestServer (t , defaultProto , ctx )
248247 defer srv .Stop ()
@@ -253,7 +252,6 @@ func TestStartupTimeout(t *testing.T) {
253252 startTime := time .Now ()
254253 cluster := NewCluster (srv .Address )
255254 cluster .Logger = log
256- cluster .LegacyLogLevel = LogLevelDebug
257255 cluster .ProtoVersion = int (defaultProto )
258256 cluster .disableControlConn = true
259257 // Set very long query connection timeout
@@ -352,15 +350,21 @@ func TestCancel(t *testing.T) {
352350type testQueryObserver struct {
353351 metrics map [string ]* hostMetrics
354352 verbose bool
355- logger StdLogger
353+ logger StructuredLogger
356354}
357355
358356func (o * testQueryObserver ) ObserveQuery (ctx context.Context , q ObservedQuery ) {
359357 host := q .Host .ConnectAddress ().String ()
360358 o .metrics [host ] = q .Metrics
361359 if o .verbose {
362- o .logger .Printf ("Observed query %q. Returned %v rows, took %v on host %q with %v attempts and total latency %v. Error: %q\n " ,
363- q .Statement , q .Rows , q .End .Sub (q .Start ), host , q .Metrics .Attempts , q .Metrics .TotalLatency , q .Err )
360+ o .logger .Debug ("Observed query %q. Returned %v rows, took %v on host %q with %v attempts and total latency %v. Error: %q\n " ,
361+ newLogFieldString ("stmt" , q .Statement ),
362+ newLogFieldInt ("rows" , q .Rows ),
363+ newLogFieldString ("duration" , q .End .Sub (q .Start ).String ()),
364+ newLogFieldString ("host" , host ),
365+ newLogFieldInt ("attempts" , q .Metrics .Attempts ),
366+ newLogFieldString ("latency" , strconv .FormatInt (q .Metrics .TotalLatency , 10 )),
367+ newLogFieldError ("err" , q .Err ))
364368 }
365369}
366370
@@ -412,7 +416,7 @@ func TestQueryRetry(t *testing.T) {
412416}
413417
414418func TestQueryMultinodeWithMetrics (t * testing.T ) {
415- log := & testLogger {}
419+ log := newTestLogger ( LogLevelDebug )
416420 defer func () {
417421 os .Stdout .WriteString (log .String ())
418422 }()
@@ -487,7 +491,7 @@ func (t *testRetryPolicy) GetRetryType(err error) RetryType {
487491}
488492
489493func TestSpeculativeExecution (t * testing.T ) {
490- log := & testLogger {}
494+ log := newTestLogger ( LogLevelDebug )
491495 defer func () {
492496 os .Stdout .WriteString (log .String ())
493497 }()
@@ -720,7 +724,7 @@ func TestStream0(t *testing.T) {
720724 r : bufio .NewReader (& buf ),
721725 },
722726 streams : streams .New (protoVersion4 ),
723- logger : newInternalLoggerFromStdLogger ( & defaultLogger {}, LogLevelNone ) ,
727+ logger : & defaultLogger {},
724728 }
725729
726730 err := conn .recv (context .Background (), false )
0 commit comments