package models

import (
	"testing"
	"time"

	"go.mongodb.org/mongo-driver/v2/bson"
)

func TestInstanceLockFieldsBSON(t *testing.T) {
	at := time.Date(2026, 9, 10, 10, 0, 0, 0, time.UTC)

	raw, err := bson.Marshal(Instance{InstanceID: "i1", LockedAt: &at, PurgeAfter: &at})
	if err != nil {
		t.Fatalf("marshal: %v", err)
	}
	var set bson.M
	if err := bson.Unmarshal(raw, &set); err != nil {
		t.Fatalf("unmarshal: %v", err)
	}
	for _, k := range []string{"locked_at", "purge_after"} {
		if _, ok := set[k]; !ok {
			t.Errorf("%s missing from %v", k, set)
		}
	}

	raw, err = bson.Marshal(Instance{InstanceID: "i1"})
	if err != nil {
		t.Fatalf("marshal: %v", err)
	}
	var unset bson.M
	if err := bson.Unmarshal(raw, &unset); err != nil {
		t.Fatalf("unmarshal: %v", err)
	}
	for _, k := range []string{"locked_at", "purge_after"} {
		if _, ok := unset[k]; ok {
			t.Errorf("%s written for an unlocked instance: %v", k, unset)
		}
	}
}