Cancel rendering helmfile parts once the missing env is detected in the previous part (#941)

* Cancel rendering helmfile parts once the missing env is detected in the previous part

Fixes #913

Signed-off-by: Yusuke Kuoka <ykuoka@gmail.com>

pkg/app/desired_state_file_loader.go

@@ -241,8 +241,16 @@ func (ld *desiredStateLoader) load(env, overrodeEnv *environment.Environment, ba
 		env = &finalState.Env
 
 		ld.logger.Debugf("merged environment: %v", env)
+
+		if len(finalState.Environments) == 0 {
+			continue
 		}
 
+		// At this point, we are sure that the env has been
+		// read from the vanilla or rendered YAML document.
+		// We can now check if the env is defined in it and fail accordingly.
+		// See https://github.com/helmfile/helmfile/issues/913
+
 		// We defer the missing env detection and failure until
 		// all the helmfile parts are loaded and merged.
 		// Otherwise, any single helmfile part missing the env would fail the whole helmfile run.
@@ -255,6 +263,16 @@ func (ld *desiredStateLoader) load(env, overrodeEnv *environment.Environment, ba
 				Cause: &state.UndefinedEnvError{Env: env.Name},
 			}
 		}
+	}
+
+	// If environments are not defined in the helmfile at all although the env is specified,
+	// it's a missing env situation. Let's fail.
+	if len(finalState.Environments) == 0 && evaluateBases && env.Name != state.DefaultEnv {
+		return nil, &state.StateLoadError{
+			Msg:   fmt.Sprintf("failed to read %s", finalState.FilePath),
+			Cause: &state.UndefinedEnvError{Env: env.Name},
+		}
+	}
 
 	return finalState, nil
 }

test/e2e/template/helmfile/testdata/snapshot/environment_missing_in_subhelmfile/config.yaml

@@ -0,0 +1,5 @@
+chartifyTempDir: environment_missing_in_subhelmfile
+helmfileArgs:
+- --environment
+- test
+- template

test/e2e/template/helmfile/testdata/snapshot/environment_missing_in_subhelmfile/helmfiles/prod.yaml

@@ -0,0 +1,15 @@
+environments:
+  prod:
+    values:
+    - foo: prod
+
+---
+
+releases:
+- name: raw
+  chart: ../../../charts/raw-0.0.1
+  values:
+  - templates:
+    - |
+      subhelmfile: {{ .Values.foo }}
+      envName: {{ .Values.envName }}

test/e2e/template/helmfile/testdata/snapshot/environment_missing_in_subhelmfile/helmfiles/test.yaml

@@ -0,0 +1,15 @@
+environments:
+  test:
+    values:
+    - foo: test
+
+---
+
+releases:
+- name: raw
+  chart: ../../../charts/raw-0.0.1
+  values:
+  - templates:
+    - |
+      subhelmfile: {{ .Values.foo }}
+      envName: {{ .Values.envName }}

test/e2e/template/helmfile/testdata/snapshot/environment_missing_in_subhelmfile/input.yaml

@@ -0,0 +1,12 @@
+environments:
+  test:
+    values:
+    - test.yaml.gotmpl
+---
+helmfiles:
+- path: helmfiles/test.yaml
+  values:
+  - envName: {{ .Values.envName }}
+- path: helmfiles/prod.yaml
+  values:
+  - envName: {{ .Values.envName }}

test/e2e/template/helmfile/testdata/snapshot/environment_missing_in_subhelmfile/output.yaml

@@ -0,0 +1,7 @@
+Building dependency release=raw, chart=../../../charts/raw-0.0.1
+Templating release=raw, chart=../../../charts/raw-0.0.1
+---
+# Source: raw/templates/resources.yaml
+subhelmfile: test
+envName: test
+

test/e2e/template/helmfile/testdata/snapshot/environment_missing_in_subhelmfile/test.yaml.gotmpl

@@ -0,0 +1 @@
+envName: {{ .Environment.Name }}