model_test.go

package sls

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

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

func TestIndex_MarshalJSON(t *testing.T) {

	type fields struct {
		Keys                   map[string]IndexKey
		Line                   *IndexLine
		Ttl                    uint32
		MaxTextLen             uint32
		LogReduce              bool
		LogReduceWhiteListDict []string
		LogReduceBlackListDict []string
	}
	tests := []struct {
		name   string
		fields fields
		want   []byte
	}{
		{
			name: "keys and line",
			fields: fields{
				Keys: map[string]IndexKey{
					"test1": {},
				},
				Line: &IndexLine{},
			},
			want: []byte(`{"keys":{"test1":{"token":null,"caseSensitive":false,"type":"","chn":false}},"line":{"token":null,"caseSensitive":false,"chn":false},"log_reduce":false}`),
		},
		{
			name: "only ttl",
			fields: fields{
				Ttl:        2,
				MaxTextLen: 3,
			},
			want: []byte(`{"ttl":2,"max_text_len":3,"log_reduce":false}`),
		},
		{
			name: "white & black",
			fields: fields{
				LogReduceWhiteListDict: []string{"key1"},
				LogReduceBlackListDict: []string{"key2"},
				LogReduce:              true,
			},
			want: []byte(`{"log_reduce":true,"log_reduce_white_list":["key1"],"log_reduce_black_list":["key2"]}`),
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			u := &Index{
				Keys:                   tt.fields.Keys,
				Line:                   tt.fields.Line,
				Ttl:                    tt.fields.Ttl,
				MaxTextLen:             tt.fields.MaxTextLen,
				LogReduce:              tt.fields.LogReduce,
				LogReduceWhiteListDict: tt.fields.LogReduceWhiteListDict,
				LogReduceBlackListDict: tt.fields.LogReduceBlackListDict,
			}
			got, _ := json.Marshal(u)
			fmt.Printf("%s", got)
			assert.Equalf(t, tt.want, got, "MarshalJSON()")
		})
	}

}