|
@@ -25,6 +25,7 @@ import org.elasticsearch.test.AbstractWireSerializingTestCase;
|
|
|
import org.hamcrest.MatcherAssert;
|
|
|
|
|
|
import java.io.IOException;
|
|
|
+import java.util.function.Function;
|
|
|
|
|
|
import static org.hamcrest.Matchers.containsString;
|
|
|
import static org.hamcrest.Matchers.equalTo;
|
|
@@ -325,4 +326,37 @@ public class ByteSizeValueTests extends AbstractWireSerializingTestCase<ByteSize
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
+
|
|
|
+ public void testOfBytes() {
|
|
|
+ testOf(ByteSizeUnit.BYTES, ByteSizeValue::ofBytes);
|
|
|
+ }
|
|
|
+
|
|
|
+ public void testOfKb() {
|
|
|
+ testOf(ByteSizeUnit.KB, ByteSizeValue::ofKb);
|
|
|
+ }
|
|
|
+
|
|
|
+ public void testOfMb() {
|
|
|
+ testOf(ByteSizeUnit.MB, ByteSizeValue::ofMb);
|
|
|
+ }
|
|
|
+
|
|
|
+ public void testOfGb() {
|
|
|
+ testOf(ByteSizeUnit.GB, ByteSizeValue::ofGb);
|
|
|
+ }
|
|
|
+
|
|
|
+ public void testOfTb() {
|
|
|
+ testOf(ByteSizeUnit.TB, ByteSizeValue::ofTb);
|
|
|
+ }
|
|
|
+
|
|
|
+ public void testOfPb() {
|
|
|
+ testOf(ByteSizeUnit.PB, ByteSizeValue::ofPb);
|
|
|
+ }
|
|
|
+
|
|
|
+ private void testOf(ByteSizeUnit unit, Function<Long, ByteSizeValue> byteSizeValueFunction) {
|
|
|
+ for (int i = 0; i < NUMBER_OF_TEST_RUNS; i++) {
|
|
|
+ long size = randomIntBetween(1, 1000);
|
|
|
+ ByteSizeValue expected = new ByteSizeValue(size, unit);
|
|
|
+ ByteSizeValue actual = byteSizeValueFunction.apply(size);
|
|
|
+ assertThat(actual, equalTo(expected));
|
|
|
+ }
|
|
|
+ }
|
|
|
}
|