Fix tests and simplify regex

SpongePowered · Dec 29, 2018 · c61cbd0 · c61cbd0
1 parent 539e04f
commit c61cbd0
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 12 deletions.
diff --git a/spongeauth/accounts/models.py b/spongeauth/accounts/models.py
@@ -23,19 +23,19 @@ def validate_username(username):
         errs.append(ValidationError(
             _('Username must be at least 3 characters long.'),
             code='username_min_length'))
-    if re.search(r'[^\w_-]', username):
+    if re.search(r'[^\w-]', username):
         errs.append(ValidationError(
             _('Username must only include numbers, letters, underscores and dashes.'),
             code='username_charset'))
-    if re.search(r'[^\w_]', username[0]):
+    if re.search(r'[\W]', username[0]):
         errs.append(ValidationError(
             _('Username must begin with a letter, number or underscore.'),
             code='username_initial_charset'))
-    if re.search(r'[^\w_]', username[-1]):
+    if re.search(r'[\W]', username[-1]):
         errs.append(ValidationError(
             _('Username must end with a letter, number or underscore.'),
             code='username_ending_charset'))
-    if re.search(r'[-_]{2,}', username):
+    if re.search(r'[^A-Za-z0-9]{2,}', username):
         errs.append(ValidationError(
             _('Username must not contain two special characters in a row.'),
             code='username_double_special'))

diff --git a/spongeauth/accounts/tests/test_admin.py b/spongeauth/accounts/tests/test_admin.py
@@ -28,7 +28,7 @@ def make_post_data(self, user, **kwargs):
         return post_data
 
     def test_does_not_validate_username_if_it_is_unchanged(self):
-        user = factories.UserFactory.create(username='ewoutvs_')
+        user = factories.UserFactory.create(username='ewoutvs__')
         post_data = self.make_post_data(user)
         form = admin.AdminUserChangeForm(post_data, instance=user)
         form.save()
@@ -42,7 +42,7 @@ def test_does_validate_username_if_it_changes(self):
 
     def test_validates_username(self):
         user = factories.UserFactory.create()
-        post_data = self.make_post_data(user, username='ewoutvs_')
+        post_data = self.make_post_data(user, username='ewoutvs__')
         form = admin.AdminUserChangeForm(post_data, instance=user)
         with pytest.raises(ValueError):
             form.save()

diff --git a/spongeauth/accounts/tests/test_username_validation.py b/spongeauth/accounts/tests/test_username_validation.py
@@ -9,15 +9,16 @@
 BAD_EXAMPLES = [
     ("lukegb", []),
     ("_lukegb", []),
+    ("_lukegb_", []),
     ("a", ['username_min_length']),
-    ("__", ['username_double_special', 'username_min_length', 'username_ending_charset']),
-    ("._", ['username_double_special', 'username_min_length', 'username_ending_charset', 'username_initial_charset']),
+    ("__", ['username_double_special', 'username_min_length']),
+    ("._", ['username_double_special', 'username_min_length', 'username_charset', 'username_initial_charset']),
     ("\N{SNOWMAN}", ['username_charset', 'username_min_length', 'username_ending_charset', 'username_initial_charset']),
-    (".png", ['username_file_suffix', 'username_initial_charset']),
-    ("lukegb.png", ['username_file_suffix']),
+    (".png", ['username_charset', 'username_initial_charset']),
+    ("lukegb.png", ['username_charset']),
     ("luke__gb", ['username_double_special']),
-    ("luke_.gb", ['username_double_special']),
-    ("lukegb_", ['username_ending_charset']),
+    ("luke_.gb", ['username_charset', 'username_double_special']),
+    ("lukegb-", ['username_ending_charset']),
     ("-lukegb", ['username_initial_charset']),
 ]