15
15
*/
16
16
package org .springframework .data .mongodb .core .aggregation ;
17
17
18
- import static org .assertj . core . api .Assertions .*;
18
+ import static org .springframework . data . mongodb . test . util .Assertions .*;
19
19
20
20
import java .util .ArrayList ;
21
21
import java .util .Arrays ;
22
22
import java .util .List ;
23
23
24
24
import org .bson .Document ;
25
25
import org .junit .Test ;
26
+
26
27
import org .springframework .data .mongodb .core .aggregation .ArrayOperators .ArrayToObject ;
27
28
28
29
/**
29
30
* Unit tests for {@link ArrayOperators}
30
- *
31
+ *
31
32
* @author Christoph Strobl
32
33
* @author Shashank Sharma
33
34
* @currentRead Royal Assassin - Robin Hobb
@@ -44,14 +45,14 @@ public class ArrayOperatorsUnitTests {
44
45
public void toArrayWithFieldReference () {
45
46
46
47
assertThat (ArrayOperators .arrayOf ("regal" ).toObject ().toDocument (Aggregation .DEFAULT_CONTEXT ))
47
- .isEqualTo (Document . parse ( "{ $arrayToObject: \" $regal\" } " ) );
48
+ .isEqualTo ("{ $arrayToObject: \" $regal\" } " );
48
49
}
49
50
50
51
@ Test // DATAMONGO-2052
51
52
public void toArrayWithExpression () {
52
53
53
54
assertThat (ArrayOperators .arrayOf (EXPRESSION ).toObject ().toDocument (Aggregation .DEFAULT_CONTEXT ))
54
- .isEqualTo (Document . parse ( "{ $arrayToObject: " + EXPRESSION_STRING + "} " ) );
55
+ .isEqualTo ("{ $arrayToObject: " + EXPRESSION_STRING + "} " );
55
56
}
56
57
57
58
@ Test // DATAMONGO-2052
@@ -62,71 +63,70 @@ public void toArrayWithArgumentList() {
62
63
source .add (Arrays .asList ("prince" , "verity" ));
63
64
64
65
assertThat (ArrayToObject .arrayToObject (source ).toDocument (Aggregation .DEFAULT_CONTEXT ))
65
- .isEqualTo (Document . parse ( "{ $arrayToObject: [ [ \" king\" , \" shrewd\" ], [ \" prince\" , \" verity\" ] ] } " ) );
66
+ .isEqualTo ("{ $arrayToObject: [ [ \" king\" , \" shrewd\" ], [ \" prince\" , \" verity\" ] ] } " );
66
67
}
67
68
68
69
@ Test // DATAMONGO-2287
69
70
public void arrayElementAtWithValueList () {
70
71
71
72
assertThat (ArrayOperators .arrayOf (VALUE_LIST ).elementAt (1 ).toDocument (Aggregation .DEFAULT_CONTEXT ))
72
- .isEqualTo (Document . parse ( "{ $arrayElemAt: [ " + VALUE_LIST_STRING + ", 1] } " ) );
73
+ .isEqualTo ("{ $arrayElemAt: [ " + VALUE_LIST_STRING + ", 1] } " );
73
74
}
74
75
75
76
@ Test // DATAMONGO-2287
76
77
public void concatWithValueList () {
77
78
78
79
assertThat (ArrayOperators .arrayOf (VALUE_LIST ).concat ("field" ).toDocument (Aggregation .DEFAULT_CONTEXT ))
79
- .isEqualTo (Document . parse ( "{ $concatArrays: [ " + VALUE_LIST_STRING + ", \" $field\" ] } " ) );
80
+ .isEqualTo ("{ $concatArrays: [ " + VALUE_LIST_STRING + ", \" $field\" ] } " );
80
81
}
81
82
82
83
@ Test // DATAMONGO-2287
83
84
public void filterWithValueList () {
84
85
85
86
assertThat (ArrayOperators .arrayOf (VALUE_LIST ).filter ().as ("var" ).by (new Document ())
86
87
.toDocument (Aggregation .DEFAULT_CONTEXT ))
87
- .isEqualTo (Document
88
- .parse ("{ $filter: { \" input\" : " + VALUE_LIST_STRING + ", \" as\" : \" var\" , \" cond\" : {} } } " ));
88
+ .isEqualTo ("{ $filter: { \" input\" : " + VALUE_LIST_STRING + ", \" as\" : \" var\" , \" cond\" : {} } } " );
89
89
}
90
90
91
91
@ Test // DATAMONGO-2287
92
92
public void lengthWithValueList () {
93
93
94
94
assertThat (ArrayOperators .arrayOf (VALUE_LIST ).length ().toDocument (Aggregation .DEFAULT_CONTEXT ))
95
- .isEqualTo (Document . parse ( "{ $size: [ " + VALUE_LIST_STRING + "] } " ) );
95
+ .isEqualTo ("{ $size: [ " + VALUE_LIST_STRING + "] } " );
96
96
}
97
97
98
98
@ Test // DATAMONGO-2287
99
99
public void sliceWithValueList () {
100
100
101
101
assertThat (ArrayOperators .arrayOf (VALUE_LIST ).slice ().itemCount (3 ).toDocument (Aggregation .DEFAULT_CONTEXT ))
102
- .isEqualTo (Document . parse ( "{ $slice: [ " + VALUE_LIST_STRING + ", 3] } " ) );
102
+ .isEqualTo ("{ $slice: [ " + VALUE_LIST_STRING + ", 3] } " );
103
103
}
104
104
105
105
@ Test // DATAMONGO-2287
106
106
public void indexOfWithValueList () {
107
107
108
108
assertThat (ArrayOperators .arrayOf (VALUE_LIST ).indexOf ("s1p" ).toDocument (Aggregation .DEFAULT_CONTEXT ))
109
- .isEqualTo (Document . parse ( "{ $indexOfArray: [ " + VALUE_LIST_STRING + ", \" s1p\" ] } " ) );
109
+ .isEqualTo ("{ $indexOfArray: [ " + VALUE_LIST_STRING + ", \" s1p\" ] } " );
110
110
}
111
111
112
112
@ Test // DATAMONGO-2287
113
113
public void reverseWithValueList () {
114
114
115
115
assertThat (ArrayOperators .arrayOf (VALUE_LIST ).reverse ().toDocument (Aggregation .DEFAULT_CONTEXT ))
116
- .isEqualTo (Document . parse ( "{ $reverseArray: [ " + VALUE_LIST_STRING + "] } " ) );
116
+ .isEqualTo ("{ $reverseArray: [ " + VALUE_LIST_STRING + "] } " );
117
117
}
118
118
119
119
@ Test // DATAMONGO-2287
120
120
public void zipWithValueList () {
121
121
122
122
assertThat (ArrayOperators .arrayOf (VALUE_LIST ).zipWith ("field" ).toDocument (Aggregation .DEFAULT_CONTEXT ))
123
- .isEqualTo (Document . parse ( "{ $zip: { \" inputs\" : [" + VALUE_LIST_STRING + ", \" $field\" ]} } " ) );
123
+ .isEqualTo ("{ $zip: { \" inputs\" : [" + VALUE_LIST_STRING + ", \" $field\" ]} } " );
124
124
}
125
125
126
126
@ Test // DATAMONGO-2287
127
127
public void inWithValueList () {
128
128
129
129
assertThat (ArrayOperators .arrayOf (VALUE_LIST ).containsValue ("$userName" ).toDocument (Aggregation .DEFAULT_CONTEXT ))
130
- .isEqualTo (Document . parse ( "{ \" $in\" : [\" $userName\" , " + VALUE_LIST_STRING + "] }" ) );
130
+ .isEqualTo ("{ \" $in\" : [\" $userName\" , " + VALUE_LIST_STRING + "] }" );
131
131
}
132
132
}
0 commit comments