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"

    "errors"

    "fmt"

    "testing"

)

​

func TestPlugin_queriesHandler(t *testing.T) {

    timePeriod := map[string]string{"TimePeriod": "42"}

​

    // create pool or acquire conn from old pool for test

    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("Plugin.queriesHandler() should return ptr to Pool for Plugin.queriesHandler()"),

            &Impl,

            args{context.Background(), sharedPool, keyQueries, timePeriod, []string{}},

            false,

        },

    }

    for _, tt := range tests {

        t.Run(tt.name, func(t *testing.T) {

            got, err := queriesHandler(tt.args.ctx, tt.args.conn, tt.args.key, tt.args.params, tt.args.extraParams...)

            if (err != nil) != tt.wantErr {

                t.Errorf("Plugin.queriesHandler() error = %v, wantErr %v", err, tt.wantErr)

                return

            }

            if len(got.(string)) == 0 && err != errors.New("cannot parse data") {

                t.Errorf("Plugin.queriesHandler() error = %v, wantErr %v", err, tt.wantErr)

                return

            }

        })

    }

}