@@ -90,8 +90,8 @@ public void testBasicRetrieval() throws ExecutionException {
9090
9191 FlagStore flagStore = userManager .getCurrentUserFlagStore ();
9292 assertEquals (2 , flagStore .getAllFlags ().size ());
93- assertEquals (true , flagStore .getFlag ("boolFlag1" ).getValue ().getAsBoolean ());
94- assertEquals (expectedStringFlagValue , flagStore .getFlag ("stringFlag1" ).getValue ().getAsString ());
93+ assertEquals (true , flagStore .getFlag ("boolFlag1" ).getValue ().booleanValue ());
94+ assertEquals (expectedStringFlagValue , flagStore .getFlag ("stringFlag1" ).getValue ().stringValue ());
9595 }
9696
9797 @ Test
@@ -182,14 +182,14 @@ public void testDeleteFlag() throws ExecutionException {
182182
183183 FlagStore flagStore = userManager .getCurrentUserFlagStore ();
184184 assertEquals (2 , flagStore .getAllFlags ().size ());
185- assertEquals (true , flagStore .getFlag ("boolFlag1" ).getValue ().getAsBoolean ());
186- assertEquals (expectedStringFlagValue , flagStore .getFlag ("stringFlag1" ).getValue ().getAsString ());
185+ assertEquals (true , flagStore .getFlag ("boolFlag1" ).getValue ().booleanValue ());
186+ assertEquals (expectedStringFlagValue , flagStore .getFlag ("stringFlag1" ).getValue ().stringValue ());
187187
188188 AwaitableCallback <Void > deleteAwait = new AwaitableCallback <>();
189189 userManager .deleteCurrentUserFlag ("{\" key\" :\" stringFlag1\" ,\" version\" :16}" , deleteAwait );
190190 deleteAwait .await ();
191191 assertNull (flagStore .getFlag ("stringFlag1" ));
192- assertEquals (true , flagStore .getFlag ("boolFlag1" ).getValue ().getAsBoolean ());
192+ assertEquals (true , flagStore .getFlag ("boolFlag1" ).getValue ().booleanValue ());
193193
194194 deleteAwait .reset ();
195195 userManager .deleteCurrentUserFlag ("{\" key\" :\" nonExistentFlag\" ,\" version\" :16,\" value\" :false}" , deleteAwait );
@@ -262,7 +262,7 @@ public void testDeleteWithVersion() throws ExecutionException {
262262 awaitableCallback .await ();
263263 awaitableCallback .reset ();
264264 FlagStore flagStore = userManager .getCurrentUserFlagStore ();
265- assertEquals ("string1" , flagStore .getFlag ("stringFlag1" ).getValue ().getAsString ());
265+ assertEquals ("string1" , flagStore .getFlag ("stringFlag1" ).getValue ().stringValue ());
266266
267267 userManager .deleteCurrentUserFlag ("{\" key\" :\" stringFlag1\" ,\" version\" :127}" , awaitableCallback );
268268 awaitableCallback .await ();
@@ -289,24 +289,24 @@ public void testPatchForAddAndReplaceFlags() throws ExecutionException {
289289 awaitableCallback .reset ();
290290
291291 FlagStore flagStore = userManager .getCurrentUserFlagStore ();
292- assertEquals (false , flagStore .getFlag ("new-flag" ).getValue ().getAsBoolean ());
292+ assertEquals (false , flagStore .getFlag ("new-flag" ).getValue ().booleanValue ());
293293
294294
295295 userManager .patchCurrentUserFlags ("{\" key\" :\" stringFlag1\" ,\" version\" :16,\" value\" :\" string2\" }" , awaitableCallback );
296296 awaitableCallback .await ();
297297 awaitableCallback .reset ();
298- assertEquals ("string2" , flagStore .getFlag ("stringFlag1" ).getValue ().getAsString ());
298+ assertEquals ("string2" , flagStore .getFlag ("stringFlag1" ).getValue ().stringValue ());
299299
300300 userManager .patchCurrentUserFlags ("{\" key\" :\" boolFlag1\" ,\" version\" :16,\" value\" :false}" , awaitableCallback );
301301 awaitableCallback .await ();
302302 awaitableCallback .reset ();
303- assertEquals (false , flagStore .getFlag ("boolFlag1" ).getValue ().getAsBoolean ());
303+ assertEquals (false , flagStore .getFlag ("boolFlag1" ).getValue ().booleanValue ());
304304
305- assertEquals (3.0f , flagStore .getFlag ("floatFlag1" ).getValue ().getAsFloat ());
305+ assertEquals (3.0f , flagStore .getFlag ("floatFlag1" ).getValue ().floatValue ());
306306
307307 userManager .patchCurrentUserFlags ("{\" key\" :\" floatFlag2\" ,\" version\" :16,\" value\" :8.0}" , awaitableCallback );
308308 awaitableCallback .await ();
309- assertEquals (8.0f , flagStore .getFlag ("floatFlag2" ).getValue ().getAsFloat ());
309+ assertEquals (8.0f , flagStore .getFlag ("floatFlag2" ).getValue ().floatValue ());
310310 }
311311
312312 @ Test
@@ -323,7 +323,7 @@ public void testPatchSucceedsForMissingVersionInPatch() throws ExecutionExceptio
323323 awaitableCallback .await ();
324324 awaitableCallback .reset ();
325325 Flag flag1 = flagStore .getFlag ("flag1" );
326- assertEquals ("value-from-patch" , flag1 .getValue ().getAsString ());
326+ assertEquals ("value-from-patch" , flag1 .getValue ().stringValue ());
327327 assertNull (flag1 .getVersion ());
328328
329329 //// case 2: value exists in shared preferences without version.
@@ -332,7 +332,7 @@ public void testPatchSucceedsForMissingVersionInPatch() throws ExecutionExceptio
332332 awaitableCallback .await ();
333333 awaitableCallback .reset ();
334334 flag1 = flagStore .getFlag ("flag1" );
335- assertEquals ("value-from-patch" , flag1 .getValue ().getAsString ());
335+ assertEquals ("value-from-patch" , flag1 .getValue ().stringValue ());
336336 assertNull (flag1 .getVersion ());
337337
338338 // version does not exist in shared preferences but exists in patch.
@@ -342,7 +342,7 @@ public void testPatchSucceedsForMissingVersionInPatch() throws ExecutionExceptio
342342 awaitableCallback .await ();
343343 awaitableCallback .reset ();
344344 flag1 = flagStore .getFlag ("flag1" );
345- assertEquals ("value-from-patch" , flag1 .getValue ().getAsString ());
345+ assertEquals ("value-from-patch" , flag1 .getValue ().stringValue ());
346346 assertEquals (558 , (int ) flag1 .getVersion ());
347347 assertEquals (3 , (int ) flag1 .getFlagVersion ());
348348 assertEquals (3 , flag1 .getVersionForEvents ());
@@ -353,7 +353,7 @@ public void testPatchSucceedsForMissingVersionInPatch() throws ExecutionExceptio
353353 awaitableCallback .await ();
354354 awaitableCallback .reset ();
355355 flag1 = flagStore .getFlag ("flag1" );
356- assertEquals ("value-from-patch" , flag1 .getValue ().getAsString ());
356+ assertEquals ("value-from-patch" , flag1 .getValue ().stringValue ());
357357 assertEquals (558 , (int ) flag1 .getVersion ());
358358 assertEquals (3 , (int ) flag1 .getFlagVersion ());
359359 assertEquals (3 , flag1 .getVersionForEvents ());
@@ -365,7 +365,7 @@ public void testPatchSucceedsForMissingVersionInPatch() throws ExecutionExceptio
365365 awaitableCallback .await ();
366366 awaitableCallback .reset ();
367367 flag1 = flagStore .getFlag ("flag1" );
368- assertEquals ("value-from-patch" , flag1 .getValue ().getAsString ());
368+ assertEquals ("value-from-patch" , flag1 .getValue ().stringValue ());
369369 assertNull (flag1 .getVersion ());
370370 assertNull (flag1 .getFlagVersion ());
371371 assertEquals (-1 , flag1 .getVersionForEvents ());
@@ -376,7 +376,7 @@ public void testPatchSucceedsForMissingVersionInPatch() throws ExecutionExceptio
376376 userManager .patchCurrentUserFlags ("{\" key\" :\" flag1\" ,\" version\" :559,\" flagVersion\" :3,\" value\" :\" value-from-patch\" ,\" variation\" :1,\" trackEvents\" :false}" , awaitableCallback );
377377 awaitableCallback .await ();
378378 flag1 = flagStore .getFlag ("flag1" );
379- assertEquals ("value-from-patch" , flag1 .getValue ().getAsString ());
379+ assertEquals ("value-from-patch" , flag1 .getValue ().stringValue ());
380380 assertEquals (559 , (int ) flag1 .getVersion ());
381381 assertEquals (3 , (int ) flag1 .getFlagVersion ());
382382 assertEquals (3 , flag1 .getVersionForEvents ());
@@ -404,15 +404,15 @@ public void testPatchWithVersion() throws ExecutionException {
404404 awaitableCallback .reset ();
405405 FlagStore flagStore = userManager .getCurrentUserFlagStore ();
406406 Flag stringFlag1 = flagStore .getFlag ("stringFlag1" );
407- assertEquals ("string1" , stringFlag1 .getValue ().getAsString ());
407+ assertEquals ("string1" , stringFlag1 .getValue ().stringValue ());
408408 assertNull (stringFlag1 .getFlagVersion ());
409409 assertEquals (125 , stringFlag1 .getVersionForEvents ());
410410
411411 userManager .patchCurrentUserFlags ("{\" key\" :\" stringFlag1\" ,\" version\" :126,\" value\" :\" string2\" }" , awaitableCallback );
412412 awaitableCallback .await ();
413413 awaitableCallback .reset ();
414414 stringFlag1 = flagStore .getFlag ("stringFlag1" );
415- assertEquals ("string2" , stringFlag1 .getValue ().getAsString ());
415+ assertEquals ("string2" , stringFlag1 .getValue ().stringValue ());
416416 assertEquals (126 , (int ) stringFlag1 .getVersion ());
417417 assertNull (stringFlag1 .getFlagVersion ());
418418 assertEquals (126 , stringFlag1 .getVersionForEvents ());
@@ -421,15 +421,15 @@ public void testPatchWithVersion() throws ExecutionException {
421421 awaitableCallback .await ();
422422 awaitableCallback .reset ();
423423 stringFlag1 = flagStore .getFlag ("stringFlag1" );
424- assertEquals ("string3" , stringFlag1 .getValue ().getAsString ());
424+ assertEquals ("string3" , stringFlag1 .getValue ().stringValue ());
425425 assertEquals (127 , (int ) stringFlag1 .getVersion ());
426426 assertEquals (3 , (int ) stringFlag1 .getFlagVersion ());
427427 assertEquals (3 , stringFlag1 .getVersionForEvents ());
428428
429429 userManager .patchCurrentUserFlags ("{\" key\" :\" stringFlag20\" ,\" version\" :1,\" value\" :\" stringValue\" }" , awaitableCallback );
430430 awaitableCallback .await ();
431431 Flag stringFlag20 = flagStore .getFlag ("stringFlag20" );
432- assertEquals ("stringValue" , stringFlag20 .getValue ().getAsString ());
432+ assertEquals ("stringValue" , stringFlag20 .getValue ().stringValue ());
433433 }
434434
435435 @ Test
@@ -511,13 +511,13 @@ public void testPutForReplaceFlags() throws ExecutionException {
511511
512512 FlagStore flagStore = userManager .getCurrentUserFlagStore ();
513513
514- assertEquals ("string2" , flagStore .getFlag ("stringFlag1" ).getValue ().getAsString ());
515- assertEquals (false , flagStore .getFlag ("boolFlag1" ).getValue ().getAsBoolean ());
514+ assertEquals ("string2" , flagStore .getFlag ("stringFlag1" ).getValue ().stringValue ());
515+ assertEquals (false , flagStore .getFlag ("boolFlag1" ).getValue ().booleanValue ());
516516
517517 // Should no exist as was deleted by PUT.
518518 assertNull (flagStore .getFlag ("floatFlag1" ));
519519
520- assertEquals (8.0f , flagStore .getFlag ("floatFlag2" ).getValue ().getAsFloat ());
520+ assertEquals (8.0f , flagStore .getFlag ("floatFlag2" ).getValue ().floatValue ());
521521 }
522522
523523 @ Test
@@ -635,7 +635,7 @@ public Void answer() {
635635
636636 private void assertFlagValue (String flagKey , String expectedValue ) {
637637 FlagStore flagStore = userManager .getCurrentUserFlagStore ();
638- assertEquals (expectedValue , flagStore .getFlag (flagKey ).getValue ().getAsString ());
638+ assertEquals (expectedValue , flagStore .getFlag (flagKey ).getValue ().stringValue ());
639639 }
640640
641641}
0 commit comments