testprovider

type DataSource struct {
	ReadResponse           *datasource.ReadResponse
	SchemaResponse         *datasource.SchemaResponse
	ValidateConfigResponse *datasource.ValidateConfigResponse
}

type ListResource struct {
	IncludeResource            bool
	SchemaResponse             *list.SchemaResponse
	ListResultsStream          *list.ListResultsStream
	ValidateListConfigResponse *list.ValidateListConfigResponse
}

type Protov5Provider struct {
	ConfigureResponse      *provider.Protov5ConfigureResponse
	DataSources            map[string]DataSource
	Resources              map[string]Resource
	SchemaResponse         *provider.Protov5SchemaResponse
	StopResponse           *provider.Protov5StopResponse
	ValidateConfigResponse *provider.Protov5ValidateConfigResponse
}

type Provider struct {
	ConfigureResponse      *provider.ConfigureResponse
	DataSources            map[string]DataSource
	ListResources          map[string]ListResource
	Resources              map[string]Resource
	StateStores            map[string]*StateStore
	SchemaResponse         *provider.SchemaResponse
	StopResponse           *provider.StopResponse
	ValidateConfigResponse *provider.ValidateConfigResponse
}

type Resource struct {
	CreateResponse *resource.CreateResponse
	// Some tests need more control over logic run during Create, the struct defined CreateResponse
	// will be passed to this function if available
	CreateFunc func(context.Context, resource.CreateRequest, *resource.CreateResponse)

	DeleteResponse      *resource.DeleteResponse
	ImportStateResponse *resource.ImportStateResponse

	// Planning happens multiple ways during a single TestStep, so statically
	// defining only the response is very problematic.
	PlanChangeFunc func(context.Context, resource.PlanChangeRequest, *resource.PlanChangeResponse)

	ReadResponse           *resource.ReadResponse
	IdentitySchemaResponse *resource.IdentitySchemaResponse
	SchemaResponse         *resource.SchemaResponse
	UpdateResponse         *resource.UpdateResponse
	UpgradeStateResponse   *resource.UpgradeStateResponse
	ValidateConfigResponse *resource.ValidateConfigResponse
}

type StateStore struct {
	SchemaResponse         *statestore.SchemaResponse
	ConfigureResponse      *statestore.ConfigureResponse
	ValidateConfigResponse *statestore.ValidateConfigResponse

	// Declaring a mock state store is slightly more complicated then other types since the implementation is likely
	// needed to be stateful to work with multiple terraform commands.
	GetStatesFunc       func(context.Context, statestore.GetStatesRequest, *statestore.GetStatesResponse)
	DeleteStateFunc     func(context.Context, statestore.DeleteStateRequest, *statestore.DeleteStateResponse)
	LockStateFunc       func(context.Context, statestore.LockStateRequest, *statestore.LockStateResponse)
	UnlockStateFunc     func(context.Context, statestore.UnlockStateRequest, *statestore.UnlockStateResponse)
	ReadStateBytesFunc  func(context.Context, statestore.ReadStateBytesRequest, *statestore.ReadStateBytesResponse)
	WriteStateBytesFunc func(context.Context, statestore.WriteStateBytesRequest, *statestore.WriteStateBytesResponse)
	// contains filtered or unexported fields
}