@@ -158,17 +158,20 @@ describe('splitChangesUpdater', () => {
158158 const fetchSplitChanges = jest . spyOn ( splitApi , 'fetchSplitChanges' ) ;
159159 const splitChangesFetcher = splitChangesFetcherFactory ( splitApi . fetchSplitChanges ) ;
160160
161- const splitsCache = new SplitsCacheInMemory ( ) ;
162- const updateSplits = jest . spyOn ( splitsCache , 'update' ) ;
161+ const splits = new SplitsCacheInMemory ( ) ;
162+ const updateSplits = jest . spyOn ( splits , 'update' ) ;
163+
164+ const segments = new SegmentsCacheInMemory ( ) ;
165+ const registerSegments = jest . spyOn ( segments , 'registerSegments' ) ;
166+
167+ const storage = { splits, segments } ;
163168
164- const segmentsCache = new SegmentsCacheInMemory ( ) ;
165- const registerSegments = jest . spyOn ( segmentsCache , 'registerSegments' ) ;
166169 const readinessManager = readinessManagerFactory ( EventEmitter , fullSettings ) ;
167170 const splitsEmitSpy = jest . spyOn ( readinessManager . splits , 'emit' ) ;
168171
169172 let splitFiltersValidation = { queryString : null , groupedFilters : { bySet : [ ] , byName : [ ] , byPrefix : [ ] } , validFilters : [ ] } ;
170173
171- let splitChangesUpdater = splitChangesUpdaterFactory ( loggerMock , splitChangesFetcher , splitsCache , segmentsCache , splitFiltersValidation , readinessManager . splits , 1000 , 1 ) ;
174+ let splitChangesUpdater = splitChangesUpdaterFactory ( loggerMock , splitChangesFetcher , storage , splitFiltersValidation , readinessManager . splits , 1000 , 1 ) ;
172175
173176 afterEach ( ( ) => {
174177 jest . clearAllMocks ( ) ;
@@ -206,7 +209,7 @@ describe('splitChangesUpdater', () => {
206209 }
207210 } ) ;
208211
209- test ( 'flag sets splits-arrived emition ' , async ( ) => {
212+ test ( 'flag sets splits-arrived emission ' , async ( ) => {
210213 const payload = splitNotifications [ 3 ] . decoded as Pick < ISplit , 'name' | 'changeNumber' | 'killed' | 'defaultTreatment' | 'trafficTypeName' | 'conditions' | 'status' | 'seed' | 'trafficAllocation' | 'trafficAllocationSeed' | 'configurations' > ;
211214 const setMocks = [
212215 { sets : [ ] , shouldEmit : false } , /* should not emit if flag does not have any set */
@@ -217,7 +220,7 @@ describe('splitChangesUpdater', () => {
217220 { sets : [ 'set_a' ] , shouldEmit : true } , /* should emit if flag is back in configured sets */
218221 ] ;
219222
220- splitChangesUpdater = splitChangesUpdaterFactory ( loggerMock , splitChangesFetcher , new SplitsCacheInMemory ( ) , segmentsCache , splitFiltersValidation , readinessManager . splits , 1000 , 1 , true ) ;
223+ splitChangesUpdater = splitChangesUpdaterFactory ( loggerMock , splitChangesFetcher , storage , splitFiltersValidation , readinessManager . splits , 1000 , 1 , true ) ;
221224
222225 let index = 0 ;
223226 let calls = 0 ;
@@ -230,7 +233,8 @@ describe('splitChangesUpdater', () => {
230233
231234 // @ts -ignore
232235 splitFiltersValidation = { queryString : null , groupedFilters : { bySet : [ 'set_a' ] , byName : [ ] , byPrefix : [ ] } , validFilters : [ ] } ;
233- splitChangesUpdater = splitChangesUpdaterFactory ( loggerMock , splitChangesFetcher , new SplitsCacheInMemory ( ) , segmentsCache , splitFiltersValidation , readinessManager . splits , 1000 , 1 , true ) ;
236+ storage . splits . clear ( ) ;
237+ splitChangesUpdater = splitChangesUpdaterFactory ( loggerMock , splitChangesFetcher , storage , splitFiltersValidation , readinessManager . splits , 1000 , 1 , true ) ;
234238 splitsEmitSpy . mockReset ( ) ;
235239 index = 0 ;
236240 for ( const setMock of setMocks ) {
0 commit comments