From 1912ebd8f6029e11a1e6f9996789abf6bfbde681 Mon Sep 17 00:00:00 2001
From: Shreyas Goenka <shreyas.goenka@databricks.com>
Date: Fri, 20 Dec 2024 12:12:22 +0530
Subject: [PATCH] fix main_test.go

---
 integration/libs/telemetry/main_test.go      | 13 +++++++++++++
 integration/libs/telemetry/telemetry_test.go |  2 +-
 2 files changed, 14 insertions(+), 1 deletion(-)
 create mode 100644 integration/libs/telemetry/main_test.go

diff --git a/integration/libs/telemetry/main_test.go b/integration/libs/telemetry/main_test.go
new file mode 100644
index 000000000..640bef8f8
--- /dev/null
+++ b/integration/libs/telemetry/main_test.go
@@ -0,0 +1,13 @@
+package telemetry_test
+
+import (
+	"testing"
+
+	"github.com/databricks/cli/integration/internal"
+)
+
+// TestMain is the entrypoint executed by the test runner.
+// See [internal.Main] for prerequisites for running integration tests.
+func TestMain(m *testing.M) {
+	internal.Main(m)
+}
diff --git a/integration/libs/telemetry/telemetry_test.go b/integration/libs/telemetry/telemetry_test.go
index f6566c117..d65a7df5c 100644
--- a/integration/libs/telemetry/telemetry_test.go
+++ b/integration/libs/telemetry/telemetry_test.go
@@ -1,4 +1,4 @@
-package internal
+package telemetry_test
 
 import (
 	"context"