@@ -97,8 +97,8 @@ public void testScalarEqual() {
9797
9898 Query <TestEntity > query = box .query ().equal (simpleInt , 2007 ).build ();
9999 assertEquals (1 , query .count ());
100- assertEquals (8 , query . findFirst ( ).getId ());
101- assertEquals (8 , query . findUnique ( ).getId ());
100+ assertEquals (8 , getFirstNotNull ( query ).getId ());
101+ assertEquals (8 , getUniqueNotNull ( query ).getId ());
102102 List <TestEntity > all = query .find ();
103103 assertEquals (1 , all .size ());
104104 assertEquals (8 , all .get (0 ).getId ());
@@ -270,9 +270,9 @@ public void testOffsetLimit() {
270270 public void testString () {
271271 List <TestEntity > entities = putTestEntitiesStrings ();
272272 int count = entities .size ();
273- assertEquals (1 , box .query ().equal (simpleString , "banana" ).build (). findUnique ( ).getId ());
273+ assertEquals (1 , getUniqueNotNull ( box .query ().equal (simpleString , "banana" ).build ()).getId ());
274274 assertEquals (count - 1 , box .query ().notEqual (simpleString , "banana" ).build ().count ());
275- assertEquals (4 , box .query ().startsWith (simpleString , "ba" ).endsWith (simpleString , "shake" ).build (). findUnique ( )
275+ assertEquals (4 , getUniqueNotNull ( box .query ().startsWith (simpleString , "ba" ).endsWith (simpleString , "shake" ).build ())
276276 .getId ());
277277 assertEquals (2 , box .query ().contains (simpleString , "nana" ).build ().count ());
278278 }
@@ -425,23 +425,23 @@ public void testByteArrayEqualsAndSetParameter() {
425425 assertEquals (1 , query .count ());
426426 TestEntity first = query .findFirst ();
427427 assertNotNull (first );
428- assertTrue ( Arrays . equals ( value , first .getSimpleByteArray () ));
428+ assertArrayEquals ( value , first .getSimpleByteArray ());
429429
430430 byte [] value2 = {1 , 2 , (byte ) 2001 };
431431 query .setParameter (simpleByteArray , value2 );
432432
433433 assertEquals (1 , query .count ());
434434 TestEntity first2 = query .findFirst ();
435435 assertNotNull (first2 );
436- assertTrue ( Arrays . equals ( value2 , first2 .getSimpleByteArray () ));
436+ assertArrayEquals ( value2 , first2 .getSimpleByteArray ());
437437
438438 byte [] value3 = {1 , 2 , (byte ) 2002 };
439439 query .setParameter ("bytes" , value3 );
440440
441441 assertEquals (1 , query .count ());
442442 TestEntity first3 = query .findFirst ();
443443 assertNotNull (first3 );
444- assertTrue ( Arrays . equals ( value3 , first3 .getSimpleByteArray () ));
444+ assertArrayEquals ( value3 , first3 .getSimpleByteArray ());
445445 }
446446
447447 @ Test
@@ -729,12 +729,12 @@ public void testSetParameterInt() {
729729
730730 putTestEntitiesScalars ();
731731 Query <TestEntity > query = box .query ().equal (simpleInt , 2007 ).parameterAlias ("foo" ).build ();
732- assertEquals (8 , query . findUnique ( ).getId ());
732+ assertEquals (8 , getUniqueNotNull ( query ).getId ());
733733 query .setParameter (simpleInt , 2004 );
734- assertEquals (5 , query . findUnique ( ).getId ());
734+ assertEquals (5 , getUniqueNotNull ( query ).getId ());
735735
736736 query .setParameter ("foo" , 2002 );
737- assertEquals (3 , query . findUnique ( ).getId ());
737+ assertEquals (3 , getUniqueNotNull ( query ).getId ());
738738 }
739739
740740 @ Test
@@ -749,7 +749,7 @@ public void testSetParameter2Ints() {
749749 assertEquals (4 , entities .get (1 ).getId ());
750750
751751 query .setParameters ("foo" , 2007 , 2007 );
752- assertEquals (8 , query . findUnique ( ).getId ());
752+ assertEquals (8 , getUniqueNotNull ( query ).getId ());
753753 }
754754
755755 @ Test
@@ -776,21 +776,21 @@ public void testSetParameter2Floats() {
776776 assertEquals (9 , entities .get (1 ).getId ());
777777
778778 query .setParameters ("foo" , 400.45 , 400.55 );
779- assertEquals (6 , query . findUnique ( ).getId ());
779+ assertEquals (6 , getUniqueNotNull ( query ).getId ());
780780 }
781781
782782 @ Test
783783 public void testSetParameterString () {
784784 putTestEntitiesStrings ();
785785 Query <TestEntity > query = box .query ().equal (simpleString , "banana" ).parameterAlias ("foo" ).build ();
786- assertEquals (1 , query . findUnique ( ).getId ());
786+ assertEquals (1 , getUniqueNotNull ( query ).getId ());
787787 query .setParameter (simpleString , "bar" );
788- assertEquals (3 , query . findUnique ( ).getId ());
788+ assertEquals (3 , getUniqueNotNull ( query ).getId ());
789789
790790 assertNull (query .setParameter (simpleString , "not here!" ).findUnique ());
791791
792792 query .setParameter ("foo" , "apple" );
793- assertEquals (2 , query . findUnique ( ).getId ());
793+ assertEquals (2 , getUniqueNotNull ( query ).getId ());
794794 }
795795
796796 /**
@@ -845,14 +845,18 @@ public void testQueryAttempts() {
845845 store .close ();
846846 BoxStoreBuilder builder = new BoxStoreBuilder (createTestModel (false )).directory (boxStoreDir )
847847 .queryAttempts (5 )
848- .failedReadTxAttemptCallback ((result , error ) -> error .printStackTrace ());
848+ .failedReadTxAttemptCallback ((result , error ) -> {
849+ if (error != null ) {
850+ error .printStackTrace ();
851+ }
852+ });
849853 builder .entity (new TestEntity_ ());
850854
851855 store = builder .build ();
852856 putTestEntitiesScalars ();
853857
854858 Query <TestEntity > query = store .boxFor (TestEntity .class ).query ().equal (simpleInt , 2007 ).build ();
855- assertEquals (2007 , query . findFirst ( ).getSimpleInt ());
859+ assertEquals (2007 , getFirstNotNull ( query ).getSimpleInt ());
856860 }
857861
858862 @ Test
@@ -953,4 +957,10 @@ private <T> T getFirstNotNull(Query<T> query) {
953957 assertNotNull (first );
954958 return first ;
955959 }
960+
961+ private <T > T getUniqueNotNull (Query <T > query ) {
962+ T first = query .findUnique ();
963+ assertNotNull (first );
964+ return first ;
965+ }
956966}
0 commit comments