fix comment for test and dont wrap the heartbeat

snowflakedb · Jun 18, 2024 · 14ecdfa · 14ecdfa
1 parent aa1b205
commit 14ecdfa
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 11 deletions.
diff --git a/function_wrapper_test.go b/function_wrapper_test.go
@@ -20,11 +20,11 @@ var closeGoWrapperCalledChannel = func(ctx context.Context, f func()) {
 
 func TestGoWrapper(t *testing.T) {
 	runDBTest(t, func(dbt *DBTest) {
-		t.Cleanup(
-			func() {
-				goWrapperCalled = false
-			},
-		)
+		oldGoroutineWrapper := GoroutineWrapper
+		t.Cleanup(func() {
+			GoroutineWrapper = oldGoroutineWrapper
+		})
+
 		GoroutineWrapper = closeGoWrapperCalledChannel
 
 		numrows := 100000

diff --git a/heartbeat.go b/heartbeat.go
@@ -41,12 +41,7 @@ func (hc *heartbeat) run() {
 
 func (hc *heartbeat) start() {
 	hc.shutdownChan = make(chan bool)
-	go GoroutineWrapper(
-		context.Background(),
-		func() {
-			hc.run()
-		},
-	)
+	go hc.run()
 	logger.Info("heartbeat started")
 }