Introduce lenient parsing in DataSize regarding whitespace

Prior to this commit, a DataSize input string could not be parsed if it
contained any whitespace.

With this commit, a DataSize input string can contain leading, trailing,
or 'in between' whitespace. For example, the following will be parsed
to the same DataSize value.

- "1024B"
- "1024 B"
- " 1024B "
- " 1024 B "

Closes gh-28643
This commit is contained in:
Sam Brannen
2022-07-04 18:26:00 +02:00
parent bf39492c34
commit 2bf5f7a6b3
2 changed files with 21 additions and 7 deletions

View File

@@ -174,7 +174,7 @@ public final class DataSize implements Comparable<DataSize>, Serializable {
public static DataSize parse(CharSequence text, @Nullable DataUnit defaultUnit) {
Assert.notNull(text, "Text must not be null");
try {
Matcher matcher = DataSizeUtils.PATTERN.matcher(text);
Matcher matcher = DataSizeUtils.PATTERN.matcher(StringUtils.trimAllWhitespace(text));
Assert.state(matcher.matches(), "Does not match data size pattern");
DataUnit unit = DataSizeUtils.determineDataUnit(matcher.group(2), defaultUnit);
long amount = Long.parseLong(matcher.group(1));

View File

@@ -1,5 +1,5 @@
/*
* Copyright 2002-2019 the original author or authors.
* Copyright 2002-2022 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -17,6 +17,8 @@
package org.springframework.util.unit;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.ValueSource;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
@@ -25,6 +27,7 @@ import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException
* Tests for {@link DataSize}.
*
* @author Stephane Nicoll
* @author Sam Brannen
*/
class DataSizeTests {
@@ -128,9 +131,17 @@ class DataSizeTests {
assertThat(DataSize.parse("-1", DataUnit.KILOBYTES)).isEqualTo(DataSize.ofKilobytes(-1));
}
@Test
void parseWithBytes() {
assertThat(DataSize.parse("1024B")).isEqualTo(DataSize.ofKilobytes(1));
@ParameterizedTest(name = "[{index}] text = ''{0}''")
@ValueSource(strings = {
"1024B",
"1024 B",
"1024B ",
" 1024B",
" 1024B ",
"\t1024 B\t"
})
void parseWithBytes(CharSequence text) {
assertThat(DataSize.parse(text)).isEqualTo(DataSize.ofKilobytes(1));
}
@Test
@@ -210,9 +221,12 @@ class DataSizeTests {
@Test
void parseWithUnsupportedUnit() {
assertThatIllegalArgumentException().isThrownBy(() ->
DataSize.parse("3WB"))
assertThatIllegalArgumentException()
.isThrownBy(() -> DataSize.parse("3WB"))
.withMessage("'3WB' is not a valid data size");
assertThatIllegalArgumentException()
.isThrownBy(() -> DataSize.parse("3 WB"))
.withMessage("'3 WB' is not a valid data size");
}
}