@@ -44,7 +44,8 @@ var _ = Describe("GenerateDocker", func() {
4444 Opts : & cmd.GenerateDockerOpts {
4545 GenerateOpts : & cmd.GenerateOpts {
4646 GlobalOptions : cmd.GlobalOptions {
47- Env : mockEnv ,
47+ WorkspaceId : - 1 ,
48+ Env : mockEnv ,
4849 },
4950 Input : defaultInput ,
5051 Output : defaultOutput ,
@@ -109,16 +110,17 @@ var _ = Describe("GenerateDocker", func() {
109110 })
110111
111112 Context ("when workspace ID is set in the env var" , func () {
113+ BeforeEach (func () {
114+ wsId = 99
115+ })
112116 It ("should use the workspace ID from environment" , func () {
113- envWsId := 99
114- mockEnv .EXPECT ().GetWorkspaceId ().Return (envWsId , nil )
115-
116117 cmd := & cmd.GenerateDockerCmd {
117118 Opts : & cmd.GenerateDockerOpts {
118119 GenerateOpts : & cmd.GenerateOpts {
119120 GlobalOptions : cmd.GlobalOptions {
120- Env : mockEnv ,
121- // WorkspaceId is nil, so it will use env var
121+ Env : mockEnv ,
122+ WorkspaceId : - 1 ,
123+ // WorkspaceId is -1 (default), so it will use env var
122124 },
123125 },
124126 BaseImage : "alpine:latest" ,
@@ -128,7 +130,8 @@ var _ = Describe("GenerateDocker", func() {
128130 ws := api.Workspace {
129131 GitUrl : * openapi_client .NewNullableString (& repoUrl ),
130132 }
131- mockClient .EXPECT ().GetWorkspace (envWsId ).Return (ws , nil )
133+ mockEnv .EXPECT ().GetWorkspaceId ().Return (wsId , nil )
134+ mockClient .EXPECT ().GetWorkspace (99 ).Return (ws , nil )
132135 mockGit .EXPECT ().CloneRepository (memoryFs , repoUrl , branch , clonedir ).Return (nil , nil )
133136
134137 err := cmd .CloneRepository (mockClient , memoryFs , mockGit , clonedir )
@@ -144,7 +147,7 @@ var _ = Describe("GenerateDocker", func() {
144147 GenerateOpts : & cmd.GenerateOpts {
145148 GlobalOptions : cmd.GlobalOptions {
146149 Env : mockEnv ,
147- WorkspaceId : & flagWsId ,
150+ WorkspaceId : flagWsId ,
148151 },
149152 },
150153 BaseImage : "alpine:latest" ,
0 commit comments