Skip to content

Commit c47d6be

Browse files
author
Thean Lim
committed
Update PerformUpdateMessage to PerformUpdateInput
- Same Struct Types - TODO: handle AckRequest (not a type in V2) so either create a AckRequest type or use PerformUpdateOutput
1 parent 2f38c52 commit c47d6be

File tree

3 files changed

+9
-9
lines changed

3 files changed

+9
-9
lines changed

agent/acs/updater/updater.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -228,8 +228,8 @@ func (u *updater) download(info *ecsacs.UpdateInfo) (err error) {
228228

229229
var exit = os.Exit
230230

231-
func (u *updater) performUpdateHandler() func(req *ecsacs.PerformUpdateMessage) {
232-
return func(req *ecsacs.PerformUpdateMessage) {
231+
func (u *updater) performUpdateHandler() func(req *ecsacs.PerformUpdateInput) {
232+
return func(req *ecsacs.PerformUpdateInput) {
233233
u.Lock()
234234
defer u.Unlock()
235235

agent/acs/updater/updater_test.go

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ func TestPerformUpdateWithUpdatesDisabled(t *testing.T) {
128128
Reason: ptr("Updates are disabled").(*string),
129129
}})
130130

131-
msg := &ecsacs.PerformUpdateMessage{
131+
msg := &ecsacs.PerformUpdateInput{
132132
ClusterArn: ptr("cluster").(*string),
133133
ContainerInstanceArn: ptr("containerInstance").(*string),
134134
MessageId: ptr("mid").(*string),
@@ -181,7 +181,7 @@ func TestFullUpdateFlow(t *testing.T) {
181181

182182
require.Equal(t, "update-tar-data", writtenFile.String(), "incorrect data written")
183183

184-
msg := &ecsacs.PerformUpdateMessage{
184+
msg := &ecsacs.PerformUpdateInput{
185185
ClusterArn: ptr("cluster").(*string),
186186
ContainerInstanceArn: ptr("containerInstance").(*string),
187187
MessageId: ptr("mid2").(*string),
@@ -248,7 +248,7 @@ func TestUndownloadedUpdate(t *testing.T) {
248248
MessageId: ptr("mid").(*string),
249249
}})
250250

251-
msg := &ecsacs.PerformUpdateMessage{
251+
msg := &ecsacs.PerformUpdateInput{
252252
ClusterArn: ptr("cluster").(*string),
253253
ContainerInstanceArn: ptr("containerInstance").(*string),
254254
MessageId: ptr("mid").(*string),
@@ -305,7 +305,7 @@ func TestDuplicateUpdateMessagesWithSuccess(t *testing.T) {
305305

306306
require.Equal(t, "update-tar-data", writtenFile.String(), "incorrect data written")
307307

308-
msg := &ecsacs.PerformUpdateMessage{
308+
msg := &ecsacs.PerformUpdateInput{
309309
ClusterArn: ptr("cluster").(*string),
310310
ContainerInstanceArn: ptr("containerInstance").(*string),
311311
MessageId: ptr("mid3").(*string),
@@ -373,7 +373,7 @@ func TestDuplicateUpdateMessagesWithFailure(t *testing.T) {
373373

374374
require.Equal(t, "update-tar-data", writtenFile.String(), "incorrect data written")
375375

376-
msg := &ecsacs.PerformUpdateMessage{
376+
msg := &ecsacs.PerformUpdateInput{
377377
ClusterArn: ptr("cluster").(*string),
378378
ContainerInstanceArn: ptr("containerInstance").(*string),
379379
MessageId: ptr("mid3").(*string),
@@ -443,7 +443,7 @@ func TestNewerUpdateMessages(t *testing.T) {
443443

444444
require.Equal(t, "newer-update-tar-data", writtenFile.String(), "incorrect data written")
445445

446-
msg := &ecsacs.PerformUpdateMessage{
446+
msg := &ecsacs.PerformUpdateInput{
447447
ClusterArn: ptr("cluster").(*string),
448448
ContainerInstanceArn: ptr("containerInstance").(*string),
449449
MessageId: ptr("mid2").(*string),

ecs-agent/acs/client/acs_client_types.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ func init() {
3737
ecsacs.PollOutput{},
3838
ecsacs.AckRequest{},
3939
ecsacs.UpdateFailureInput{},
40-
ecsacs.PerformUpdateMessage{},
40+
ecsacs.PerformUpdateInput{},
4141
ecsacs.StageUpdateMessage{},
4242
ecsacs.RefreshTaskIAMRoleCredentialsInput{},
4343
ecsacs.RefreshTaskIAMRoleCredentialsOutput{},

0 commit comments

Comments
 (0)