Source of handlers_test.go - NVIDIA GPU

Raw file
Source viewDiff to previous
4.34 KB
xxxxxxxxxx
 
/*
** Copyright (C) 2001-2024 Zabbix SIA
**
** Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
** documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
** rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
** permit persons to whom the Software is furnished to do so, subject to the following conditions:
**
** The above copyright notice and this permission notice shall be included in all copies or substantial portions
** of the Software.
**
** THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
** WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
** COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
** TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
** SOFTWARE.
**/
​
package handlers
​
import (
    "context"
    "testing"
​
    "github.com/google/go-cmp/cmp"
    "golang.zabbix.com/plugin/nvidia/pkg/nvml"
    nvmlmock "golang.zabbix.com/plugin/nvidia/pkg/nvml-mock"
    "golang.zabbix.com/sdk/errs"
)
​
func TestWithJSONResponse(t *testing.T) {
    t.Parallel()
​
    type args struct {
        value  any
        gotErr bool
    }
​
    tests := []struct {
        name    string
        args    args
        want    any
        wantErr bool
    }{
        {
            "+valid",
            args{
                value: "foobar",
            },
            `"foobar"`,
            false,
        },
        {
            "+jsonObject",
            args{
                value: map[string]string{
                    "foo":  "bar",
                    "test": "true",
                },
            },
            `{"foo":"bar","test":"true"}`,
            false,
        },
        {
            "-jsonMarshalErr",
            args{
                value: map[struct{ test string }]string{
                    {test: "1"}: "bar",
                    {test: "2"}: "foo",
                },
            },
            nil,
            true,
        },
        {
            "-handlerErr",
            args{gotErr: true},
            nil,
            true,
        },
    }
​
    for _, tt := range tests {
        tt := tt
        t.Run(tt.name, func(t *testing.T) {
            t.Parallel()
​
            got, err := WithJSONResponse(
                func(ctx context.Context, metricParams map[string]string, extraParams ...string) (any, error) {
                    if tt.args.gotErr {
                        return nil, errs.New("fail")