| 284 | } |
| 285 | |
| 286 | func TestLogConnect(t *testing.T) { |
| 287 | t.Parallel() |
| 288 | |
| 289 | ctx, cancel := context.WithTimeout(context.Background(), 120*time.Second) |
| 290 | defer cancel() |
| 291 | |
| 292 | logger := &testLogger{} |
| 293 | tracer := &tracelog.TraceLog{ |
| 294 | Logger: logger, |
| 295 | LogLevel: tracelog.LogLevelTrace, |
| 296 | } |
| 297 | |
| 298 | config := defaultConnTestRunner.CreateConfig(ctx, t) |
| 299 | config.Tracer = tracer |
| 300 | |
| 301 | conn1, err := pgx.ConnectConfig(ctx, config) |
| 302 | require.NoError(t, err) |
| 303 | defer conn1.Close(ctx) |
| 304 | require.Len(t, logger.logs, 1) |
| 305 | require.Equal(t, "Connect", logger.logs[0].msg) |
| 306 | require.Equal(t, tracelog.LogLevelInfo, logger.logs[0].lvl) |
| 307 | |
| 308 | logger.Clear() |
| 309 | |
| 310 | config, err = pgx.ParseConfig("host=/invalid") |
| 311 | require.NoError(t, err) |
| 312 | config.Tracer = tracer |
| 313 | |
| 314 | conn2, err := pgx.ConnectConfig(ctx, config) |
| 315 | require.Nil(t, conn2) |
| 316 | require.Error(t, err) |
| 317 | require.Len(t, logger.logs, 1) |
| 318 | require.Equal(t, "Connect", logger.logs[0].msg) |
| 319 | require.Equal(t, tracelog.LogLevelError, logger.logs[0].lvl) |
| 320 | } |
| 321 | |
| 322 | func TestLogBatchStatementsOnExec(t *testing.T) { |
| 323 | t.Parallel() |