PMM-13057 Fix test.

percona · Oct 10, 2024 · 4c996e5 · 4c996e5
1 parent a23b179
commit 4c996e5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/managed/services/telemetry/datasource_envvars_test.go b/managed/services/telemetry/datasource_envvars_test.go
@@ -21,7 +21,7 @@ import (
 	"os"
 	"testing"
 
-	pmmv1 "github.com/percona/saas/gen/telemetry/events/pmm"
+	genericv1 "github.com/percona/saas/gen/telemetry/generic"
 	"github.com/sirupsen/logrus"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
@@ -94,7 +94,7 @@ func TestEnvVarsDatasource(t *testing.T) {
 		metrics, err := dsEnvVars.FetchMetrics(ctx, *config)
 		require.NoError(t, err)
 
-		expected := []*pmmv1.ServerMetric_Metric{
+		expected := []*genericv1.GenericReport_Metric{
 			{Key: "test_env_var1", Value: "1"},
 			{Key: "test_env_var2", Value: "test"},
 			{Key: "test_env_var3", Value: "true"},
@@ -138,7 +138,7 @@ func TestEnvVarsDatasource(t *testing.T) {
 		metrics, err := dsEnvVars.FetchMetrics(ctx, *config)
 		require.NoError(t, err)
 
-		expected := []*pmmv1.ServerMetric_Metric{
+		expected := []*genericv1.GenericReport_Metric{
 			{Key: "test_env_var6", Value: "1"},
 			{Key: "test_env_var7", Value: "1"},
 			{Key: "test_env_var8", Value: "1"},