Adding test coverage

Author: Barry Barrette

openapi_python_client/parser/properties/enum_property.py

@@ -195,8 +195,9 @@ def values_from_list(
             value = cast(Union[str, int], value)
             if isinstance(value, int):
                 if use_var_names:
-                    key = var_names[i].upper()
-                    output[key] = value
+                    key = var_names[i]
+                    sanitized_key = utils.snake_case(key).upper()
+                    output[sanitized_key] = value
                 elif value < 0:
                     output[f"VALUE_NEGATIVE_{-value}"] = value
                 else:

tests/test_parser/test_properties/test_init.py

@@ -352,7 +352,7 @@ def test_values_from_list(self):
 
         data = ["abc", "123", "a23", "1bc", 4, -3, "a Thing WIth spaces", ""]
 
-        result = EnumProperty.values_from_list(data, Class("ClassName", "module_name"))
+        result = EnumProperty.values_from_list(data, Class("ClassName", "module_name"), [])
 
         assert result == {
             "ABC": "abc",
@@ -371,7 +371,21 @@ def test_values_from_list_duplicate(self):
         data = ["abc", "123", "a23", "abc"]
 
         with pytest.raises(ValueError):
-            EnumProperty.values_from_list(data, Class("ClassName", "module_name"))
+            EnumProperty.values_from_list(data, Class("ClassName", "module_name"), [])
+
+    def test_int_enum_var_names_extension(self):
+        from openapi_python_client.parser.properties import EnumProperty
+
+        data = [-1, 1, 2]
+        var_names = ["Negative One", "One", "Two"]
+
+        result = EnumProperty.values_from_list(data, Class("ClassName", "module_name"), var_names)
+
+        assert result == {
+            "NEGATIVE_ONE": -1,
+            "ONE": 1,
+            "TWO": 2,
+        }
 
 
 class TestLiteralEnumProperty: