126 lines
3.0 KiB
Go
126 lines
3.0 KiB
Go
package cluster
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
acidv1 "github.com/zalando/postgres-operator/pkg/apis/acid.zalan.do/v1"
|
|
"github.com/zalando/postgres-operator/pkg/util/config"
|
|
"github.com/zalando/postgres-operator/pkg/util/k8sutil"
|
|
|
|
appsv1 "k8s.io/api/apps/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
)
|
|
|
|
func int32ToPointer(value int32) *int32 {
|
|
return &value
|
|
}
|
|
|
|
func deploymentUpdated(cluster *Cluster, err error) error {
|
|
if cluster.ConnectionPool.Deployment.Spec.Replicas == nil ||
|
|
*cluster.ConnectionPool.Deployment.Spec.Replicas != 2 {
|
|
return fmt.Errorf("Wrong nubmer of instances")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func objectsAreSaved(cluster *Cluster, err error) error {
|
|
if cluster.ConnectionPool == nil {
|
|
return fmt.Errorf("Connection pool resources are empty")
|
|
}
|
|
|
|
if cluster.ConnectionPool.Deployment == nil {
|
|
return fmt.Errorf("Deployment was not saved")
|
|
}
|
|
|
|
if cluster.ConnectionPool.Service == nil {
|
|
return fmt.Errorf("Service was not saved")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func TestConnPoolSynchronization(t *testing.T) {
|
|
testName := "Test connection pool synchronization"
|
|
var cluster = New(
|
|
Config{
|
|
OpConfig: config.Config{
|
|
ProtectedRoles: []string{"admin"},
|
|
Auth: config.Auth{
|
|
SuperUsername: superUserName,
|
|
ReplicationUsername: replicationUserName,
|
|
},
|
|
ConnectionPool: config.ConnectionPool{
|
|
ConnPoolDefaultCPURequest: "100m",
|
|
ConnPoolDefaultCPULimit: "100m",
|
|
ConnPoolDefaultMemoryRequest: "100M",
|
|
ConnPoolDefaultMemoryLimit: "100M",
|
|
},
|
|
},
|
|
}, k8sutil.KubernetesClient{}, acidv1.Postgresql{}, logger)
|
|
|
|
cluster.Statefulset = &appsv1.StatefulSet{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "test-sts",
|
|
},
|
|
}
|
|
|
|
clusterMissingObjects := *cluster
|
|
clusterMissingObjects.KubeClient = k8sutil.ClientMissingObjects()
|
|
|
|
clusterMock := *cluster
|
|
clusterMock.KubeClient = k8sutil.NewMockKubernetesClient()
|
|
|
|
tests := []struct {
|
|
subTest string
|
|
oldSpec *acidv1.Postgresql
|
|
newSpec *acidv1.Postgresql
|
|
cluster *Cluster
|
|
check func(cluster *Cluster, err error) error
|
|
}{
|
|
{
|
|
subTest: "create if doesn't exist",
|
|
oldSpec: &acidv1.Postgresql{
|
|
Spec: acidv1.PostgresSpec{
|
|
ConnectionPool: &acidv1.ConnectionPool{},
|
|
},
|
|
},
|
|
newSpec: &acidv1.Postgresql{
|
|
Spec: acidv1.PostgresSpec{
|
|
ConnectionPool: &acidv1.ConnectionPool{},
|
|
},
|
|
},
|
|
cluster: &clusterMissingObjects,
|
|
check: objectsAreSaved,
|
|
},
|
|
{
|
|
subTest: "update deployment",
|
|
oldSpec: &acidv1.Postgresql{
|
|
Spec: acidv1.PostgresSpec{
|
|
ConnectionPool: &acidv1.ConnectionPool{
|
|
NumberOfInstances: int32ToPointer(1),
|
|
},
|
|
},
|
|
},
|
|
newSpec: &acidv1.Postgresql{
|
|
Spec: acidv1.PostgresSpec{
|
|
ConnectionPool: &acidv1.ConnectionPool{
|
|
NumberOfInstances: int32ToPointer(2),
|
|
},
|
|
},
|
|
},
|
|
cluster: &clusterMock,
|
|
check: deploymentUpdated,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
err := tt.cluster.syncConnectionPool(tt.oldSpec, tt.newSpec)
|
|
|
|
if err := tt.check(tt.cluster, err); err != nil {
|
|
t.Errorf("%s [%s]: Could not synchronize, %+v",
|
|
testName, tt.subTest, err)
|
|
}
|
|
}
|
|
}
|