Strip brackets and extra spaces

ethereum · Feb 13, 2023 · 2c3fb22 · 2c3fb22
1 parent d83c312
commit 2c3fb22
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/staking_deposit/utils/validation.py b/staking_deposit/utils/validation.py
@@ -238,6 +238,8 @@ def validate_bls_withdrawal_credentials(bls_withdrawal_credentials: str) -> byte
 
 
 def normalize_input_list(input: str) -> Sequence[str]:
+    input = input.strip('[({})]')
+    input = re.sub(' +', ' ', input)
     return re.split(r'; |, | |,|;', input)
 
 

diff --git a/tests/test_utils/test_validation.py b/tests/test_utils/test_validation.py
@@ -5,6 +5,7 @@
 
 from staking_deposit.exceptions import ValidationError
 from staking_deposit.utils.validation import (
+    normalize_input_list,
     validate_int_range,
     validate_password_strength,
 )
@@ -43,3 +44,19 @@ def test_validate_int_range(num: Any, low: int, high: int, valid: bool) -> None:
     else:
         with pytest.raises(ValidationError):
             validate_int_range(num, low, high)
+
+
+@pytest.mark.parametrize(
+    'input, result',
+    [
+        ('1', ['1']),
+        ('1,2,3', ['1', '2', '3']),
+        ('[1,2,3]', ['1', '2', '3']),
+        ('(1,2,3)', ['1', '2', '3']),
+        ('{1,2,3}', ['1', '2', '3']),
+        ('1 2 3', ['1', '2', '3']),
+        ('1  2  3', ['1', '2', '3']),
+    ]
+)
+def test_normalize_input_list(input, result):
+    assert normalize_input_list(input) == result