55 lines
1.6 KiB
Go
55 lines
1.6 KiB
Go
package organization
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"git.sendico.io/tech/sxchnge/bff/pkg/db/repository"
|
|
"git.sendico.io/tech/sxchnge/bff/pkg/merrors"
|
|
"go.mongodb.org/mongo-driver/v2/mongo"
|
|
)
|
|
|
|
func TestDropSupersededProvisioningIndexesIgnoresMissingIndexes(t *testing.T) {
|
|
repo := &dropIndexRepository{dropErrors: []error{
|
|
mongo.CommandError{Code: mongoIndexNotFoundCode},
|
|
mongo.CommandError{Code: mongoNamespaceNotFoundCode},
|
|
}}
|
|
|
|
if err := dropSupersededProvisioningIndexes(context.Background(), repo); err != nil {
|
|
t.Fatalf("missing superseded indexes must be ignored: %v", err)
|
|
}
|
|
if len(repo.dropped) != len(supersededProvisioningIndexes) {
|
|
t.Fatalf("drop count mismatch: got=%d want=%d", len(repo.dropped), len(supersededProvisioningIndexes))
|
|
}
|
|
}
|
|
|
|
func TestDropSupersededProvisioningIndexesPreservesUnexpectedFailure(t *testing.T) {
|
|
wantErr := merrors.ServiceUnavailable("MongoDB is unavailable")
|
|
repo := &dropIndexRepository{dropErrors: []error{wantErr}}
|
|
|
|
err := dropSupersededProvisioningIndexes(context.Background(), repo)
|
|
if !errors.Is(err, wantErr) {
|
|
t.Fatalf("unexpected drop failure must be preserved: got=%v want=%v", err, wantErr)
|
|
}
|
|
if len(repo.dropped) != 1 {
|
|
t.Fatalf("cleanup must stop after an unexpected failure: dropped=%v", repo.dropped)
|
|
}
|
|
}
|
|
|
|
type dropIndexRepository struct {
|
|
repository.Repository
|
|
dropErrors []error
|
|
dropped []string
|
|
}
|
|
|
|
func (r *dropIndexRepository) DropIndex(_ context.Context, name string) error {
|
|
r.dropped = append(r.dropped, name)
|
|
if len(r.dropErrors) == 0 {
|
|
return nil
|
|
}
|
|
err := r.dropErrors[0]
|
|
r.dropErrors = r.dropErrors[1:]
|
|
return err
|
|
}
|