milvus/tests/integration/suite_options.go

24 lines
745 B
Go

package integration
type clusterSuiteOption struct {
dropAllCollectionsWhenTestTearDown bool
notResetDeploymentWhenTestTearDown bool
}
type ClusterSuiteOption func(o *clusterSuiteOption)
// WithDropAllCollectionsWhenTestTearDown drop all collections when test tear down
func WithDropAllCollectionsWhenTestTearDown() ClusterSuiteOption {
return func(o *clusterSuiteOption) {
o.dropAllCollectionsWhenTestTearDown = true
}
}
// WithoutResetDeploymentWhenTestTearDown reset deployment when test tear down
// the reset operation will keep only 1 default component for milvus cluster.
func WithoutResetDeploymentWhenTestTearDown() ClusterSuiteOption {
return func(o *clusterSuiteOption) {
o.notResetDeploymentWhenTestTearDown = true
}
}