@@ -97,8 +97,8 @@ public void testScalarEqual() {
97
97
98
98
Query <TestEntity > query = box .query ().equal (simpleInt , 2007 ).build ();
99
99
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 ());
102
102
List <TestEntity > all = query .find ();
103
103
assertEquals (1 , all .size ());
104
104
assertEquals (8 , all .get (0 ).getId ());
@@ -270,9 +270,9 @@ public void testOffsetLimit() {
270
270
public void testString () {
271
271
List <TestEntity > entities = putTestEntitiesStrings ();
272
272
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 ());
274
274
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 ())
276
276
.getId ());
277
277
assertEquals (2 , box .query ().contains (simpleString , "nana" ).build ().count ());
278
278
}
@@ -425,23 +425,23 @@ public void testByteArrayEqualsAndSetParameter() {
425
425
assertEquals (1 , query .count ());
426
426
TestEntity first = query .findFirst ();
427
427
assertNotNull (first );
428
- assertTrue ( Arrays . equals ( value , first .getSimpleByteArray () ));
428
+ assertArrayEquals ( value , first .getSimpleByteArray ());
429
429
430
430
byte [] value2 = {1 , 2 , (byte ) 2001 };
431
431
query .setParameter (simpleByteArray , value2 );
432
432
433
433
assertEquals (1 , query .count ());
434
434
TestEntity first2 = query .findFirst ();
435
435
assertNotNull (first2 );
436
- assertTrue ( Arrays . equals ( value2 , first2 .getSimpleByteArray () ));
436
+ assertArrayEquals ( value2 , first2 .getSimpleByteArray ());
437
437
438
438
byte [] value3 = {1 , 2 , (byte ) 2002 };
439
439
query .setParameter ("bytes" , value3 );
440
440
441
441
assertEquals (1 , query .count ());
442
442
TestEntity first3 = query .findFirst ();
443
443
assertNotNull (first3 );
444
- assertTrue ( Arrays . equals ( value3 , first3 .getSimpleByteArray () ));
444
+ assertArrayEquals ( value3 , first3 .getSimpleByteArray ());
445
445
}
446
446
447
447
@ Test
@@ -729,12 +729,12 @@ public void testSetParameterInt() {
729
729
730
730
putTestEntitiesScalars ();
731
731
Query <TestEntity > query = box .query ().equal (simpleInt , 2007 ).parameterAlias ("foo" ).build ();
732
- assertEquals (8 , query . findUnique ( ).getId ());
732
+ assertEquals (8 , getUniqueNotNull ( query ).getId ());
733
733
query .setParameter (simpleInt , 2004 );
734
- assertEquals (5 , query . findUnique ( ).getId ());
734
+ assertEquals (5 , getUniqueNotNull ( query ).getId ());
735
735
736
736
query .setParameter ("foo" , 2002 );
737
- assertEquals (3 , query . findUnique ( ).getId ());
737
+ assertEquals (3 , getUniqueNotNull ( query ).getId ());
738
738
}
739
739
740
740
@ Test
@@ -749,7 +749,7 @@ public void testSetParameter2Ints() {
749
749
assertEquals (4 , entities .get (1 ).getId ());
750
750
751
751
query .setParameters ("foo" , 2007 , 2007 );
752
- assertEquals (8 , query . findUnique ( ).getId ());
752
+ assertEquals (8 , getUniqueNotNull ( query ).getId ());
753
753
}
754
754
755
755
@ Test
@@ -776,21 +776,21 @@ public void testSetParameter2Floats() {
776
776
assertEquals (9 , entities .get (1 ).getId ());
777
777
778
778
query .setParameters ("foo" , 400.45 , 400.55 );
779
- assertEquals (6 , query . findUnique ( ).getId ());
779
+ assertEquals (6 , getUniqueNotNull ( query ).getId ());
780
780
}
781
781
782
782
@ Test
783
783
public void testSetParameterString () {
784
784
putTestEntitiesStrings ();
785
785
Query <TestEntity > query = box .query ().equal (simpleString , "banana" ).parameterAlias ("foo" ).build ();
786
- assertEquals (1 , query . findUnique ( ).getId ());
786
+ assertEquals (1 , getUniqueNotNull ( query ).getId ());
787
787
query .setParameter (simpleString , "bar" );
788
- assertEquals (3 , query . findUnique ( ).getId ());
788
+ assertEquals (3 , getUniqueNotNull ( query ).getId ());
789
789
790
790
assertNull (query .setParameter (simpleString , "not here!" ).findUnique ());
791
791
792
792
query .setParameter ("foo" , "apple" );
793
- assertEquals (2 , query . findUnique ( ).getId ());
793
+ assertEquals (2 , getUniqueNotNull ( query ).getId ());
794
794
}
795
795
796
796
/**
@@ -845,14 +845,18 @@ public void testQueryAttempts() {
845
845
store .close ();
846
846
BoxStoreBuilder builder = new BoxStoreBuilder (createTestModel (false )).directory (boxStoreDir )
847
847
.queryAttempts (5 )
848
- .failedReadTxAttemptCallback ((result , error ) -> error .printStackTrace ());
848
+ .failedReadTxAttemptCallback ((result , error ) -> {
849
+ if (error != null ) {
850
+ error .printStackTrace ();
851
+ }
852
+ });
849
853
builder .entity (new TestEntity_ ());
850
854
851
855
store = builder .build ();
852
856
putTestEntitiesScalars ();
853
857
854
858
Query <TestEntity > query = store .boxFor (TestEntity .class ).query ().equal (simpleInt , 2007 ).build ();
855
- assertEquals (2007 , query . findFirst ( ).getSimpleInt ());
859
+ assertEquals (2007 , getFirstNotNull ( query ).getSimpleInt ());
856
860
}
857
861
858
862
@ Test
@@ -953,4 +957,10 @@ private <T> T getFirstNotNull(Query<T> query) {
953
957
assertNotNull (first );
954
958
return first ;
955
959
}
960
+
961
+ private <T > T getUniqueNotNull (Query <T > query ) {
962
+ T first = query .findUnique ();
963
+ assertNotNull (first );
964
+ return first ;
965
+ }
956
966
}
0 commit comments