@@ -49,44 +49,7 @@ var _ = Describe("DistributedModelStore", Label("Distributed"), func() {
4949 Expect (m ).To (Equal (expected ))
5050 })
5151
52- It ("falls back to DB when local cache misses, returns model with correct address" , func () {
53- // Register a node and load a model on it in the DB
54- node := & nodes.BackendNode {
55- Name : "remote-node" , Address : "remote:9000" ,
56- }
57- Expect (registry .Register (context .Background (), node , true )).To (Succeed ())
58- Expect (registry .SetNodeModel (context .Background (), node .ID , "db-model" , "loaded" )).To (Succeed ())
59-
60- m , ok := dStore .Get ("db-model" )
61- Expect (ok ).To (BeTrue ())
62- Expect (m ).ToNot (BeNil ())
63- Expect (m .ID ).To (Equal ("db-model" ))
64- })
65-
66- It ("caches DB result locally — second Get hits local cache" , func () {
67- node := & nodes.BackendNode {
68- Name : "cache-node" , Address : "cache:9000" ,
69- }
70- Expect (registry .Register (context .Background (), node , true )).To (Succeed ())
71- Expect (registry .SetNodeModel (context .Background (), node .ID , "cached-model" , "loaded" )).To (Succeed ())
72-
73- // First Get — falls back to DB
74- m1 , ok := dStore .Get ("cached-model" )
75- Expect (ok ).To (BeTrue ())
76-
77- // Should now be in local store
78- localM , localOK := localStore .Get ("cached-model" )
79- Expect (localOK ).To (BeTrue ())
80- Expect (localM ).To (Equal (m1 ))
81-
82- // Second Get — hits local cache (even if we remove from DB)
83- Expect (registry .RemoveNodeModel (context .Background (), node .ID , "cached-model" )).To (Succeed ())
84- m2 , ok := dStore .Get ("cached-model" )
85- Expect (ok ).To (BeTrue ())
86- Expect (m2 ).To (Equal (m1 ))
87- })
88-
89- It ("returns (nil, false) when model not in local or DB" , func () {
52+ It ("returns (nil, false) when model is not in local cache" , func () {
9053 m , ok := dStore .Get ("ghost-model" )
9154 Expect (ok ).To (BeFalse ())
9255 Expect (m ).To (BeNil ())
0 commit comments