@@ -120,6 +120,34 @@ public void testGetAll() {
120120 Mockito .verify (_userStorageWrapper , Mockito .times (1 )).getMany (Mockito .anyObject ());
121121 }
122122
123+ @ Test
124+ public void testGetSplitNames () {
125+ Split split = getSplit (SPLIT_NAME );
126+ Split split2 = getSplit (SPLIT_NAME +"2" );
127+ List <Split > listResultExpected = Stream .of (split , split2 ).collect (Collectors .toList ());
128+ Set <String > keysResult = Stream .of (SPLIT_NAME , SPLIT_NAME +"2" ).collect (Collectors .toSet ());
129+ Mockito .when (_userStorageWrapper .getKeysByPrefix (Mockito .anyObject ())).
130+ thenReturn (keysResult );
131+ List <String > splitsResult = _userCustomSplitAdapterConsumer .splitNames ();
132+ Assert .assertNotNull (splitsResult );
133+ Assert .assertEquals (listResultExpected .size (), splitsResult .size ());
134+ Assert .assertEquals (SPLIT_NAME , splitsResult .get (1 ));
135+ Assert .assertEquals (SPLIT_NAME +"2" , splitsResult .get (0 ));
136+ Mockito .verify (_userStorageWrapper , Mockito .times (1 )).getKeysByPrefix (Mockito .anyString ());
137+
138+ // default prefix
139+ listResultExpected = Stream .of (split , split2 ).collect (Collectors .toList ());
140+ keysResult = Stream .of ("SPLITIO.split." + SPLIT_NAME , "SPLITIO.split." + SPLIT_NAME +"2" ).collect (Collectors .toSet ());
141+ Mockito .when (_userStorageWrapper .getKeysByPrefix (Mockito .anyObject ())).
142+ thenReturn (keysResult );
143+
144+ splitsResult = _userCustomSplitAdapterConsumer .splitNames ();
145+ Assert .assertNotNull (splitsResult );
146+ Assert .assertEquals (listResultExpected .size (), splitsResult .size ());
147+ Assert .assertEquals (SPLIT_NAME , splitsResult .get (1 ));
148+ Assert .assertEquals (SPLIT_NAME +"2" , splitsResult .get (0 ));
149+ }
150+
123151 @ Test
124152 public void testGetAllWithWrapperFailing () {
125153 Mockito .when (_userStorageWrapper .get (PrefixAdapter .buildGetAllSplit ())).
0 commit comments