chore(dbrp mapper): Move DBRP mapper to platform
Nathaniel Cook
parent
ca66f0df74
commit
23ab3f1981
|
|
@ -0,0 +1,74 @@
|
||||||
|
package platform
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"regexp"
|
||||||
|
)
|
||||||
|
|
||||||
|
// DBRPMapperService provides a mapping of cluster, database and retention policy to an organization ID and bucket ID.
|
||||||
|
type DBRPMapperService interface {
|
||||||
|
// FindBy returns the dbrp mapping the for cluster, db and rp.
|
||||||
|
FindBy(ctx context.Context, cluster, db, rp string) (*DBRPMapping, error)
|
||||||
|
// Find returns the first dbrp mapping the matches the filter.
|
||||||
|
Find(ctx context.Context, filter DBRPMappingFilter) (*DBRPMapping, error)
|
||||||
|
// FindMany returns a list of dbrp mappings that match filter and the total count of matching dbrp mappings.
|
||||||
|
FindMany(ctx context.Context, filter DBRPMappingFilter, opt ...FindOptions) ([]*DBRPMapping, int, error)
|
||||||
|
// Create creates a new dbrp mapping, if a different mapping exists an error is returned.
|
||||||
|
Create(ctx context.Context, dbrpMap *DBRPMapping) error
|
||||||
|
// Delete removes a dbrp mapping.
|
||||||
|
// Deleting a mapping that does not exists is not an error.
|
||||||
|
Delete(ctx context.Context, cluster, db, rp string) error
|
||||||
|
}
|
||||||
|
|
||||||
|
// DBRPMapping represents a mapping of a cluster, database and retention policy to an organization ID and bucket ID.
|
||||||
|
type DBRPMapping struct {
|
||||||
|
Cluster string `json:"cluster"`
|
||||||
|
Database string `json:"database"`
|
||||||
|
RetentionPolicy string `json:"retention_policy"`
|
||||||
|
|
||||||
|
// Default indicates if this mapping is the default for the cluster and database.
|
||||||
|
Default bool `json:"default"`
|
||||||
|
|
||||||
|
OrganizationID ID `json:"organization_id"`
|
||||||
|
BucketID ID `json:"bucket_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var validPattern = regexp.MustCompile(`^[\.\w]+$`)
|
||||||
|
|
||||||
|
// Validate reports any validation errors for the mapping.
|
||||||
|
func (m DBRPMapping) Validate() error {
|
||||||
|
if m.Cluster == "" {
|
||||||
|
return errors.New("Cluster is required")
|
||||||
|
}
|
||||||
|
if !validPattern.MatchString(m.Cluster) {
|
||||||
|
return errors.New("Cluster must contain only letters, numbers, '_' and '.'")
|
||||||
|
}
|
||||||
|
if m.Database == "" {
|
||||||
|
return errors.New("Database is required")
|
||||||
|
}
|
||||||
|
if !validPattern.MatchString(m.Database) {
|
||||||
|
return errors.New("Database must contain only letters, numbers, '_' and '.'")
|
||||||
|
}
|
||||||
|
if m.RetentionPolicy == "" {
|
||||||
|
return errors.New("RetentionPolicy is required")
|
||||||
|
}
|
||||||
|
if !validPattern.MatchString(m.RetentionPolicy) {
|
||||||
|
return errors.New("RetentionPolicy must contain only letters, numbers, '_' and '.'")
|
||||||
|
}
|
||||||
|
if len(m.OrganizationID) == 0 {
|
||||||
|
return errors.New("OrganizationID is required")
|
||||||
|
}
|
||||||
|
if len(m.BucketID) == 0 {
|
||||||
|
return errors.New("BucketID is required")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// DBRPMappingFilter represents a set of filters that restrict the returned results by cluster, database and retention policy.
|
||||||
|
type DBRPMappingFilter struct {
|
||||||
|
Cluster *string
|
||||||
|
Database *string
|
||||||
|
RetentionPolicy *string
|
||||||
|
Default *bool
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,760 @@
|
||||||
|
package testing
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/google/go-cmp/cmp"
|
||||||
|
"github.com/influxdata/platform"
|
||||||
|
)
|
||||||
|
|
||||||
|
var dbrpMappingCmpOptions = cmp.Options{
|
||||||
|
cmp.Comparer(func(x, y []byte) bool {
|
||||||
|
return bytes.Equal(x, y)
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
|
||||||
|
// DBRPMappingFields will include the dbrpMappings
|
||||||
|
type DBRPMappingFields struct {
|
||||||
|
DBRPMappings []*platform.DBRPMapping
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateDBRPMapping testing
|
||||||
|
func CreateDBRPMapping(
|
||||||
|
init func(DBRPMappingFields, *testing.T) (platform.DBRPMapperService, func()),
|
||||||
|
t *testing.T,
|
||||||
|
) {
|
||||||
|
type args struct {
|
||||||
|
dbrpMapping *platform.DBRPMapping
|
||||||
|
}
|
||||||
|
type wants struct {
|
||||||
|
err error
|
||||||
|
dbrpMappings []*platform.DBRPMapping
|
||||||
|
}
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
fields DBRPMappingFields
|
||||||
|
args args
|
||||||
|
wants wants
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "create dbrpMappings with empty set",
|
||||||
|
fields: DBRPMappingFields{
|
||||||
|
DBRPMappings: []*platform.DBRPMapping{},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
dbrpMapping: &platform.DBRPMapping{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
dbrpMappings: []*platform.DBRPMapping{{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
}},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "basic create dbrpMapping",
|
||||||
|
fields: DBRPMappingFields{
|
||||||
|
DBRPMappings: []*platform.DBRPMapping{{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
}},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
dbrpMapping: &platform.DBRPMapping{
|
||||||
|
Cluster: "cluster2",
|
||||||
|
Database: "database2",
|
||||||
|
RetentionPolicy: "retention_policy2",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org2"),
|
||||||
|
BucketID: platform.ID("bucket2"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
dbrpMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Cluster: "cluster2",
|
||||||
|
Database: "database2",
|
||||||
|
RetentionPolicy: "retention_policy2",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org2"),
|
||||||
|
BucketID: platform.ID("bucket2"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "idempotent create dbrpMapping",
|
||||||
|
fields: DBRPMappingFields{
|
||||||
|
DBRPMappings: []*platform.DBRPMapping{{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
}},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
dbrpMapping: &platform.DBRPMapping{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
dbrpMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "error on create existing dbrpMapping",
|
||||||
|
fields: DBRPMappingFields{
|
||||||
|
DBRPMappings: []*platform.DBRPMapping{{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
}},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
dbrpMapping: &platform.DBRPMapping{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
err: errors.New("dbrp mapping already exists"),
|
||||||
|
dbrpMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
s, done := init(tt.fields, t)
|
||||||
|
defer done()
|
||||||
|
ctx := context.TODO()
|
||||||
|
err := s.Create(ctx, tt.args.dbrpMapping)
|
||||||
|
if (err != nil) != (tt.wants.err != nil) {
|
||||||
|
t.Fatalf("expected error '%v' got '%v'", tt.wants.err, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil && tt.wants.err != nil {
|
||||||
|
if err.Error() != tt.wants.err.Error() {
|
||||||
|
t.Fatalf("expected error messages to match '%v' got '%v'", tt.wants.err, err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
dbrpMappings, _, err := s.FindMany(ctx, platform.DBRPMappingFilter{})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("failed to retrieve dbrpMappings: %v", err)
|
||||||
|
}
|
||||||
|
if diff := cmp.Diff(dbrpMappings, tt.wants.dbrpMappings, dbrpMappingCmpOptions...); diff != "" {
|
||||||
|
t.Errorf("dbrpMappings are different -got/+want\ndiff %s", diff)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// FindDBRPMappings testing
|
||||||
|
func FindDBRPMappings(
|
||||||
|
init func(DBRPMappingFields, *testing.T) (platform.DBRPMapperService, func()),
|
||||||
|
t *testing.T,
|
||||||
|
) {
|
||||||
|
type args struct {
|
||||||
|
filter platform.DBRPMappingFilter
|
||||||
|
}
|
||||||
|
|
||||||
|
type wants struct {
|
||||||
|
dbrpMappings []*platform.DBRPMapping
|
||||||
|
err error
|
||||||
|
}
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
fields DBRPMappingFields
|
||||||
|
args args
|
||||||
|
wants wants
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "find all dbrpMappings",
|
||||||
|
fields: DBRPMappingFields{
|
||||||
|
DBRPMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Cluster: "cluster2",
|
||||||
|
Database: "database2",
|
||||||
|
RetentionPolicy: "retention_policy2",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org2"),
|
||||||
|
BucketID: platform.ID("bucket2"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
filter: platform.DBRPMappingFilter{},
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
dbrpMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Cluster: "cluster2",
|
||||||
|
Database: "database2",
|
||||||
|
RetentionPolicy: "retention_policy2",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org2"),
|
||||||
|
BucketID: platform.ID("bucket2"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "find dbrpMappings by cluster",
|
||||||
|
fields: DBRPMappingFields{
|
||||||
|
DBRPMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Cluster: "cluster2",
|
||||||
|
Database: "database2",
|
||||||
|
RetentionPolicy: "retention_policy2",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org2"),
|
||||||
|
BucketID: platform.ID("bucket2"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
filter: platform.DBRPMappingFilter{
|
||||||
|
Cluster: strPtr("cluster2"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
dbrpMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster2",
|
||||||
|
Database: "database2",
|
||||||
|
RetentionPolicy: "retention_policy2",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org2"),
|
||||||
|
BucketID: platform.ID("bucket2"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "find default rp from dbrpMappings",
|
||||||
|
fields: DBRPMappingFields{
|
||||||
|
DBRPMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyA",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org"),
|
||||||
|
BucketID: platform.ID("bucketA"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Cluster: "cluster",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyB",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org"),
|
||||||
|
BucketID: platform.ID("bucketB"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
filter: platform.DBRPMappingFilter{
|
||||||
|
Cluster: strPtr("cluster"),
|
||||||
|
Database: strPtr("database"),
|
||||||
|
Default: boolPtr(true),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
dbrpMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyB",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org"),
|
||||||
|
BucketID: platform.ID("bucketB"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
s, done := init(tt.fields, t)
|
||||||
|
defer done()
|
||||||
|
ctx := context.TODO()
|
||||||
|
|
||||||
|
dbrpMappings, _, err := s.FindMany(ctx, tt.args.filter)
|
||||||
|
if (err != nil) != (tt.wants.err != nil) {
|
||||||
|
t.Fatalf("expected errors to be equal '%v' got '%v'", tt.wants.err, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil && tt.wants.err != nil {
|
||||||
|
if err.Error() != tt.wants.err.Error() {
|
||||||
|
t.Fatalf("expected error '%v' got '%v'", tt.wants.err, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if diff := cmp.Diff(dbrpMappings, tt.wants.dbrpMappings, dbrpMappingCmpOptions...); diff != "" {
|
||||||
|
t.Errorf("dbrpMappings are different -got/+want\ndiff %s", diff)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// FindDBRPMappingByKey testing
|
||||||
|
func FindDBRPMappingByKey(
|
||||||
|
init func(DBRPMappingFields, *testing.T) (platform.DBRPMapperService, func()),
|
||||||
|
t *testing.T,
|
||||||
|
) {
|
||||||
|
type args struct {
|
||||||
|
Cluster,
|
||||||
|
Database,
|
||||||
|
RetentionPolicy string
|
||||||
|
}
|
||||||
|
|
||||||
|
type wants struct {
|
||||||
|
dbrpMapping *platform.DBRPMapping
|
||||||
|
err error
|
||||||
|
}
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
fields DBRPMappingFields
|
||||||
|
args args
|
||||||
|
wants wants
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "find dbrpMappings by cluster db and rp",
|
||||||
|
fields: DBRPMappingFields{
|
||||||
|
DBRPMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyA",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org"),
|
||||||
|
BucketID: platform.ID("bucketA"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Cluster: "cluster",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyB",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org"),
|
||||||
|
BucketID: platform.ID("bucketB"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
Cluster: "cluster",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyB",
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
dbrpMapping: &platform.DBRPMapping{
|
||||||
|
Cluster: "cluster",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyB",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org"),
|
||||||
|
BucketID: platform.ID("bucketB"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "find non existing dbrpMapping",
|
||||||
|
fields: DBRPMappingFields{
|
||||||
|
DBRPMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyA",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org"),
|
||||||
|
BucketID: platform.ID("bucketA"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
Cluster: "clusterX",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyA",
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
err: errors.New("dbrp mapping not found"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
s, done := init(tt.fields, t)
|
||||||
|
defer done()
|
||||||
|
ctx := context.TODO()
|
||||||
|
|
||||||
|
dbrpMapping, err := s.FindBy(ctx, tt.args.Cluster, tt.args.Database, tt.args.RetentionPolicy)
|
||||||
|
if (err != nil) != (tt.wants.err != nil) {
|
||||||
|
t.Fatalf("expected error '%v' got '%v'", tt.wants.err, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil && tt.wants.err != nil {
|
||||||
|
if err.Error() != tt.wants.err.Error() {
|
||||||
|
t.Fatalf("expected error messages to match '%v' got '%v'", tt.wants.err, err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if diff := cmp.Diff(dbrpMapping, tt.wants.dbrpMapping, dbrpMappingCmpOptions...); diff != "" {
|
||||||
|
t.Errorf("dbrpMappings are different -got/+want\ndiff %s", diff)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// FindDBRPMapping testing
|
||||||
|
func FindDBRPMapping(
|
||||||
|
init func(DBRPMappingFields, *testing.T) (platform.DBRPMapperService, func()),
|
||||||
|
t *testing.T,
|
||||||
|
) {
|
||||||
|
type args struct {
|
||||||
|
filter platform.DBRPMappingFilter
|
||||||
|
}
|
||||||
|
|
||||||
|
type wants struct {
|
||||||
|
dbrpMapping *platform.DBRPMapping
|
||||||
|
err error
|
||||||
|
}
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
fields DBRPMappingFields
|
||||||
|
args args
|
||||||
|
wants wants
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "find dbrpMappings by cluster db and rp",
|
||||||
|
fields: DBRPMappingFields{
|
||||||
|
DBRPMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyA",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org"),
|
||||||
|
BucketID: platform.ID("bucketA"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Cluster: "cluster",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyB",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org"),
|
||||||
|
BucketID: platform.ID("bucketB"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
filter: platform.DBRPMappingFilter{
|
||||||
|
Cluster: strPtr("cluster"),
|
||||||
|
Database: strPtr("database"),
|
||||||
|
RetentionPolicy: strPtr("retention_policyB"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
dbrpMapping: &platform.DBRPMapping{
|
||||||
|
Cluster: "cluster",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyB",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org"),
|
||||||
|
BucketID: platform.ID("bucketB"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "find default rp from dbrpMappings",
|
||||||
|
fields: DBRPMappingFields{
|
||||||
|
DBRPMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyA",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org"),
|
||||||
|
BucketID: platform.ID("bucketA"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Cluster: "cluster",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyB",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org"),
|
||||||
|
BucketID: platform.ID("bucketB"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
filter: platform.DBRPMappingFilter{
|
||||||
|
Cluster: strPtr("cluster"),
|
||||||
|
Database: strPtr("database"),
|
||||||
|
Default: boolPtr(true),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
dbrpMapping: &platform.DBRPMapping{
|
||||||
|
Cluster: "cluster",
|
||||||
|
Database: "database",
|
||||||
|
RetentionPolicy: "retention_policyB",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org"),
|
||||||
|
BucketID: platform.ID("bucketB"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
s, done := init(tt.fields, t)
|
||||||
|
defer done()
|
||||||
|
ctx := context.TODO()
|
||||||
|
|
||||||
|
dbrpMapping, err := s.Find(ctx, tt.args.filter)
|
||||||
|
if (err != nil) != (tt.wants.err != nil) {
|
||||||
|
t.Fatalf("expected error '%v' got '%v'", tt.wants.err, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil && tt.wants.err != nil {
|
||||||
|
if err.Error() != tt.wants.err.Error() {
|
||||||
|
t.Fatalf("expected error messages to match '%v' got '%v'", tt.wants.err, err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if diff := cmp.Diff(dbrpMapping, tt.wants.dbrpMapping, dbrpMappingCmpOptions...); diff != "" {
|
||||||
|
t.Errorf("dbrpMappings are different -got/+want\ndiff %s", diff)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteDBRPMapping testing
|
||||||
|
func DeleteDBRPMapping(
|
||||||
|
init func(DBRPMappingFields, *testing.T) (platform.DBRPMapperService, func()),
|
||||||
|
t *testing.T,
|
||||||
|
) {
|
||||||
|
type args struct {
|
||||||
|
Cluster, Database, RetentionPolicy string
|
||||||
|
}
|
||||||
|
type wants struct {
|
||||||
|
err error
|
||||||
|
dbrpMappings []*platform.DBRPMapping
|
||||||
|
}
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
fields DBRPMappingFields
|
||||||
|
args args
|
||||||
|
wants wants
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "delete existing dbrpMapping",
|
||||||
|
fields: DBRPMappingFields{
|
||||||
|
DBRPMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Cluster: "cluster2",
|
||||||
|
Database: "database2",
|
||||||
|
RetentionPolicy: "retention_policy2",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org2"),
|
||||||
|
BucketID: platform.ID("bucket2"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
dbrpMappings: []*platform.DBRPMapping{{
|
||||||
|
Cluster: "cluster2",
|
||||||
|
Database: "database2",
|
||||||
|
RetentionPolicy: "retention_policy2",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org2"),
|
||||||
|
BucketID: platform.ID("bucket2"),
|
||||||
|
}},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "delete dbrpMappings using key that does not exist",
|
||||||
|
fields: DBRPMappingFields{
|
||||||
|
DBRPMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Cluster: "cluster2",
|
||||||
|
Database: "database2",
|
||||||
|
RetentionPolicy: "retention_policy2",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org2"),
|
||||||
|
BucketID: platform.ID("bucket2"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
Cluster: "cluster3",
|
||||||
|
Database: "db",
|
||||||
|
RetentionPolicy: "rp",
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
dbrpMappings: []*platform.DBRPMapping{
|
||||||
|
{
|
||||||
|
Cluster: "cluster1",
|
||||||
|
Database: "database1",
|
||||||
|
RetentionPolicy: "retention_policy1",
|
||||||
|
Default: false,
|
||||||
|
OrganizationID: platform.ID("org1"),
|
||||||
|
BucketID: platform.ID("bucket1"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Cluster: "cluster2",
|
||||||
|
Database: "database2",
|
||||||
|
RetentionPolicy: "retention_policy2",
|
||||||
|
Default: true,
|
||||||
|
OrganizationID: platform.ID("org2"),
|
||||||
|
BucketID: platform.ID("bucket2"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
s, done := init(tt.fields, t)
|
||||||
|
defer done()
|
||||||
|
ctx := context.TODO()
|
||||||
|
err := s.Delete(ctx, tt.args.Cluster, tt.args.Database, tt.args.RetentionPolicy)
|
||||||
|
if (err != nil) != (tt.wants.err != nil) {
|
||||||
|
t.Fatalf("expected error '%v' got '%v'", tt.wants.err, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil && tt.wants.err != nil {
|
||||||
|
if err.Error() != tt.wants.err.Error() {
|
||||||
|
t.Fatalf("expected error messages to match '%v' got '%v'", tt.wants.err, err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
filter := platform.DBRPMappingFilter{}
|
||||||
|
dbrpMappings, _, err := s.FindMany(ctx, filter)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("failed to retrieve dbrpMappings: %v", err)
|
||||||
|
}
|
||||||
|
if diff := cmp.Diff(dbrpMappings, tt.wants.dbrpMappings, dbrpMappingCmpOptions...); diff != "" {
|
||||||
|
t.Errorf("dbrpMappings are different -got/+want\ndiff %s", diff)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func strPtr(s string) *string {
|
||||||
|
return &s
|
||||||
|
}
|
||||||
|
func boolPtr(b bool) *bool {
|
||||||
|
return &b
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue