PostgreSQL

//go:build postgresql_tests

// +build postgresql_tests

​

/*

** Zabbix

** Copyright 2001-2024 Zabbix SIA

**

** Licensed under the Apache License, Version 2.0 (the "License");

** you may not use this file except in compliance with the License.

** You may obtain a copy of the License at

**

**     http://www.apache.org/licenses/LICENSE-2.0

**

** Unless required by applicable law or agreed to in writing, software

** distributed under the License is distributed on an "AS IS" BASIS,

** WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

** See the License for the specific language governing permissions and

** limitations under the License.

**/

​

package plugin

​

import (

    "context"

    "fmt"

    "testing"

)

​

func TestPlugin_replicationHandler(t *testing.T) {

    // create pool or acquire conn from old pool

    sharedPool, err := getConnPool()

    if err != nil {

        t.Fatal(err)

    }

​

    type args struct {

        ctx         context.Context

        conn        *PGConn

        key         string

        params      map[string]string

        extraParams []string

    }

    tests := []struct {

        name    string

        p       *Plugin

        args    args

        wantErr bool

    }{

        {

            fmt.Sprintf("replicationHandler should return ptr to Pool for replication.count"),

            &Impl,

            args{context.Background(), sharedPool, keyReplicationCount, nil, []string{}},

            false,

        },

        {

            fmt.Sprintf("replicationHandler should return ptr to Pool for replication.status"),

            &Impl,

            args{context.Background(), sharedPool, keyReplicationStatus, nil, []string{}},

            false,

        },

        {

            fmt.Sprintf("replicationHandler should return ptr to Pool for replication.lag.sec"),

            &Impl,