code: func (_m *API) VirtualizationWait(ctx context.Context, ID string, state string) (*types.VirtualizationWaitResult, error) {
  	ret := _m.Called(ctx, ID, state)
  
  	var r0 *types.VirtualizationWaitResult
  	if rf, ok := ret.Get(0).(func(context.Context, string, string) *types.VirtualizationWaitResult); ok {
  		r0 = rf(ctx, ID, state)
  	} else {
  		if ret.Get(0) != nil {
  			r0 = ret.Get(0).(*types.VirtualizationWaitResult)
  		}
  	}
  
  	var r1 error
  	if rf, ok := ret.Get(1).(func(context.Context, string, string) error); ok {
  		r1 = rf(ctx, ID, state)
  	} else {
  		r1 = ret.Error(1)
  	}
  
  	return r0, r1
  }
docs: // VirtualizationWait provides a mock function with given fields: ctx, ID, state