Skip to content

Commit f3b8001

Browse files
committed
add a test for multiple input files
1 parent 6edd2ae commit f3b8001

1 file changed

Lines changed: 16 additions & 2 deletions

File tree

hbase-hbck2/src/test/java/org/apache/hbase/TestHBCK2.java

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -188,13 +188,27 @@ public void testAssigns() throws IOException {
188188
assertEquals(org.apache.hadoop.hbase.procedure2.Procedure.NO_PROC_ID, pid);
189189
}
190190

191-
// test input files
191+
// Test input files
192192
unassigns(regions, regionStrsArray);
193193
File testFile = new File(TEST_UTIL.getDataTestDir().toString(), "inputForAssignsTest");
194194
writeStringsToAFile(testFile, regionStrsArray);
195195
String result = testRunWithArgs(new String[]{ASSIGNS, "-i", testFile.toString()});
196196
validateRegionEndState(getPidsFromResult(result), regions, true);
197-
}
197+
198+
// Test multiple input files
199+
unassigns(regions, regionStrsArray);
200+
List<String> params = new ArrayList<>();
201+
params.add(ASSIGNS);
202+
params.add("-i");
203+
for (String regionStr : regionStrsArray) {
204+
File tempTestFile = new File(TEST_UTIL.getDataTestDir().toString(),
205+
"inputForAssignsTest-" + regionStr);
206+
writeStringsToAFile(tempTestFile, new String[]{regionStr});
207+
params.add(tempTestFile.toString());
208+
}
209+
result = testRunWithArgs(params.toArray(new String[]{}));
210+
validateRegionEndState(getPidsFromResult(result), regions, true);
211+
}
198212
}
199213
}
200214

0 commit comments

Comments
 (0)