@@ -2,6 +2,7 @@ package store
22
33import (
44 "context"
5+ "k8s.io/client-go/tools/cache"
56 "testing"
67 "time"
78
@@ -20,23 +21,23 @@ func TestVersioner(t *testing.T) {
2021 t .Run ("increment version for new resource" , func (t * testing.T ) {
2122 v := newVersioner ()
2223 version := v .increment ("test-work" )
23- if version != 0 {
24- t .Errorf ("expected version 0 for new resource, got %d" , version )
24+ if version != 1 {
25+ t .Errorf ("expected version 1 for new resource, got %d" , version )
2526 }
2627 })
2728
2829 t .Run ("increment version for existing resource" , func (t * testing.T ) {
2930 v := newVersioner ()
3031 v .increment ("test-work" )
3132 version := v .increment ("test-work" )
32- if version != 1 {
33- t .Errorf ("expected version 1 for existing resource, got %d" , version )
33+ if version != 2 {
34+ t .Errorf ("expected version 2 for existing resource, got %d" , version )
3435 }
3536 })
3637
3738 t .Run ("increment multiple times" , func (t * testing.T ) {
3839 v := newVersioner ()
39- for i := 0 ; i < 5 ; i ++ {
40+ for i := 1 ; i <= 5 ; i ++ {
4041 version := v .increment ("test-work" )
4142 if version != int64 (i ) {
4243 t .Errorf ("expected version %d, got %d" , i , version )
@@ -49,8 +50,8 @@ func TestVersioner(t *testing.T) {
4950 v .increment ("test-work" )
5051 v .delete ("test-work" )
5152 version := v .increment ("test-work" )
52- if version != 0 {
53- t .Errorf ("expected version 0 after delete, got %d" , version )
53+ if version != 1 {
54+ t .Errorf ("expected version 1 after delete, got %d" , version )
5455 }
5556 })
5657
@@ -70,14 +71,15 @@ func TestVersioner(t *testing.T) {
7071 }
7172
7273 finalVersion := v .increment ("test-work" )
73- if finalVersion != 10 {
74- t .Errorf ("expected version 10 after 10 concurrent increments, got %d" , finalVersion )
74+ if finalVersion != 11 {
75+ t .Errorf ("expected version 11 after 10 concurrent increments, got %d" , finalVersion )
7576 }
7677 })
7778}
7879
7980func TestAgentInformerWatcherStore_Add (t * testing.T ) {
8081 store := NewAgentInformerWatcherStore ()
82+ store .Store = cache .NewIndexer (cache .DeletionHandlingMetaNamespaceKeyFunc , cache.Indexers {})
8183
8284 // Start consuming events to prevent blocking
8385 watcher , err := store .GetWatcher ("" , metav1.ListOptions {})
@@ -118,8 +120,8 @@ func TestAgentInformerWatcherStore_Add(t *testing.T) {
118120 t .Fatalf ("unexpected error adding work: %v" , err )
119121 }
120122
121- if work .ResourceVersion != "0 " {
122- t .Errorf ("expected resource version 0 , got %s" , work .ResourceVersion )
123+ if work .ResourceVersion != "1 " {
124+ t .Errorf ("expected resource version 1 , got %s" , work .ResourceVersion )
123125 }
124126
125127 // Add another work
@@ -135,13 +137,14 @@ func TestAgentInformerWatcherStore_Add(t *testing.T) {
135137 t .Fatalf ("unexpected error adding work: %v" , err )
136138 }
137139
138- if work2 .ResourceVersion != "1 " {
139- t .Errorf ("expected resource version 1 for second add, got %s" , work2 .ResourceVersion )
140+ if work2 .ResourceVersion != "2 " {
141+ t .Errorf ("expected resource version 2 for second add, got %s" , work2 .ResourceVersion )
140142 }
141143}
142144
143145func TestAgentInformerWatcherStore_Update (t * testing.T ) {
144146 store := NewAgentInformerWatcherStore ()
147+ store .Store = cache .NewIndexer (cache .DeletionHandlingMetaNamespaceKeyFunc , cache.Indexers {})
145148
146149 // Start consuming events to prevent blocking
147150 watcher , err := store .GetWatcher ("" , metav1.ListOptions {})
@@ -198,8 +201,8 @@ func TestAgentInformerWatcherStore_Update(t *testing.T) {
198201 t .Fatalf ("unexpected error updating work: %v" , err )
199202 }
200203
201- if updatedWork .ResourceVersion != "1 " {
202- t .Errorf ("expected resource version 1 after update, got %s" , updatedWork .ResourceVersion )
204+ if updatedWork .ResourceVersion != "2 " {
205+ t .Errorf ("expected resource version 2 after update, got %s" , updatedWork .ResourceVersion )
203206 }
204207
205208 // Update again
@@ -208,13 +211,14 @@ func TestAgentInformerWatcherStore_Update(t *testing.T) {
208211 t .Fatalf ("unexpected error updating work: %v" , err )
209212 }
210213
211- if updatedWork .ResourceVersion != "2 " {
212- t .Errorf ("expected resource version 2 after second update, got %s" , updatedWork .ResourceVersion )
214+ if updatedWork .ResourceVersion != "3 " {
215+ t .Errorf ("expected resource version 3 after second update, got %s" , updatedWork .ResourceVersion )
213216 }
214217}
215218
216219func TestAgentInformerWatcherStore_Delete (t * testing.T ) {
217220 store := NewAgentInformerWatcherStore ()
221+ store .Store = cache .NewIndexer (cache .DeletionHandlingMetaNamespaceKeyFunc , cache.Indexers {})
218222
219223 // Start consuming events to prevent blocking
220224 watcher , err := store .GetWatcher ("" , metav1.ListOptions {})
@@ -267,7 +271,7 @@ func TestAgentInformerWatcherStore_Delete(t *testing.T) {
267271 t .Fatalf ("unexpected error deleting work: %v" , err )
268272 }
269273
270- // Add the same work again, version should be reset to 0
274+ // Add the same work again, version should be reset to 1
271275 newWork := & workv1.ManifestWork {
272276 ObjectMeta : metav1.ObjectMeta {
273277 Name : "test-work" ,
@@ -280,13 +284,14 @@ func TestAgentInformerWatcherStore_Delete(t *testing.T) {
280284 t .Fatalf ("unexpected error adding work after delete: %v" , err )
281285 }
282286
283- if newWork .ResourceVersion != "0 " {
284- t .Errorf ("expected resource version 0 after delete and re-add, got %s" , newWork .ResourceVersion )
287+ if newWork .ResourceVersion != "1 " {
288+ t .Errorf ("expected resource version 1 after delete and re-add, got %s" , newWork .ResourceVersion )
285289 }
286290}
287291
288292func TestAgentInformerWatcherStore_ResourceVersionIncrement (t * testing.T ) {
289293 store := NewAgentInformerWatcherStore ()
294+ store .Store = cache .NewIndexer (cache .DeletionHandlingMetaNamespaceKeyFunc , cache.Indexers {})
290295
291296 // Start consuming events to prevent blocking
292297 watcher , err := store .GetWatcher ("" , metav1.ListOptions {})
@@ -323,7 +328,7 @@ func TestAgentInformerWatcherStore_ResourceVersionIncrement(t *testing.T) {
323328 }
324329
325330 // Verify resource versions increment properly
326- expectedVersions := []string {"0 " , "1 " , "2 " , "3 " }
331+ expectedVersions := []string {"1 " , "2 " , "3 " , "4 " }
327332
328333 // Add
329334 err = store .Add (work )
0 commit comments