130 lines
3.6 KiB
Go
130 lines
3.6 KiB
Go
package cluster
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
acidv1 "github.com/zalando/postgres-operator/pkg/apis/acid.zalan.do/v1"
|
|
"github.com/zalando/postgres-operator/pkg/util"
|
|
"github.com/zalando/postgres-operator/pkg/util/config"
|
|
"github.com/zalando/postgres-operator/pkg/util/k8sutil"
|
|
)
|
|
|
|
func newGenerateConfigCluster() *Cluster {
|
|
maxDB := int32(60)
|
|
instances := int32(2)
|
|
cluster := New(
|
|
Config{OpConfig: config.Config{
|
|
ConnectionPooler: config.ConnectionPooler{
|
|
User: "pooler",
|
|
Schema: "pooler",
|
|
Mode: "transaction",
|
|
MaxDBConnections: &maxDB,
|
|
NumberOfInstances: &instances,
|
|
GenerateConfig: true,
|
|
AuthType: "scram-sha-256",
|
|
ConfigPath: "/etc/pgbouncer/pgbouncer.ini",
|
|
Args: []string{"/etc/pgbouncer/pgbouncer.ini"},
|
|
},
|
|
Resources: config.Resources{
|
|
EnableOwnerReferences: util.True(),
|
|
},
|
|
}},
|
|
k8sutil.NewMockKubernetesClient(), acidv1.Postgresql{}, logger, eventRecorder)
|
|
cluster.Name = "acid-test"
|
|
cluster.Namespace = "default"
|
|
cluster.Spec = acidv1.PostgresSpec{ConnectionPooler: &acidv1.ConnectionPooler{}}
|
|
return cluster
|
|
}
|
|
|
|
func TestGeneratePgBouncerIni(t *testing.T) {
|
|
cluster := newGenerateConfigCluster()
|
|
|
|
ini, err := cluster.generatePgBouncerIni(Master)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
|
|
for _, want := range []string{
|
|
"[databases]",
|
|
"[pgbouncer]",
|
|
"pool_mode = transaction",
|
|
"auth_type = scram-sha-256",
|
|
"auth_file = /etc/pgbouncer/userlist.txt",
|
|
"auth_query = SELECT * FROM pooler.user_lookup($1)",
|
|
"server_tls_sslmode = require",
|
|
"default_pool_size = 15",
|
|
"max_db_connections = 30",
|
|
} {
|
|
if !strings.Contains(ini, want) {
|
|
t.Errorf("rendered ini missing %q\n---\n%s", want, ini)
|
|
}
|
|
}
|
|
|
|
if strings.Contains(ini, "client_tls_cert_file") {
|
|
t.Errorf("did not expect client_tls_cert_file without spec.TLS\n%s", ini)
|
|
}
|
|
}
|
|
|
|
func TestGeneratePgBouncerIniWithTLS(t *testing.T) {
|
|
cluster := newGenerateConfigCluster()
|
|
cluster.Spec.TLS = &acidv1.TLSDescription{SecretName: "pg-tls"}
|
|
|
|
ini, err := cluster.generatePgBouncerIni(Master)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
for _, want := range []string{
|
|
"client_tls_sslmode = require",
|
|
"client_tls_key_file = /tls/tls.key",
|
|
"client_tls_cert_file = /tls/tls.crt",
|
|
} {
|
|
if !strings.Contains(ini, want) {
|
|
t.Errorf("rendered ini missing %q\n---\n%s", want, ini)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestConnectionPoolerConfigChecksumStability(t *testing.T) {
|
|
cluster := newGenerateConfigCluster()
|
|
|
|
sum1, err := cluster.connectionPoolerConfigChecksum(Master)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
sum2, err := cluster.connectionPoolerConfigChecksum(Master)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if sum1 != sum2 {
|
|
t.Errorf("checksum not stable: %q != %q", sum1, sum2)
|
|
}
|
|
|
|
cluster.OpConfig.ConnectionPooler.AuthType = "md5"
|
|
sum3, err := cluster.connectionPoolerConfigChecksum(Master)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if sum1 == sum3 {
|
|
t.Errorf("checksum should change when config changes")
|
|
}
|
|
}
|
|
|
|
func TestGenerateConnectionPoolerConfigMap(t *testing.T) {
|
|
cluster := newGenerateConfigCluster()
|
|
|
|
cm, err := cluster.generateConnectionPoolerConfigMap(Master)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if cm.Name != cluster.connectionPoolerName(Master)+"-config" {
|
|
t.Errorf("unexpected config map name %q", cm.Name)
|
|
}
|
|
if _, ok := cm.Data["pgbouncer.ini"]; !ok {
|
|
t.Errorf("config map missing pgbouncer.ini key, got %#v", cm.Data)
|
|
}
|
|
if len(cm.OwnerReferences) == 0 {
|
|
t.Errorf("config map should have owner references")
|
|
}
|
|
}
|