store

type AWSError struct {
	Code      string
	Message   string
	Bucket    string
	Key       string
	VersionId string
}

type DeleteObjectsVersionResponse struct {
	Deleted   []DeletedPackage
	AWSErrors []AWSError
}

type DeletedPackage struct {
	NodeId       string
	DeleteMarker bool
}

type DynamoDBFixture struct {
	Fixture
	Client *dynamodb.Client
	// Tables is a set of table names
	Tables map[string]bool
}

type DynamoDBStore struct {
	Client *dynamodb.Client
}

type Fixture struct {
	T *testing.T
}

type GetDeleteMarkerVersionsResponse map[string]*S3ObjectInfo

type ModelScanner struct {
	ColumnNames       []string
	ColumnNamesString string
}

type NoLogger struct{}

type NoSQLStore interface {
	GetDeleteMarkerVersions(ctx context.Context, restoring ...*models.RestorePackageInfo) (GetDeleteMarkerVersionsResponse, error)
	RemoveDeleteRecords(ctx context.Context, restoring []*models.RestorePackageInfo) error
	logging.Logger
}

type ObjectStore interface {
	DeleteObjectsVersion(ctx context.Context, objInfos ...S3ObjectInfo) (DeleteObjectsVersionResponse, error)
	logging.Logger
}

type PackageScanner struct {
	*ModelScanner
	QualifiedColumnNamesString string
}

type PackageStateTransition struct {
	// NodeId: package node id
	NodeId string
	// Expected: expected current state of package
	Expected packageState.State
	// Target: desired state of package
	Target packageState.State
}

type PostgresConfig struct {
	Host     string
	Port     string
	User     string
	Password string
	DBName   string
	SSLMode  string
}

type PostgresOption struct {
	Name  string
	Value string
}

type PostgresStoreFactory struct {
	DB *sql.DB
}

type Queries struct {
	OrgId int
	logging.Logger
	// contains filtered or unexported fields
}

type QueueStore interface {
	SendRestorePackage(ctx context.Context, restoreMessage models.RestorePackageMessage) error
}

type RowScanner interface {
	Scan(dest ...any) error
}

type S3Fixture struct {
	Fixture
	Client *s3.Client
	// Buckets is a set of bucket names
	Buckets map[string]bool
}

type S3ObjectInfo struct {
	NodeId    string `dynamodbav:"NodeId"`
	Bucket    string `dynamodbav:"S3Bucket"`
	Key       string `dynamodbav:"S3Key"`
	VersionId string `dynamodbav:"S3ObjectVersion"`
	Size      string `dynamodbav:"ObjectSize"`
}

type S3Store struct {
	Client *s3.Client
}

type SQLStore interface {
	UpdatePackageName(ctx context.Context, packageId int64, newName string) error
	GetDatasetByNodeId(ctx context.Context, dsNodeId string) (*pgdb.Dataset, error)
	// TransitionPackageState updates the state of the given package from expectedState to targetState and returns the resulting package.
	// If the package is not already in expectedState, then models.PackageNotFoundError is returned.
	TransitionPackageState(ctx context.Context, datasetId int64, packageId string, expectedState, targetState packageState.State) (*pgdb.Package, error)
	// TransitionPackageStateBulk updates the state of the given packages from expectedStates to targetStates and returns the resulting packages.
	TransitionPackageStateBulk(ctx context.Context, datasetId int64, transitions []PackageStateTransition) ([]*pgdb.Package, error)
	// TransitionDescendantPackageState updates the state of any descendants of the given package which have state == expectedState to targetState and returns the updated packages.
	// It does not update the state of the package with id packageId, only its descendants if any.
	TransitionDescendantPackageState(ctx context.Context, datasetId, packageId int64, expectedState, targetState packageState.State) ([]*pgdb.Package, error)
	// TransitionAncestorPackageState updates the state of any ancestors of the package with the given parentId which have state == expectedState to targetState and returns the updated packages.
	TransitionAncestorPackageState(ctx context.Context, parentId int64, expectedState, targetState packageState.State) ([]*pgdb.Package, error)
	NewSavepoint(ctx context.Context, name string) error
	RollbackToSavepoint(ctx context.Context, name string) error
	ReleaseSavepoint(ctx context.Context, name string) error
	IncrementOrganizationStorage(ctx context.Context, organizationId int64, sizeIncrement int64) error
	IncrementDatasetStorage(ctx context.Context, datasetId int64, sizeIncrement int64) error
	IncrementPackageStorage(ctx context.Context, packageId int64, sizeIncrement int64) error
	IncrementPackageStorageAncestors(ctx context.Context, parentId int64, size int64) error
	GetPackageByNodeId(ctx context.Context, packageId string) (*pgdb.Package, error)
	logging.Logger
}

type SQLStoreFactory interface {
	NewSimpleStore(orgId int) SQLStore
	ExecStoreTx(ctx context.Context, orgId int, fn func(store SQLStore) error) error
}

type TestDB struct {
	*sql.DB
	// contains filtered or unexported fields
}

type TestPackage struct {
	pgdb.Package
}

type TestPackageAttribute struct {
	packageInfo.PackageAttribute
}

type TestPackageAttributes []TestPackageAttribute