diff --git a/internal/pkg/cli/env_init.go b/internal/pkg/cli/env_init.go index 1e8e6316680..1729b12be13 100644 --- a/internal/pkg/cli/env_init.go +++ b/internal/pkg/cli/env_init.go @@ -156,6 +156,7 @@ func (o *initEnvOpts) Execute() error { if err != nil { return fmt.Errorf("get environment struct for %s: %w", o.EnvName, err) } + env.Prod = o.IsProduction // 3. Add the stack set instance to the project stackset. if err := o.addToStackset(project, env); err != nil { diff --git a/internal/pkg/cli/env_init_test.go b/internal/pkg/cli/env_init_test.go index ca9ff62ca96..1b478a502f8 100644 --- a/internal/pkg/cli/env_init_test.go +++ b/internal/pkg/cli/env_init_test.go @@ -153,6 +153,7 @@ func TestInitEnvOpts_Execute(t *testing.T) { testCases := map[string]struct { inProjectName string inEnvName string + inProd bool expectProjectGetter func(m *mocks.MockProjectGetter) expectEnvCreator func(m *mocks.MockEnvironmentCreator) @@ -391,6 +392,7 @@ func TestInitEnvOpts_Execute(t *testing.T) { "success": { inProjectName: "phonetool", inEnvName: "test", + inProd: true, expectProjectGetter: func(m *mocks.MockProjectGetter) { m.EXPECT().GetProject("phonetool").Return(&archer.Project{Name: "phonetool"}, nil) @@ -415,6 +417,7 @@ func TestInitEnvOpts_Execute(t *testing.T) { Project: "phonetool", Name: "test", AccountID: "1234", + Prod: true, Region: "mars-1", }, Err: nil, @@ -425,6 +428,7 @@ func TestInitEnvOpts_Execute(t *testing.T) { AccountID: "1234", Region: "mars-1", Name: "test", + Prod: false, Project: "phonetool", }, nil) m.EXPECT().AddEnvToProject(gomock.Any(), gomock.Any()).Return(nil) @@ -434,6 +438,7 @@ func TestInitEnvOpts_Execute(t *testing.T) { Project: "phonetool", Name: "test", AccountID: "1234", + Prod: true, Region: "mars-1", }).Return(nil) }, @@ -582,8 +587,9 @@ func TestInitEnvOpts_Execute(t *testing.T) { opts := &initEnvOpts{ initEnvVars: initEnvVars{ - EnvName: tc.inEnvName, - GlobalOpts: &GlobalOpts{projectName: tc.inProjectName}, + EnvName: tc.inEnvName, + GlobalOpts: &GlobalOpts{projectName: tc.inProjectName}, + IsProduction: tc.inProd, }, projectGetter: mockProjectGetter, envCreator: mockEnvCreator,