[chore][receiver/snowflake] Enable goleak checks

open-telemetry · Apr 4, 2024 · 5c09feb · 5c09feb
1 parent eb308a9
commit 5c09feb
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/receiver/snowflakereceiver/client_test.go b/receiver/snowflakereceiver/client_test.go
@@ -18,7 +18,7 @@ import (
 )
 
 func TestDefaultClientCreation(t *testing.T) {
-	_, err := newDefaultClient(componenttest.NewNopTelemetrySettings(), Config{
+	c, err := newDefaultClient(componenttest.NewNopTelemetrySettings(), Config{
 		Username:  "testuser",
 		Password:  "testPassword",
 		Account:   "testAccount",
@@ -27,7 +27,8 @@ func TestDefaultClientCreation(t *testing.T) {
 		Database:  "testDatabase",
 		Role:      "testRole",
 	})
-	assert.Equal(t, nil, err)
+	assert.NoError(t, err)
+	assert.NoError(t, c.client.Close())
 }
 
 // test query wrapper

diff --git a/receiver/snowflakereceiver/package_test.go b/receiver/snowflakereceiver/package_test.go
@@ -0,0 +1,14 @@
+// Copyright The OpenTelemetry Authors
+// SPDX-License-Identifier: Apache-2.0
+
+package snowflakereceiver
+
+import (
+	"testing"
+
+	"go.uber.org/goleak"
+)
+
+func TestMain(m *testing.M) {
+	goleak.VerifyTestMain(m)
+}
diff --git a/receiver/snowflakereceiver/scraper_test.go b/receiver/snowflakereceiver/scraper_test.go
@@ -72,6 +72,7 @@ func TestStart(t *testing.T) {
 	scraper := newSnowflakeMetricsScraper(receivertest.NewNopCreateSettings(), cfg)
 	err := scraper.start(context.Background(), componenttest.NewNopHost())
 	require.NoError(t, err, "Problem starting scraper")
+	require.NoError(t, scraper.shutdown(context.Background()))
 }
 
 // wrapper type for convenience