fixed cleanup for database/sql driver

asmyasnikov · asmyasnikov · commit 9db993e639e5 · 2025-03-04T18:54:01.000+03:00
diff --git a/tests/integration/helpers_test.go b/tests/integration/helpers_test.go
@@ -153,13 +153,11 @@ func (scope *scopeT) SQLDriver(opts ...ydb.ConnectorOption) *sql.DB {
 
 		db := sql.OpenDB(connector)
 
-		scope.Logf("Ping db")
-		err = db.PingContext(scope.Ctx)
-		if err != nil {
-			return nil, err
+		clean := func() {
+			scope.Require.NoError(db.Close())
 		}
 
-		return fixenv.NewGenericResult(db), nil
+		return fixenv.NewGenericResultWithCleanup(db, clean), nil
 	}
 	return fixenv.CacheResult(scope.Env, f)
 }

Original file line number	Diff line number	Diff line change
`@@ -153,13 +153,11 @@ func (scope scopeT) SQLDriver(opts ...ydb.ConnectorOption) sql.DB {`
`153`	`153`
`154`	`154`	`db := sql.OpenDB(connector)`
`155`	`155`
`156`		`- scope.Logf("Ping db")`
`157`		`- err = db.PingContext(scope.Ctx)`
`158`		`- if err != nil {`
`159`		`- return nil, err`
	`156`	`+ clean := func() {`
	`157`	`+ scope.Require.NoError(db.Close())`
`160`	`158`	`}`
`161`	`159`
`162`		`- return fixenv.NewGenericResult(db), nil`
	`160`	`+ return fixenv.NewGenericResultWithCleanup(db, clean), nil`
`163`	`161`	`}`
`164`	`162`	`return fixenv.CacheResult(scope.Env, f)`
`165`	`163`	`}`