package apollo

import (
	"testing"

	"github.com/apolloconfig/agollo/v4/storage"

	"github.com/go-kratos/kratos/v2/config"
	"github.com/go-kratos/kratos/v2/encoding"
)

func Test_onChange(t *testing.T) {
	s := map[string]struct {
		Name string `yaml:"name"`
	}{
		"app": {
			Name: "new",
		},
	}
	codec := encoding.GetCodec(yaml)
	val, _ := codec.Marshal(s)
	c := customChangeListener{}
	tests := []struct {
		name      string
		namespace string
		changes   map[string]*storage.ConfigChange
		kvs       []*config.KeyValue
	}{
		{
			"test yaml onChange",
			"app.yaml",
			map[string]*storage.ConfigChange{
				"name": {
					OldValue:   "old",
					NewValue:   "new",
					ChangeType: storage.MODIFIED,
				},
			},
			[]*config.KeyValue{
				{
					Key:    "app.yaml",
					Value:  val,
					Format: yaml,
				},
			},
		},
		{
			"test json onChange",
			"app.json",
			map[string]*storage.ConfigChange{
				"content": {
					OldValue:   `{"name":"old"}`,
					NewValue:   `{"name":"new"}`,
					ChangeType: storage.MODIFIED,
				},
			},
			[]*config.KeyValue{
				{
					Key:    "app.json",
					Value:  []byte(`{"name":"new"}`),
					Format: json,
				},
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			kvs := c.onChange(tt.namespace, tt.changes)
			if len(kvs) != len(tt.kvs) {
				t.Errorf("len(kvs) = %v, want %v", len(kvs), len(tt.kvs))
			}
			for i := range kvs {
				if kvs[i].Format != tt.kvs[i].Format || kvs[i].Key != tt.kvs[i].Key || string(kvs[i].Value) != string(tt.kvs[i].Value) {
					t.Errorf("got %v, want %v", kvs[i], tt.kvs[i])
				}
			}
		})
	}
}