PostgreSQL

/*

** 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"

    "reflect"

    "testing"

​

    "github.com/DATA-DOG/go-sqlmock"

)

​

func Test_versionHandler(t *testing.T) {

    type mock struct {

        row *sqlmock.Rows

        err error

    }

​

    tests := []struct {

        name    string

        mock    mock

        want    any

        wantErr bool

    }{

        {

            "+valid",

            mock{

                row: sqlmock.NewRows([]string{"version"}).AddRow("postgres 69"),

            },

            "postgres 69",

            false,

        },

        {

            "-queryErr",

            mock{

                row: sqlmock.NewRows([]string{"version"}).AddRow("postgres 69"),

                err: errors.New("query err"),

            },

            nil,

            true,

        },

        {

            "-noRows",

            mock{

                row: sqlmock.NewRows([]string{"version"}),

            },

            nil,