test: Add unit tests for new ClassName processing

Author: dbanty

openapi_python_client/utils.py

@@ -27,7 +27,8 @@ def __new__(cls, value: str, prefix: str) -> "ClassName":
         new_value = fix_reserved_words(pascal_case(sanitize(value)))
 
         if not new_value.isidentifier():
-            new_value = f"{prefix}{new_value}"
+            value = f"{prefix}{new_value}"
+            new_value = fix_reserved_words(pascal_case(sanitize(value)))
         return str.__new__(cls, new_value)
 
     def __deepcopy__(self, _: Any) -> "ClassName":

tests/test_utils.py

@@ -20,6 +20,23 @@ def test_empty_is_prefixed(self):
         assert utils.PythonIdentifier(value="", prefix="something") == "something"
 
 
+class TestClassName:
+    def test_valid_is_not_changed(self):
+        assert utils.ClassName(value="ValidClass", prefix="field") == "ValidClass"
+
+    def test_numbers_are_prefixed(self):
+        assert utils.ClassName(value="1", prefix="field") == "Field1"
+
+    def test_invalid_symbols_are_stripped(self):
+        assert utils.ClassName(value="$abc", prefix="prefix") == "Abc"
+
+    def test_keywords_are_postfixed(self):
+        assert utils.ClassName(value="none", prefix="prefix") == "None_"
+
+    def test_empty_is_prefixed(self):
+        assert utils.ClassName(value="", prefix="something") == "Something"
+
+
 @pytest.mark.parametrize(
     "before, after",
     [