testutil

type CommitCopyInput struct {
	StorageCopyID int64
	ContentID     int64
	CopyIndex     int
	PieceCID      string
	PieceID       *idtypes.OnChainID
	RetrievalURL  string
	TransactionID string
}

type MockCache struct {
	ListUploadIDsFunc            func(context.Context) ([]string, error)
	WalkContentFilesFunc         func(context.Context, bool, func(string, int64) error) error
	QuarantineFunc               func(context.Context, string, int64) error
	DeleteQuarantinedFunc        func(context.Context, string, int64) error
	PutFunc                      func(ctx context.Context, bucket, key string, r io.Reader, size int64) (*cache.ObjectInfo, error)
	PutStagedFunc                func(ctx context.Context, bucket, key string, r io.Reader, size int64) (*cache.StagedObject, error)
	GetFunc                      func(ctx context.Context, bucket, key string) (io.ReadCloser, *cache.ObjectInfo, error)
	DeleteFunc                   func(ctx context.Context, bucket, key string) error
	ExistsFunc                   func(ctx context.Context, bucket, key string) bool
	UsedBytesFunc                func() int64
	CapacitySnapshotFunc         func() cache.CapacitySnapshot
	ConsumeRefusedWriteBytesFunc func() int64
	CreateBucketDirFunc          func(ctx context.Context, bucket string) error
	DeleteBucketDirFunc          func(ctx context.Context, bucket string) error
	PutPartStagedFunc            func(ctx context.Context, uploadID string, partNumber int, r io.Reader, size int64) (*cache.StagedObject, error)
	AssemblePartsFunc            func(ctx context.Context, bucket, key, uploadID string, partNumbers []int) (*cache.StagedObject, []string, error)
	DeleteUploadFunc             func(ctx context.Context, uploadID string) error
}

type MockChainEpochReader struct {
	CurrentEpochFunc func(ctx context.Context) (int64, error)
}

type MockCommitNonces struct {

	// Err fails every read while set.
	Err error
	// contains filtered or unexported fields
}

type MockServiceTerminator struct {
	TerminateServiceFunc func(ctx context.Context, dataSetID sdktypes.BigInt) (*synapse.TerminationResult, error)
	// VerifyServicePayerFunc defaults to accepting every data set.
	VerifyServicePayerFunc func(ctx context.Context, dataSetID sdktypes.BigInt) error
	// ContextIdentityValue overrides DefaultContextIdentity.
	ContextIdentityValue storage.ContextIdentity
}

type MockStorageClient struct {
	ContextIdentityValue    storage.ContextIdentity
	PDPVerifierAddressValue common.Address
	UploadFunc              func(ctx context.Context, r io.Reader, opts *storage.UploadOptions) (*storage.UploadResult, error)
	DownloadFunc            func(ctx context.Context, pieceCID cid.Cid, opts *storage.DownloadOptions) (io.ReadCloser, error)
	PrepareUploadFunc       func(ctx context.Context, dataSize uint64, targets []synapse.StorageTarget) (*sdkcosts.MultiContextCosts, error)
	SelectUploadTargetsFunc func(ctx context.Context, opts storage.SelectUploadContextsOptions) ([]synapse.StorageTarget, error)
	OpenProviderTargetFunc  func(ctx context.Context, providerID sdktypes.BigInt, opts storage.NewProviderContextOptions) (synapse.ProviderTarget, error)
	OpenDataSetTargetFunc   func(ctx context.Context, dataSetID sdktypes.BigInt, opts storage.NewDataSetContextOptions) (synapse.DataSetTarget, error)
	OpenTargetFunc          func(ctx context.Context, opts *OpenTargetOptions) (synapse.StorageTarget, error)
	FindMatchingDataSetFunc func(ctx context.Context, providerID sdktypes.BigInt, metadata map[string]string, withCDN bool) (*storage.DataSetRef, error)
	OpenCleanupContextFunc  func(ctx context.Context, dataSetID sdktypes.BigInt, opts storage.NewDataSetContextOptions) (synapse.CleanupContext, error)
	DeletionStateFunc       func(ctx context.Context, dataSetID, pieceID sdktypes.BigInt) (synapse.CleanupPieceState, error)
	DeletionStatesFunc      func(ctx context.Context, dataSetID sdktypes.BigInt, pieceIDs []sdktypes.BigInt) (synapse.CleanupPiecesState, error)
	ProviderPiecesKnownFunc func(ctx context.Context, serviceURL string, dataSetID sdktypes.BigInt, pieceIDs []sdktypes.BigInt) (synapse.ProviderPiecesState, error)
}

type MockStorageTarget struct {
	ProviderIDValue      sdktypes.BigInt
	DataSetIDValue       *sdktypes.BigInt
	ClientDataSetIDValue sdktypes.BigInt
	ServiceURLValue      string
	WithCDNValue         bool
	CreateDataSetFunc    func(context.Context, *storage.CreateDataSetOptions) (*storage.CreateDataSetResult, error)
	WaitDataSetFunc      func(context.Context, string, sdktypes.BigInt) (*storage.CreateDataSetResult, error)
	// ContextIdentityValue overrides DefaultContextIdentity.
	ContextIdentityValue      storage.ContextIdentity
	FindDataSetByClientIDFunc func(context.Context, sdktypes.BigInt) (storage.DataSetRef, bool, error)
	CreateStoreSessionFunc    func(context.Context) (string, error)
	UploadStoreSessionFunc    func(context.Context, string, string, io.Reader, *storage.StoreOptions) (*storage.StoreResult, error)
	FinalizeStoreSessionFunc  func(context.Context, string, string, cid.Cid) error
	PresignForCommitFunc      func(context.Context, []storage.PieceInput) ([]byte, error)
	SubmitPullFunc            func(context.Context, storage.PullRequest) (*storage.PullResult, error)
	SubmitCommitFunc          func(context.Context, storage.CommitRequest) (*storage.CommitSubmission, error)
	GetCommitStatusFunc       func(context.Context, string) (*storage.CommitStatus, error)
	PieceStatusFunc           func(context.Context, cid.Cid) (*storage.PieceStatus, error)
	// CheckWritableFunc defaults to a writable data set.
	CheckWritableFunc func(context.Context) error
}

type MockWalletQuerier struct {
	GetWalletInfoFunc func(ctx context.Context) (*synapse.WalletInfo, error)
}

type OpenTargetOptions struct {
	ProviderID      *sdktypes.BigInt
	DataSetID       *sdktypes.BigInt
	DataSetMetadata map[string]string
	WithCDN         *bool
}