code: func (m *MockTaskEngineState) AllImageStates() []*image.ImageState {
  	ret := m.ctrl.Call(m, "AllImageStates")
  	ret0, _ := ret[0].([]*image.ImageState)
  	return ret0
  }
docs: // AllImageStates mocks base method