15
15
*/
16
16
package org .springframework .batch .item .data ;
17
17
18
+ import java .util .ArrayList ;
19
+ import java .util .HashMap ;
20
+ import java .util .Map ;
21
+
18
22
import org .junit .Before ;
19
23
import org .junit .Test ;
20
24
import org .mockito .ArgumentCaptor ;
21
25
import org .mockito .Mock ;
22
26
import org .mockito .MockitoAnnotations ;
27
+
23
28
import org .springframework .data .domain .Sort ;
24
29
import org .springframework .data .mongodb .core .MongoOperations ;
25
30
import org .springframework .data .mongodb .core .query .Query ;
26
31
27
- import java .util .ArrayList ;
28
- import java .util .HashMap ;
29
- import java .util .Map ;
30
-
31
32
import static org .junit .Assert .assertEquals ;
32
33
import static org .junit .Assert .assertFalse ;
33
34
import static org .junit .Assert .assertNull ;
@@ -120,8 +121,8 @@ public void testBasicQueryFirstPage() {
120
121
Query query = queryContainer .getValue ();
121
122
assertEquals (50 , query .getLimit ());
122
123
assertEquals (0 , query .getSkip ());
123
- assertEquals ("{ }" , query .getQueryObject ().toString ());
124
- assertEquals ("{ \" name\" : -1}" , query .getSortObject ().toString ());
124
+ assertEquals ("{ }" , query .getQueryObject ().toJson ());
125
+ assertEquals ("{ \" name\" : -1 }" , query .getSortObject ().toJson ());
125
126
}
126
127
127
128
@ Test
@@ -137,8 +138,8 @@ public void testBasicQuerySecondPage() {
137
138
138
139
assertEquals (50 , query .getLimit ());
139
140
assertEquals (100 , query .getSkip ());
140
- assertEquals ("{ }" , query .getQueryObject ().toString ());
141
- assertEquals ("{ \" name\" : -1}" , query .getSortObject ().toString ());
141
+ assertEquals ("{ }" , query .getQueryObject ().toJson ());
142
+ assertEquals ("{ \" name\" : -1 }" , query .getSortObject ().toJson ());
142
143
assertNull (query .getFieldsObject ());
143
144
}
144
145
@@ -154,9 +155,11 @@ public void testQueryWithFields() {
154
155
Query query = queryContainer .getValue ();
155
156
assertEquals (50 , query .getLimit ());
156
157
assertEquals (0 , query .getSkip ());
157
- assertEquals ("{ }" , query .getQueryObject ().toString ());
158
- assertEquals ("{ \" name\" : -1}" , query .getSortObject ().toString ());
159
- assertEquals ("{ \" name\" : 1 , \" age\" : 1 , \" _id\" : 0}" , query .getFieldsObject ().toString ());
158
+ assertEquals ("{ }" , query .getQueryObject ().toJson ());
159
+ assertEquals ("{ \" name\" : -1 }" , query .getSortObject ().toJson ());
160
+ assertEquals (1 , query .getFieldsObject ().get ("name" ));
161
+ assertEquals (1 , query .getFieldsObject ().get ("age" ));
162
+ assertEquals (0 , query .getFieldsObject ().get ("_id" ));
160
163
}
161
164
162
165
@ Test
@@ -171,8 +174,8 @@ public void testQueryWithHint() {
171
174
Query query = queryContainer .getValue ();
172
175
assertEquals (50 , query .getLimit ());
173
176
assertEquals (0 , query .getSkip ());
174
- assertEquals ("{ }" , query .getQueryObject ().toString ());
175
- assertEquals ("{ \" name\" : -1}" , query .getSortObject ().toString ());
177
+ assertEquals ("{ }" , query .getQueryObject ().toJson ());
178
+ assertEquals ("{ \" name\" : -1 }" , query .getSortObject ().toJson ());
176
179
assertEquals ("{ $natural : 1}" , query .getHint ());
177
180
}
178
181
@@ -193,8 +196,8 @@ public void testQueryWithParameters() {
193
196
Query query = queryContainer .getValue ();
194
197
assertEquals (50 , query .getLimit ());
195
198
assertEquals (0 , query .getSkip ());
196
- assertEquals ("{ \" name\" : \" foo\" }" , query .getQueryObject ().toString ());
197
- assertEquals ("{ \" name\" : -1}" , query .getSortObject ().toString ());
199
+ assertEquals ("{ \" name\" : \" foo\" }" , query .getQueryObject ().toJson ());
200
+ assertEquals ("{ \" name\" : -1 }" , query .getSortObject ().toJson ());
198
201
}
199
202
200
203
@ SuppressWarnings ("serial" )
@@ -216,8 +219,8 @@ public void testQueryWithCollection() {
216
219
Query query = queryContainer .getValue ();
217
220
assertEquals (50 , query .getLimit ());
218
221
assertEquals (0 , query .getSkip ());
219
- assertEquals ("{ \" name\" : \" foo\" }" , query .getQueryObject ().toString ());
220
- assertEquals ("{ \" name\" : -1}" , query .getSortObject ().toString ());
222
+ assertEquals ("{ \" name\" : \" foo\" }" , query .getQueryObject ().toJson ());
223
+ assertEquals ("{ \" name\" : -1 }" , query .getSortObject ().toJson ());
221
224
assertEquals ("collection" , collectionContainer .getValue ());
222
225
}
223
226
}
0 commit comments