18
18
*/
19
19
package org .elasticsearch .hadoop .serialization ;
20
20
21
+ import java .util .ArrayList ;
21
22
import java .util .Arrays ;
22
23
import java .util .Collection ;
24
+ import java .util .Collections ;
23
25
import java .util .LinkedHashMap ;
24
26
import java .util .Map ;
25
27
@@ -65,56 +67,29 @@ public static Collection<Object[]> data() {
65
67
throw new IllegalStateException ("CommandTest needs new version updates." );
66
68
}
67
69
68
- return Arrays .asList (new Object [][]{
69
- {ConfigurationOptions .ES_OPERATION_INDEX , false , EsMajorVersion .V_1_X },
70
- {ConfigurationOptions .ES_OPERATION_CREATE , false , EsMajorVersion .V_1_X },
71
- {ConfigurationOptions .ES_OPERATION_UPDATE , false , EsMajorVersion .V_1_X },
72
- {ConfigurationOptions .ES_OPERATION_DELETE , false , EsMajorVersion .V_1_X },
73
- {ConfigurationOptions .ES_OPERATION_INDEX , true , EsMajorVersion .V_1_X },
74
- {ConfigurationOptions .ES_OPERATION_CREATE , true , EsMajorVersion .V_1_X },
75
- {ConfigurationOptions .ES_OPERATION_UPDATE , true , EsMajorVersion .V_1_X },
76
- {ConfigurationOptions .ES_OPERATION_DELETE , true , EsMajorVersion .V_1_X },
77
- {ConfigurationOptions .ES_OPERATION_INDEX , false , EsMajorVersion .V_2_X },
78
- {ConfigurationOptions .ES_OPERATION_CREATE , false , EsMajorVersion .V_2_X },
79
- {ConfigurationOptions .ES_OPERATION_UPDATE , false , EsMajorVersion .V_2_X },
80
- {ConfigurationOptions .ES_OPERATION_DELETE , false , EsMajorVersion .V_2_X },
81
- {ConfigurationOptions .ES_OPERATION_INDEX , true , EsMajorVersion .V_2_X },
82
- {ConfigurationOptions .ES_OPERATION_CREATE , true , EsMajorVersion .V_2_X },
83
- {ConfigurationOptions .ES_OPERATION_UPDATE , true , EsMajorVersion .V_2_X },
84
- {ConfigurationOptions .ES_OPERATION_DELETE , true , EsMajorVersion .V_2_X },
85
- {ConfigurationOptions .ES_OPERATION_INDEX , false , EsMajorVersion .V_5_X },
86
- {ConfigurationOptions .ES_OPERATION_CREATE , false , EsMajorVersion .V_5_X },
87
- {ConfigurationOptions .ES_OPERATION_UPDATE , false , EsMajorVersion .V_5_X },
88
- {ConfigurationOptions .ES_OPERATION_DELETE , false , EsMajorVersion .V_5_X },
89
- {ConfigurationOptions .ES_OPERATION_INDEX , true , EsMajorVersion .V_5_X },
90
- {ConfigurationOptions .ES_OPERATION_CREATE , true , EsMajorVersion .V_5_X },
91
- {ConfigurationOptions .ES_OPERATION_UPDATE , true , EsMajorVersion .V_5_X },
92
- {ConfigurationOptions .ES_OPERATION_DELETE , true , EsMajorVersion .V_5_X },
93
- {ConfigurationOptions .ES_OPERATION_INDEX , false , EsMajorVersion .V_6_X },
94
- {ConfigurationOptions .ES_OPERATION_CREATE , false , EsMajorVersion .V_6_X },
95
- {ConfigurationOptions .ES_OPERATION_UPDATE , false , EsMajorVersion .V_6_X },
96
- {ConfigurationOptions .ES_OPERATION_DELETE , false , EsMajorVersion .V_6_X },
97
- {ConfigurationOptions .ES_OPERATION_INDEX , true , EsMajorVersion .V_6_X },
98
- {ConfigurationOptions .ES_OPERATION_CREATE , true , EsMajorVersion .V_6_X },
99
- {ConfigurationOptions .ES_OPERATION_UPDATE , true , EsMajorVersion .V_6_X },
100
- {ConfigurationOptions .ES_OPERATION_DELETE , true , EsMajorVersion .V_6_X },
101
- {ConfigurationOptions .ES_OPERATION_INDEX , false , EsMajorVersion .V_7_X },
102
- {ConfigurationOptions .ES_OPERATION_CREATE , false , EsMajorVersion .V_7_X },
103
- {ConfigurationOptions .ES_OPERATION_UPDATE , false , EsMajorVersion .V_7_X },
104
- {ConfigurationOptions .ES_OPERATION_DELETE , false , EsMajorVersion .V_7_X },
105
- {ConfigurationOptions .ES_OPERATION_INDEX , true , EsMajorVersion .V_7_X },
106
- {ConfigurationOptions .ES_OPERATION_CREATE , true , EsMajorVersion .V_7_X },
107
- {ConfigurationOptions .ES_OPERATION_UPDATE , true , EsMajorVersion .V_7_X },
108
- {ConfigurationOptions .ES_OPERATION_DELETE , true , EsMajorVersion .V_7_X },
109
- {ConfigurationOptions .ES_OPERATION_INDEX , false , EsMajorVersion .V_8_X },
110
- {ConfigurationOptions .ES_OPERATION_CREATE , false , EsMajorVersion .V_8_X },
111
- {ConfigurationOptions .ES_OPERATION_UPDATE , false , EsMajorVersion .V_8_X },
112
- {ConfigurationOptions .ES_OPERATION_DELETE , false , EsMajorVersion .V_8_X },
113
- {ConfigurationOptions .ES_OPERATION_INDEX , true , EsMajorVersion .V_8_X },
114
- {ConfigurationOptions .ES_OPERATION_CREATE , true , EsMajorVersion .V_8_X },
115
- {ConfigurationOptions .ES_OPERATION_UPDATE , true , EsMajorVersion .V_8_X },
116
- {ConfigurationOptions .ES_OPERATION_DELETE , true , EsMajorVersion .V_8_X }
117
- });
70
+ Collection <Object []> result = new ArrayList <>();
71
+
72
+ String [] operations = new String []{ConfigurationOptions .ES_OPERATION_INDEX ,
73
+ ConfigurationOptions .ES_OPERATION_CREATE ,
74
+ ConfigurationOptions .ES_OPERATION_UPDATE ,
75
+ ConfigurationOptions .ES_OPERATION_DELETE };
76
+ boolean [] asJsons = new boolean []{false , true };
77
+ EsMajorVersion [] versions = new EsMajorVersion []{EsMajorVersion .V_1_X ,
78
+ EsMajorVersion .V_2_X ,
79
+ EsMajorVersion .V_5_X ,
80
+ EsMajorVersion .V_6_X ,
81
+ EsMajorVersion .V_7_X ,
82
+ EsMajorVersion .V_8_X };
83
+
84
+ for (EsMajorVersion version : versions ) {
85
+ for (boolean asJson : asJsons ) {
86
+ for (String operation : operations ) {
87
+ result .add (new Object []{ operation , asJson , version });
88
+ }
89
+ }
90
+ }
91
+
92
+ return result ;
118
93
}
119
94
120
95
public CommandTest (String operation , boolean jsonInput , EsMajorVersion version ) {
@@ -131,8 +106,7 @@ public void prepare() {
131
106
map .put ("n" , 1 );
132
107
map .put ("s" , "v" );
133
108
data = map ;
134
- }
135
- else {
109
+ } else {
136
110
data = "{\" n\" :1,\" s\" :\" v\" }" ;
137
111
}
138
112
}
@@ -287,17 +261,15 @@ public void testIdPattern() throws Exception {
287
261
Settings settings = settings ();
288
262
if (version .onOrAfter (EsMajorVersion .V_8_X )) {
289
263
settings .setResourceWrite ("{n}" );
290
- }
291
- else {
264
+ } else {
292
265
settings .setResourceWrite ("foo/{n}" );
293
266
}
294
267
295
268
create (settings ).write (data ).copyTo (ba );
296
269
String header ;
297
270
if (version .onOrAfter (EsMajorVersion .V_8_X )) {
298
271
header = "{\" _index\" :\" 1\" " + (isUpdateOp () ? ",\" _id\" :2" : "" ) + "}" ;
299
- }
300
- else {
272
+ } else {
301
273
header = "{\" _index\" :\" foo\" ,\" _type\" :\" 1\" " + (isUpdateOp () ? ",\" _id\" :2" : "" ) + "}" ;
302
274
}
303
275
String result = "{\" " + operation + "\" :" + header + "}" + map ();
@@ -345,7 +317,7 @@ public void testUpdateOnlyInlineScript5X() throws Exception {
345
317
create (set ).write (data ).copyTo (ba );
346
318
String result =
347
319
"{\" " + operation + "\" :{\" _id\" :2,\" _retry_on_conflict\" :3}}\n " +
348
- "{\" script\" :{\" inline\" :\" counter = 3\" ,\" lang\" :\" groovy\" }}\n " ;
320
+ "{\" script\" :{\" inline\" :\" counter = 3\" ,\" lang\" :\" groovy\" }}\n " ;
349
321
assertEquals (result , ba .toString ());
350
322
}
351
323
@@ -475,7 +447,7 @@ public void testUpdateOnlyParamInlineScript5X() throws Exception {
475
447
476
448
String result =
477
449
"{\" " + operation + "\" :{\" _id\" :1}}\n " +
478
- "{\" script\" :{\" inline\" :\" counter = param1; anothercounter = param2\" ,\" lang\" :\" groovy\" ,\" params\" :{\" param1\" :1,\" param2\" :1}}}\n " ;
450
+ "{\" script\" :{\" inline\" :\" counter = param1; anothercounter = param2\" ,\" lang\" :\" groovy\" ,\" params\" :{\" param1\" :1,\" param2\" :1}}}\n " ;
479
451
480
452
assertEquals (result , ba .toString ());
481
453
}
@@ -553,8 +525,7 @@ private Settings settings() {
553
525
set .setProperty (ConfigurationOptions .ES_INPUT_JSON , Boolean .toString (jsonInput ));
554
526
if (isDeleteOP ()) {
555
527
InitializationUtils .setValueWriterIfNotSet (set , DeleteBulkFactory .NoDataWriter .class , null );
556
- }
557
- else {
528
+ } else {
558
529
InitializationUtils .setValueWriterIfNotSet (set , JdkValueWriter .class , null );
559
530
}
560
531
InitializationUtils .setFieldExtractorIfNotSet (set , MapFieldExtractor .class , null );
@@ -564,8 +535,7 @@ private Settings settings() {
564
535
set .setProperty (ConfigurationOptions .ES_WRITE_OPERATION , operation );
565
536
if (version .onOrAfter (EsMajorVersion .V_8_X )) {
566
537
set .setResourceWrite ("foo" );
567
- }
568
- else {
538
+ } else {
569
539
set .setResourceWrite ("foo/bar" );
570
540
}
571
541
if (isUpdateOp ()) {
0 commit comments