node

type ConnectionOpts struct {
	Timeout         time.Duration
	TimeBeforeRetry time.Duration
	IgnoreError     bool
	Sudo            bool
}

type Driver interface {
	// Init initializes the node driver under the given scheduler
	Init() error

	// DeleteNode deletes the given node
	DeleteNode(node Node, timeout time.Duration) error

	// String returns the string name of this driver.
	String() string

	// RebootNode reboots the given node
	RebootNode(node Node, options RebootNodeOpts) error

	// RunCommand runs the given command on the node and returns the output
	RunCommand(node Node, command string, options ConnectionOpts) (string, error)

	// ShutdownNode shuts down the given node
	ShutdownNode(node Node, options ShutdownNodeOpts) error

	// FindFiles finds and returns the files for the given path regex and the node
	FindFiles(path string, node Node, options FindOpts) (string, error)

	// Systemctl runs a systemctl command for the given service on the node
	Systemctl(node Node, service string, options SystemctlOpts) error

	// TestConnection tests connection to given node. returns nil if driver can connect to given node
	TestConnection(node Node, options ConnectionOpts) error

	// YankDrive simulates a failure on the provided drive on the given node.
	// It returns the bus ID of the drive which can be used to recover it back
	YankDrive(node Node, driveNameToFail string, options ConnectionOpts) (string, error)

	// RecoverDrive recovers the given drive from failure on the given node.
	RecoverDrive(node Node, driveNameToRecover string, driveUUID string, options ConnectionOpts) error

	// SystemCheck checks whether core files are present on the given node.
	SystemCheck(node Node, options ConnectionOpts) (string, error)

	// SetASGClusterSize sets node count per zone for an asg cluster
	SetASGClusterSize(perZoneCount int64, timeout time.Duration) error

	// GetASGClusterSize gets node count for an asg cluster
	GetASGClusterSize() (int64, error)

	// SetClusterVersion sets desired version for cluster and its node pools
	SetClusterVersion(version string, timeout time.Duration) error

	// GetClusterVersion returns version of cluster and its node pools
	GetClusterVersion() (clusterVersion string, nodePoolsVersion []string, err error)

	// GetZones returns list of zones in which ASG cluster is running
	GetZones() ([]string, error)
}

type ErrFailedToFindFileOnNode struct {
	Node  Node
	Cause string
}

type ErrFailedToRebootNode struct {
	Node  Node
	Cause string
}

type ErrFailedToRecoverDrive struct {
	Node  Node
	Cause string
}

type ErrFailedToRunCommand struct {
	Addr  string
	Node  Node
	Cause string
}

type ErrFailedToRunSystemctlOnNode struct {
	Node  Node
	Cause string
}

type ErrFailedToShutdownNode struct {
	Node  Node
	Cause string
}

type ErrFailedToSystemCheck struct {
	Node  Node
	Cause string
}

type ErrFailedToTestConnection struct {
	Node  Node
	Cause string
}

type ErrFailedToYankDrive struct {
	Node  Node
	Cause string
}

type FindOpts struct {
	Name     string
	MinDepth int
	MaxDepth int
	Type     FindType
	Empty    bool
	ConnectionOpts
}

type FindType string

type Node struct {
	api.StorageNode

	VolDriverNodeID          string
	Name                     string
	Addresses                []string
	UsableAddr               string
	Type                     Type
	Zone                     string
	Region                   string
	IsStorageDriverInstalled bool
	IsMetadataNode           bool
	StoragePools             []StoragePool
	// contains filtered or unexported fields
}

type RebootNodeOpts struct {
	Force bool
	ConnectionOpts
}

type ShutdownNodeOpts struct {
	Force bool
	ConnectionOpts
}

type StoragePool struct {
	*api.StoragePool
	// StoragePoolAtInit in the storage pool that's captured when the test initializes. This is useful for tests that
	// want to track changes in a pool since the test was started. For e.g tracking pool expansion changes
	StoragePoolAtInit *api.StoragePool
	// WorkloadSize is the size in bytes of the workload that will be launched by test on this storage pool
	WorkloadSize uint64
}

type SystemctlOpts struct {
	Action string
	ConnectionOpts
}

type TestConnectionOpts struct {
	ConnectionOpts
}

type Type string