wehub-resource-sync
30 lines
1.1 KiB
Go
30 lines
1.1 KiB
Go
package prefixcache_test
|
|
|
|
import (
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
|
|
"github.com/mudler/LocalAI/core/services/nodes/prefixcache"
|
|
)
|
|
|
|
var _ = Describe("RoutePolicy", func() {
|
|
It("parses known values and defaults unknown to Default (zero)", func() {
|
|
Expect(prefixcache.ParsePolicy("round_robin")).To(Equal(prefixcache.RoutePolicyRoundRobin))
|
|
Expect(prefixcache.ParsePolicy("prefix_cache")).To(Equal(prefixcache.RoutePolicyPrefixCache))
|
|
Expect(prefixcache.ParsePolicy("")).To(Equal(prefixcache.RoutePolicyDefault))
|
|
Expect(prefixcache.ParsePolicy("bogus")).To(Equal(prefixcache.RoutePolicyDefault))
|
|
})
|
|
|
|
It("stringifies", func() {
|
|
Expect(prefixcache.RoutePolicyPrefixCache.String()).To(Equal("prefix_cache"))
|
|
Expect(prefixcache.RoutePolicyRoundRobin.String()).To(Equal("round_robin"))
|
|
})
|
|
|
|
It("resolves per-model against a global default", func() {
|
|
Expect(prefixcache.RoutePolicyDefault.Resolve(prefixcache.RoutePolicyPrefixCache)).
|
|
To(Equal(prefixcache.RoutePolicyPrefixCache))
|
|
Expect(prefixcache.RoutePolicyRoundRobin.Resolve(prefixcache.RoutePolicyPrefixCache)).
|
|
To(Equal(prefixcache.RoutePolicyRoundRobin))
|
|
})
|
|
})
|