package main

import (
	"encoding/json"
	"fmt"
	"testing"

	"github.com/stretchr/testify/assert"
)

func Test_zabbixParamOutput(t *testing.T) {
	tests := []struct {
		name  string
		input ZabbixAPIParams
		want  string
	}{
		{
			name: "Mode extend",
			input: ZabbixAPIParams{
				Output: &zabbixParamOutput{
					Mode: "extend",
				},
				GroupIDs: []string{"test1", "test2"},
			},
			want: `{ "output": "extend", "groupids": ["test1", "test2"] }`,
		},
		{
			name: "Fields",
			input: ZabbixAPIParams{
				Output: &zabbixParamOutput{
					Fields: []string{"name", "key_", "hostid"},
				},
				GroupIDs: []string{"test1", "test2"},
			},
			want: `{ "output": ["name", "key_", "hostid"], "groupids": ["test1", "test2"] }`,
		},
		{
			name: "No Output",
			input: ZabbixAPIParams{
				GroupIDs: []string{"test1", "test2"},
			},
			want: `{ "groupids": ["test1", "test2"] }`,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			jsonOut, err := json.Marshal(tt.input)
			fmt.Printf("Output: %s\n", jsonOut)
			assert.NoError(t, err)
			if !assert.JSONEq(t, tt.want, string(jsonOut)) {
				return
			}

			objOut := ZabbixAPIParams{}
			err = json.Unmarshal(jsonOut, &objOut)
			assert.NoError(t, err)
			assert.Equal(t, tt.input, objOut)
		})
	}
}