@@ -28,6 +28,7 @@ var _ = Describe("Continue deployment command", func() {
28
28
fakeActor * v7fakes.FakeActor
29
29
binaryName string
30
30
appName string
31
+ noWait bool
31
32
spaceGUID string
32
33
executeErr error
33
34
)
@@ -44,6 +45,7 @@ var _ = Describe("Continue deployment command", func() {
44
45
45
46
cmd = ContinueDeploymentCommand {
46
47
RequiredArgs : flag.AppName {AppName : appName },
48
+ NoWait : noWait ,
47
49
BaseCommand : BaseCommand {
48
50
UI : testUI ,
49
51
Config : fakeConfig ,
@@ -126,10 +128,13 @@ var _ = Describe("Continue deployment command", func() {
126
128
127
129
When ("getting the app succeeds" , func () {
128
130
var appGUID string
131
+ var returnedApplication resources.Application
132
+
129
133
BeforeEach (func () {
130
134
appGUID = "some-app-guid"
135
+ returnedApplication = resources.Application {Name : appName , GUID : appGUID }
131
136
fakeActor .GetApplicationByNameAndSpaceReturns (
132
- resources. Application { Name : appName , GUID : appGUID } ,
137
+ returnedApplication ,
133
138
v7action.Warnings {"get-app-warning" },
134
139
nil ,
135
140
)
@@ -202,6 +207,32 @@ var _ = Describe("Continue deployment command", func() {
202
207
Expect (executeErr ).ToNot (HaveOccurred ())
203
208
})
204
209
210
+ When ("the --no-wait flag is not provided" , func () {
211
+ It ("polls and waits" , func () {
212
+ Expect (fakeActor .PollStartForDeploymentCallCount ()).To (Equal (1 ))
213
+
214
+ invokedApplication , invokedGuid , invokedNoWait , _ := fakeActor .PollStartForDeploymentArgsForCall (0 )
215
+ Expect (invokedApplication ).To (Equal (returnedApplication ))
216
+ Expect (invokedGuid ).To (Equal (deploymentGUID ))
217
+ Expect (invokedNoWait ).To (Equal (false ))
218
+ })
219
+ })
220
+
221
+ When ("the --no-wait flag is provided" , func () {
222
+ BeforeEach (func () {
223
+ cmd .NoWait = true
224
+ })
225
+
226
+ It ("polls without waiting" , func () {
227
+ Expect (fakeActor .PollStartForDeploymentCallCount ()).To (Equal (1 ))
228
+
229
+ invokedApplication , invokedGuid , invokedNoWait , _ := fakeActor .PollStartForDeploymentArgsForCall (0 )
230
+ Expect (invokedApplication ).To (Equal (returnedApplication ))
231
+ Expect (invokedGuid ).To (Equal (deploymentGUID ))
232
+ Expect (invokedNoWait ).To (Equal (true ))
233
+ })
234
+ })
235
+
205
236
When ("polling the application fails" , func () {
206
237
BeforeEach (func () {
207
238
fakeActor .PollStartForDeploymentReturns (
0 commit comments