package messaging_test

import (
	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"

	"github.com/mudler/LocalAI/core/services/messaging"
)

var _ = Describe("PrefixCache subjects", func() {
	It("exposes stable subject constants", func() {
		Expect(messaging.SubjectPrefixCacheObserve).To(Equal("prefixcache.observe"))
		Expect(messaging.SubjectPrefixCacheInvalidate).To(Equal("prefixcache.invalidate"))
	})

	It("carries a replica index on the observe event", func() {
		ev := messaging.PrefixCacheObserveEvent{Model: "m", Chain: []uint64{1, 2}, NodeID: "A", Replica: 3}
		Expect(ev.Replica).To(Equal(3))
	})

	It("uses a negative replica on the invalidate event to mean all replicas of a node", func() {
		all := messaging.PrefixCacheInvalidateEvent{Model: "m", NodeID: "A", Replica: -1}
		Expect(all.Replica).To(BeNumerically("<", 0))
		one := messaging.PrefixCacheInvalidateEvent{Model: "m", NodeID: "A", Replica: 0}
		Expect(one.Replica).To(Equal(0))
	})
})