Add tests

elastic · Aug 27, 2024 · 2de477a · 2de477a
1 parent 5cf76f3
commit 2de477a
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/internal/install/application_configuration_test.go b/internal/install/application_configuration_test.go
@@ -51,3 +51,22 @@ func TestSelectElasticAgentImageName_NextStackInOwnNamespace(t *testing.T) {
 	selected := selectElasticAgentImageName(version)
 	assert.Equal(t, selected, elasticAgentCompleteImageName)
 }
+
+func TestSelectElasticAgentImageName_WolfiImage(t *testing.T) {
+	version := "8.16.0-SNAPSHOT"
+	selected := selectElasticAgentImageName(version)
+	assert.Equal(t, selected, elasticAgentWolfiImageName)
+}
+
+func TestSelectElasticAgentImageName_DisableWolfiImageEnvVar(t *testing.T) {
+	version := "8.16.0-SNAPSHOT"
+	t.Setenv(disableElasticAgentWolfiEnvVar, "true")
+	selected := selectElasticAgentImageName(version)
+	assert.Equal(t, selected, elasticAgentCompleteImageName)
+}
+func TestSelectElasticAgentImageName_EnableWolfiImageEnvVar(t *testing.T) {
+	version := "8.16.0-SNAPSHOT"
+	t.Setenv(disableElasticAgentWolfiEnvVar, "false")
+	selected := selectElasticAgentImageName(version)
+	assert.Equal(t, selected, elasticAgentWolfiImageName)
+}