@@ -70,6 +70,8 @@ class ToggleServiceSpec extends Specification {
7070
7171 IncludeArchivedToggleService includeArchivedToggleService
7272
73+ Set<String > toggleKeys = new HashSet ();
74+
7375 def projectKey
7476 def environmentKey
7577 def toggleKey
@@ -102,6 +104,7 @@ class ToggleServiceSpec extends Specification {
102104 toggleKey = " feature_toggle_unit_test"
103105 toggleName = " test_toggle"
104106 sdkKey = " server-123456"
107+ toggleKeys. add(toggleKey)
105108 rules = " {\" rules\" :[{\" conditions\" :[{\" type\" :\" string\" ,\" subject\" :\" city\" ,\" predicate\" :\" is one of\" ," +
106109 " \" objects\" :[\" Paris\" ]},{\" type\" :\" segment\" ,\" subject\" :\"\" ,\" predicate\" :\" is in\" ," +
107110 " \" objects\" :[\" test_segment\" ]},{\" type\" :\" number\" ,\" subject\" :\" age\" ,\" predicate\" :\" =\" ," +
@@ -142,7 +145,7 @@ class ToggleServiceSpec extends Specification {
142145 false ) >> [new Targeting (toggleKey : toggleKey)]
143146 1 * tagRepository. findByNameIn([" test" ]) >> [new Tag (name : " test" , toggles : [new Toggle (key : toggleKey)])]
144147 1 * toggleTagRepository. findByTagIdIn(_) >> [new ToggleTagRelation (toggleKey : toggleKey)]
145- 1 * eventRepository. findAll(_ ) >> [ new Event ( toggleKey : toggleKey)]
148+ 1 * eventRepository. findAllAccessedToggleKeyGreaterThanOrEqualToEndDate(_, _, _ ) >> toggleKeys
146149 1 * toggleRepository. findAll(_, _) >> new PageImpl<> ([new Toggle (key : toggleKey, projectKey : projectKey)],
147150 Pageable . ofSize(1 ), 1 )
148151 1 * targetingRepository. findByProjectKeyAndEnvironmentKeyAndToggleKeyIn(projectKey, environmentKey, [toggleKey]) >>
@@ -176,7 +179,8 @@ class ToggleServiceSpec extends Specification {
176179 false ) >> [new Targeting (toggleKey : toggleKey)]
177180 1 * tagRepository. findByNameIn([" test" ]) >> [new Tag (name : " test" , toggles : [new Toggle (key : toggleKey)])]
178181 1 * toggleTagRepository. findByTagIdIn(_) >> [new ToggleTagRelation (toggleKey : toggleKey)]
179- 2 * eventRepository. findAll(_) >> [new Event (toggleKey : toggleKey)]
182+ 1 * eventRepository. findAllAccessedToggleKey(_, _) >> toggleKeys
183+ 1 * eventRepository. findAllAccessedToggleKeyGreaterThanOrEqualToEndDate(_, _, _) >> toggleKeys
180184 1 * toggleRepository. findAll(_, _) >> new PageImpl<> ([new Toggle (key : toggleKey, projectKey : projectKey)],
181185 Pageable . ofSize(1 ), 1 )
182186 1 * targetingRepository. findByProjectKeyAndEnvironmentKeyAndToggleKeyIn(projectKey, environmentKey, [toggleKey]) >>
@@ -210,7 +214,7 @@ class ToggleServiceSpec extends Specification {
210214 false ) >> [new Targeting (toggleKey : toggleKey)]
211215 1 * tagRepository. findByNameIn([" test" ]) >> [new Tag (name : " test" , toggles : [new Toggle (key : toggleKey)])]
212216 1 * toggleTagRepository. findByTagIdIn(_) >> [new ToggleTagRelation (toggleKey : toggleKey)]
213- 1 * eventRepository. findAll(_ ) >> [ new Event ( toggleKey : toggleKey)]
217+ 1 * eventRepository. findAllAccessedToggleKey(_, _ ) >> toggleKeys
214218 1 * toggleRepository. findAll(_) >> [new Toggle (key : toggleKey, projectKey : projectKey)]
215219 1 * toggleRepository. findAll(_, _) >> new PageImpl<> ([new Toggle (key : toggleKey, projectKey : projectKey)],
216220 Pageable . ofSize(1 ), 1 )
@@ -245,7 +249,7 @@ class ToggleServiceSpec extends Specification {
245249 false ) >> [new Targeting (toggleKey : toggleKey)]
246250 1 * tagRepository. findByNameIn([" test" ]) >> [new Tag (name : " test" , toggles : [new Toggle (key : toggleKey)])]
247251 1 * toggleTagRepository. findByTagIdIn(_) >> [new ToggleTagRelation (toggleKey : toggleKey)]
248- 1 * eventRepository. findAll(_ ) >> [ new Event ( toggleKey : toggleKey)]
252+ 1 * eventRepository. findAllAccessedToggleKeyGreaterThanOrEqualToEndDate(_, _, _ ) >> toggleKeys
249253 1 * toggleRepository. findAll(_, _) >> new PageImpl<> ([new Toggle (key : toggleKey, projectKey : projectKey)],
250254 Pageable . ofSize(1 ), 1 )
251255 1 * targetingRepository. findByProjectKeyAndEnvironmentKeyAndToggleKeyIn(projectKey, environmentKey, [toggleKey]) >>
0 commit comments