Adding tests

Author: ldecheverz-split

client/src/test/java/io/split/engine/common/SynchronizerTest.java

@@ -4,6 +4,7 @@
 import io.split.cache.SplitCache;
 import io.split.engine.experiments.SplitFetcherImp;
 import io.split.engine.experiments.SplitSynchronizationTask;
+import io.split.engine.segments.SegmentFetcher;
 import io.split.engine.segments.SegmentSynchronizationTask;
 import org.junit.Before;
 import org.junit.Test;
@@ -15,14 +16,15 @@ public class SynchronizerTest {
     private SplitFetcherImp _splitFetcher;
     private SplitCache _splitCache;
     private Synchronizer _synchronizer;
+    private SegmentCache _segmentCache;
 
     @Before
     public void beforeMethod() {
         _refreshableSplitFetcherTask = Mockito.mock(SplitSynchronizationTask.class);
         _segmentFetcher = Mockito.mock(SegmentSynchronizationTask.class);
         _splitFetcher = Mockito.mock(SplitFetcherImp.class);
         _splitCache = Mockito.mock(SplitCache.class);
-        SegmentCache _segmentCache = Mockito.mock(SegmentCache.class);
+        _segmentCache = Mockito.mock(SegmentCache.class);
 
         _synchronizer = new SynchronizerImp(_refreshableSplitFetcherTask, _splitFetcher, _segmentFetcher, _splitCache, _segmentCache);
     }
@@ -51,4 +53,23 @@ public void stopPeriodicFetching() {
         Mockito.verify(_refreshableSplitFetcherTask, Mockito.times(1)).stop();
         Mockito.verify(_segmentFetcher, Mockito.times(1)).stop();
     }
+
+    @Test
+    public void streamingRetryOnSplit() {
+        Mockito.when(_splitCache.getChangeNumber()).thenReturn(0l).thenReturn(0l).thenReturn(1l);
+        _synchronizer.refreshSplits(1l);
+
+        Mockito.verify(_splitCache, Mockito.times(3)).getChangeNumber();
+    }
+
+    @Test
+    public void streamingRetryOnSegment() {
+        SegmentFetcher fetcher = Mockito.mock(SegmentFetcher.class);
+        Mockito.when(_segmentFetcher.getFetcher(Mockito.anyString())).thenReturn(fetcher);
+        Mockito.when(_segmentCache.getChangeNumber(Mockito.anyString())).thenReturn(0l).thenReturn(0l).thenReturn(1l);
+        _synchronizer.refreshSegment("Segment",1l);
+
+        Mockito.verify(_segmentCache, Mockito.times(3)).getChangeNumber(Mockito.anyString());
+    }
+
 }