Expose FaultInjection status in TaskResponse

ecs-agent/tmds/handlers/v4/handlers_test.go

@@ -128,7 +128,7 @@ var (
 	}
 )
 
-// Returns a standard agent task response
+// taskResponse returns a standard agent task response
 func taskResponse() *state.TaskResponse {
 	return &state.TaskResponse{
 		TaskResponse: &v2.TaskResponse{
@@ -175,6 +175,13 @@ func taskResponse() *state.TaskResponse {
 	}
 }
 
+// taskResponseWithFaultInjectionEnabled returns a standard agent task response with FaultInjection enabled
+func taskResponseWithFaultInjectionEnabled() *state.TaskResponse {
+	taskResponse := taskResponse()
+	taskResponse.FaultInjectionEnabled = true
+	return taskResponse
+}
+
 func TestContainerMetadata(t *testing.T) {
 	var setup = func(t *testing.T) (*mux.Router, *gomock.Controller, *mock_state.MockAgentState,
 		*mock_metrics.MockEntryFactory,
@@ -288,6 +295,25 @@ func TestTaskMetadata(t *testing.T) {
 			expectedResponseBody: *expectedTaskResponse,
 		})
 	})
+
+	t.Run("happy case with FaultInjection enabled", func(t *testing.T) {
+		metadata := taskResponseWithFaultInjectionEnabled()
+		expectedTaskResponse := taskResponseWithFaultInjectionEnabled()
+		expectedTaskResponse.CredentialsID = ""            // credentials ID not expected
+		expectedTaskResponse.TaskNetworkConfig = nil       // TaskNetworkConfig is not expected and would be used internally
+		expectedTaskResponse.FaultInjectionEnabled = false // FaultInjectionEnabled is not expected and would be used internally
+
+		handler, _, agentState, _ := setup(t)
+		agentState.EXPECT().
+			GetTaskMetadata(endpointContainerID).
+			Return(*metadata, nil)
+		testTMDSRequest(t, handler, TMDSTestCase[state.TaskResponse]{
+			path:                 path,
+			expectedStatusCode:   http.StatusOK,
+			expectedResponseBody: *expectedTaskResponse,
+		})
+	})
+
 	t.Run("task lookup failure", func(t *testing.T) {
 		handler, _, agentState, _ := setup(t)
 		agentState.EXPECT().

ecs-agent/tmds/handlers/v4/state/response.go

@@ -38,6 +38,7 @@ type TaskResponse struct {
 	EphemeralStorageMetrics *EphemeralStorageMetrics `json:"EphemeralStorageMetrics,omitempty"`
 	CredentialsID           string                   `json:"-"`
 	TaskNetworkConfig       *TaskNetworkConfig       `json:"-"`
+	FaultInjectionEnabled   bool                     `json:"-"`
 }
 
 // TaskNetworkConfig contains required network configurations for network faults injection.