Update vendor for ecs agent with TMDs changes.

Author: avirak

agent/handlers/v4/tmdsstate.go

@@ -249,6 +249,18 @@ func (s *TMDSAgentState) GetTaskStats(v3EndpointID string) (map[string]*tmdsv4.S
 	return taskStatsResponse, nil
 }
 
+func (s *TMDSAgentState) GetTasksMetadata() ([]tmdsv4.TaskResponse, error) {
+	return nil, tmdsv4.NewErrorMetadataFetchFailure("tasks metadata endpoint not supported")
+}
+
+func (s *TMDSAgentState) GetTasksMetadataWithTags() ([]tmdsv4.TaskResponse, error) {
+	return nil, tmdsv4.NewErrorMetadataFetchFailure("tasks metadata with tags endpoint not supported")
+}
+
+func (s *TMDSAgentState) GetTasksStats(v3EndpointID string) ([]map[string]*tmdsv4.StatsResponse, error) {
+	return nil, tmdsv4.NewErrorStatsFetchFailure("tasks stats endpoint not supported", nil)
+}
+
 // sortContainersCNIPauseFirst sorts containers so that CNI_PAUSE containers appear first.
 // Other containers maintain their relative order.
 func sortContainersCNIPauseFirst(containers []tmdsv4.ContainerResponse) {

agent/handlers/v4/tmdsstate_test.go

@@ -17,6 +17,7 @@
 package v4
 
 import (
+	"errors"
 	"testing"
 
 	apicontainer "github.com/aws/amazon-ecs-agent/agent/api/container"
@@ -179,3 +180,30 @@ func TestGetTaskMetadataWithTags(t *testing.T) {
 		})
 	}
 }
+
+func TestGetTasksMetadata_NotSupported(t *testing.T) {
+	state := &TMDSAgentState{}
+	_, err := state.GetTasksMetadata()
+
+	var metadataErr *tmdsv4.ErrorMetadataFetchFailure
+	assert.True(t, errors.As(err, &metadataErr))
+	assert.Contains(t, err.Error(), "not supported")
+}
+
+func TestGetTasksMetadataWithTags_NotSupported(t *testing.T) {
+	state := &TMDSAgentState{}
+	_, err := state.GetTasksMetadataWithTags()
+
+	var metadataErr *tmdsv4.ErrorMetadataFetchFailure
+	assert.True(t, errors.As(err, &metadataErr))
+	assert.Contains(t, err.Error(), "not supported")
+}
+
+func TestGetTasksStats_NotSupported(t *testing.T) {
+	state := &TMDSAgentState{}
+	_, err := state.GetTasksStats("test")
+
+	var statsErr *tmdsv4.ErrorStatsFetchFailure
+	assert.True(t, errors.As(err, &statsErr))
+	assert.Contains(t, err.Error(), "not supported")
+}

agent/vendor/github.com/aws/amazon-ecs-agent/ecs-agent/tmds/handlers/utils/helpers.go

@@ -212,7 +212,7 @@ func tasksMetadataHandler(
 		} else {
 			tasksMetadata, err = agentState.GetTasksMetadata()
 		}
-		
+
 		if err != nil {
 			logger.Error("Failed to get v4 tasks metadata", logger.Fields{
 				field.Error: err,

agent/vendor/github.com/aws/amazon-ecs-agent/ecs-agent/tmds/handlers/v4/handlers.go

@@ -504,9 +504,9 @@ func TestTasksMetadata(t *testing.T) {
 		expectedTaskResponse := taskResponse()
 		expectedTaskResponse.CredentialsID = ""      // credentials ID not expected
 		expectedTaskResponse.TaskNetworkConfig = nil // TaskNetworkConfig is not expected and would be used internally.
-		
+
 		tasksMetadata := []state.TaskResponse{*expectedTaskResponse}
-		
+
 		handler, _, agentState, _ := setup(t)
 		agentState.EXPECT().
 			GetTasksMetadata().
@@ -583,9 +583,9 @@ func TestTasksMetadataWithTags(t *testing.T) {
 		expectedTaskResponse := taskResponse()
 		expectedTaskResponse.CredentialsID = ""      // credentials ID not expected
 		expectedTaskResponse.TaskNetworkConfig = nil // TaskNetworkConfig is not expected and would be used internally.
-		
+
 		tasksMetadata := []state.TaskResponse{*expectedTaskResponse}
-		
+
 		handler, _, agentState, _ := setup(t)
 		agentState.EXPECT().
 			GetTasksMetadataWithTags().
@@ -953,25 +953,25 @@ func TestTasksMetadataHandler_IncludeTags(t *testing.T) {
 
 	t.Run("calls GetTasksMetadata when includeTags is false", func(t *testing.T) {
 		agentState.EXPECT().GetTasksMetadata().Return([]state.TaskResponse{}, nil)
-		
+
 		handler := tasksMetadataHandler(agentState, metricsFactory, false)
 		req := httptest.NewRequest("GET", "/v4/test/tasks", nil)
 		w := httptest.NewRecorder()
-		
+
 		handler(w, req)
-		
+
 		assert.Equal(t, http.StatusOK, w.Code)
 	})
 
 	t.Run("calls GetTasksMetadataWithTags when includeTags is true", func(t *testing.T) {
 		agentState.EXPECT().GetTasksMetadataWithTags().Return([]state.TaskResponse{}, nil)
-		
+
 		handler := tasksMetadataHandler(agentState, metricsFactory, true)
 		req := httptest.NewRequest("GET", "/v4/test/tasks", nil)
 		w := httptest.NewRecorder()
-		
+
 		handler(w, req)
-		
+
 		assert.Equal(t, http.StatusOK, w.Code)
 	})
 }