package store

import (
	"context"
	"testing"
)

func TestSetTaskFolderMappingRoundTrip(t *testing.T) {
	s := testStore(t)
	ctx := context.Background()
	e1, _ := s.CreateEndpoint(ctx, Endpoint{RoleLabel: "s", Host: "a", Port: 993, TLSMode: "ssl"})
	e2, _ := s.CreateEndpoint(ctx, Endpoint{RoleLabel: "d", Host: "b", Port: 993, TLSMode: "ssl"})
	taskID, _ := s.CreateTask(ctx, Task{Name: "t", SrcEndpointID: e1, DstEndpointID: e2})

	m := map[string]string{"Спам": "Spam", "Отправленные": "Sent"}
	if err := s.SetTaskFolderMapping(ctx, taskID, m); err != nil {
		t.Fatalf("set mapping: %v", err)
	}
	got, err := s.GetTask(ctx, taskID)
	if err != nil {
		t.Fatalf("get: %v", err)
	}
	if got.FolderMapping["Спам"] != "Spam" || got.FolderMapping["Отправленные"] != "Sent" {
		t.Fatalf("mapping round-trip failed: %+v", got.FolderMapping)
	}
	// nil clears to empty object, not null
	if err := s.SetTaskFolderMapping(ctx, taskID, nil); err != nil {
		t.Fatalf("clear: %v", err)
	}
	got, _ = s.GetTask(ctx, taskID)
	if got.FolderMapping == nil {
		t.Fatal("mapping should be non-nil empty map after clear")
	}
}