add generate_setter() function

Author: MrMaydo

src/method_generator.py

@@ -22,6 +22,10 @@
     "null", "true", "false"
 }
 
+indent_lvl1 = "    "
+indent_lvl2 = indent_lvl1 * 2
+indent_lvl3 = indent_lvl1 * 3
+
 
 @dataclass
 class Field:
@@ -37,14 +41,35 @@ def generate_getter(attr: Field) -> str:
 
     getter_name = "get" + attr_name[0].upper() + attr_name[1:]
 
-    getter = f"""
-    public {attr_type} {getter_name}() {{
-        return {attr_name};
-    }}
-    """
+    getter = [
+        "",
+        f"{indent_lvl1}public {attr_type} {getter_name}() {{",
+        f"{indent_lvl2}return {attr_name};",
+        f"{indent_lvl1}}}"
+    ]
+    getter = "\n".join(getter)
+
     return getter
 
 
+def generate_setter(attr: Field) -> str:
+    attr_name = attr.name
+    attr_type = attr.type
+
+    _validate_java_identifier(attr_name)
+
+    setter_name = "set" + attr_name[0].upper() + attr_name[1:]
+
+    setter = [
+        "",
+        f"{indent_lvl1}public void {setter_name}({attr_type} {attr_name}) {{",
+        f"{indent_lvl2}this.{attr_name} = {attr_name};",
+        f"{indent_lvl1}}}"
+    ]
+    setter = "\n".join(setter)
+    return setter
+
+
 def _validate_java_identifier(name: str) -> None:
     if not name:
         raise ValueError("Field name cannot be empty")

tests/test_method_generator.py

@@ -1,6 +1,6 @@
 import pytest
 
-from src.method_generator import generate_getter, Field
+from src.method_generator import generate_getter, Field, generate_setter
 
 JAVA_KEYWORDS = {
     "abstract", "assert", "boolean", "break", "byte", "case", "catch",
@@ -26,13 +26,12 @@
 def test_generate_getter_integer():
     attr = Field(
         name="exampleAttribute",
-        type="Integer"
+        type="int"
     )
     expected = """
-    public Integer getExampleAttribute() {
+    public int getExampleAttribute() {
         return exampleAttribute;
-    }
-    """
+    }"""
     assert (generate_getter(attr) == expected)
 
 
@@ -41,8 +40,16 @@ def test_generate_getter_string():
     expected = """
     public String getSomeName() {
         return someName;
-    }
-    """
+    }"""
+    assert (generate_getter(attr) == expected)
+
+
+def test_generate_getter_custom_object():
+    attr = Field(name="customData", type="CustomObject")
+    expected = """
+    public CustomObject getCustomData() {
+        return customData;
+    }"""
     assert (generate_getter(attr) == expected)
 
 
@@ -54,3 +61,39 @@ def test_generate_getter_invalid_name():
         with pytest.raises(ValueError):
             generate_getter(attr)
 
+
+def test_generate_setter_integer():
+    attr = Field(name="exampleAttribute", type="int")
+    expected = f"""
+    public void setExampleAttribute(int exampleAttribute) {{
+        this.exampleAttribute = exampleAttribute;
+    }}"""
+    assert (generate_setter(attr) == expected)
+
+
+def test_generate_setter_string():
+    attr = Field(name="someName", type="String")
+    expected = f"""
+    public void setSomeName(String someName) {{
+        this.someName = someName;
+    }}"""
+    assert (generate_setter(attr) == expected)
+
+
+def test_generate_setter_custom_object():
+    attr = Field(name="customData", type="CustomObject")
+    expected = f"""
+    public void setCustomData(CustomObject customData) {{
+        this.customData = customData;
+    }}"""
+    assert (generate_setter(attr) == expected)
+
+
+def test_generate_setter_invalid_name():
+    EMPTY_NAMES = {None, ""}
+    invalid_names = JAVA_KEYWORDS | JAVA_BUILTIN_TYPES | JAVA_LITERALS | EMPTY_NAMES
+    for name in invalid_names:
+        attr = Field(name=name, type="String")
+        with pytest.raises(ValueError):
+            generate_setter(attr)
+