Test boilerplate

Author: Julien-Ben

api/v1/mdb/mongodb_types.go

@@ -443,6 +443,8 @@ type MongoDbSpec struct {
 	// +kubebuilder:pruning:PreserveUnknownFields
 	// +optional
 	MemberConfig []automationconfig.MemberOptions `json:"memberConfig,omitempty"`
+
+	ClusterSpecList ClusterSpecList `json:"clusterSpecList,omitempty"`
 }
 
 func (m *MongoDbSpec) GetExternalDomain() *string {

api/v1/mdb/mongodbbuilder.go

@@ -23,6 +23,12 @@ func NewDefaultReplicaSetBuilder() *MongoDBBuilder {
 	return defaultMongoDB(ReplicaSet)
 }
 
+func NewDefaultMultiReplicaSetBuilder() *MongoDBBuilder {
+	return defaultMongoDB(ReplicaSet).
+		SetMultiClusterTopology().
+		SetDefaultClusterSpecList()
+}
+
 func NewDefaultShardedClusterBuilder() *MongoDBBuilder {
 	return defaultMongoDB(ShardedCluster).
 		SetShardCountSpec(3).
@@ -264,6 +270,24 @@ func (b *MongoDBBuilder) AddDummyOpsManagerConfig() *MongoDBBuilder {
 	return b
 }
 
+func (b *MongoDBBuilder) SetDefaultClusterSpecList() *MongoDBBuilder {
+	b.mdb.Spec.ClusterSpecList = ClusterSpecList{
+		{
+			ClusterName: "test-cluster-0",
+			Members:     1,
+		},
+		{
+			ClusterName: "test-cluster-1",
+			Members:     1,
+		},
+		{
+			ClusterName: "test-cluster-2",
+			Members:     1,
+		},
+	}
+	return b
+}
+
 func (b *MongoDBBuilder) SetAllClusterSpecLists(clusterSpecList ClusterSpecList) *MongoDBBuilder {
 	b.mdb.Spec.ShardSpec.ClusterSpecList = clusterSpecList
 	b.mdb.Spec.ConfigSrvSpec.ClusterSpecList = clusterSpecList

controllers/operator/mongodbreplicaset_controller_multi_test.go

@@ -0,0 +1,58 @@
+package operator
+
+import (
+	"context"
+	mdbv1 "github.com/mongodb/mongodb-kubernetes/api/v1/mdb"
+	"github.com/mongodb/mongodb-kubernetes/controllers/om"
+	"github.com/mongodb/mongodb-kubernetes/controllers/operator/mock"
+	kubernetesClient "github.com/mongodb/mongodb-kubernetes/mongodb-community-operator/pkg/kube/client"
+	"github.com/mongodb/mongodb-kubernetes/pkg/images"
+	"github.com/mongodb/mongodb-kubernetes/pkg/kube"
+	"github.com/mongodb/mongodb-kubernetes/pkg/util"
+	"github.com/stretchr/testify/assert"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	"sigs.k8s.io/controller-runtime/pkg/reconcile"
+	"testing"
+)
+
+// NedDefaultMultiReplicaSetBuilder
+
+func TestCreateMultiClusterReplicaSet(t *testing.T) {
+	ctx := context.Background()
+	rs := mdbv1.NewDefaultMultiReplicaSetBuilder().Build()
+
+	reconciler, client, _, _ := defaultMultiClusterReplicaSetReconciler(ctx, nil, "", "", mrs)
+	checkMultiReplicaSetReconcileSuccessful(ctx, t, reconciler, rs, client, false)
+}
+
+func checkMultiReplicaSetReconcileSuccessful(ctx context.Context, t *testing.T, reconciler reconcile.Reconciler, m *mdbv1.MongoDB, client client.Client, shouldRequeue bool) {
+	err := client.Update(ctx, m)
+	assert.NoError(t, err)
+
+	result, e := reconciler.Reconcile(ctx, requestFromObject(m))
+	assert.NoError(t, e)
+	if shouldRequeue {
+		assert.True(t, result.Requeue || result.RequeueAfter > 0)
+	} else {
+		assert.Equal(t, reconcile.Result{RequeueAfter: util.TWENTY_FOUR_HOURS}, result)
+	}
+
+	// fetch the last updates as the reconciliation loop can update the mdb resource.
+	err = client.Get(ctx, kube.ObjectKey(m.Namespace, m.Name), m)
+	assert.NoError(t, err)
+}
+
+func multiClusterReplicaSetReconciler(ctx context.Context, imageUrls images.ImageUrls, initDatabaseNonStaticImageVersion, databaseNonStaticImageVersion string, m *mdbv1.MongoDB) (*ReconcileMongoDbReplicaSet, kubernetesClient.Client, map[string]client.Client, *om.CachedOMConnectionFactory) {
+	kubeClient, omConnectionFactory := mock.NewDefaultFakeClient(m)
+	memberClusterMap := getFakeMultiClusterMap(omConnectionFactory)
+	return newReplicaSetReconciler(ctx, kubeClient, imageUrls, initDatabaseNonStaticImageVersion, databaseNonStaticImageVersion, false, false, memberClusterMap, omConnectionFactory.GetConnectionFunc), kubeClient, memberClusterMap, omConnectionFactory
+}
+
+func defaultMultiClusterReplicaSetReconciler(ctx context.Context, imageUrls images.ImageUrls, initDatabaseNonStaticImageVersion, databaseNonStaticImageVersion string, rs *mdbv1.MongoDB) (*ReconcileMongoDbReplicaSet, kubernetesClient.Client, map[string]client.Client, *om.CachedOMConnectionFactory) {
+	multiReplicaSetController, client, clusterMap, omConnectionFactory := multiClusterReplicaSetReconciler(ctx, imageUrls, initDatabaseNonStaticImageVersion, databaseNonStaticImageVersion, rs)
+	omConnectionFactory.SetPostCreateHook(func(connection om.Connection) {
+		connection.(*om.MockedOmConnection).Hostnames = calculateHostNamesForExternalDomains(rs)
+	})
+
+	return multiReplicaSetController, client, clusterMap, omConnectionFactory
+}