add equals generation

Author: MrMaydo

src/method_generator.py

@@ -106,6 +106,34 @@ def generate_setter(field: Field) -> str:
     return setter
 
 
+def generate_equals(class_name: str, fields: List[Field]) -> str:
+    equals = [
+        "",
+        f"{indent_lvl1}@Override",
+        f"{indent_lvl1}public boolean equals(Object obj) {{",
+
+        f"{indent_lvl2}if (this == obj)",
+        f"{indent_lvl3}return true;",
+
+        f"{indent_lvl2}if (!(obj instanceof {class_name}))",
+        f"{indent_lvl3}return false;",
+
+        f"{indent_lvl2}{class_name} that = ({class_name}) obj;"
+    ]
+
+    for i, field in enumerate(fields):
+        _validate_java_identifier(field.name)
+        getter_name = "get" + field.name[0].upper() + field.name[1:]
+        semicolon = ";" if i == (len(fields) - 1) else ""
+        if i == 0:
+            equals.append(f"{indent_lvl2}return Objects.equals({getter_name}(), that.{getter_name}()){semicolon}")
+        else:
+            equals.append(f"{indent_lvl2}        && Objects.equals({getter_name}(), that.{getter_name}()){semicolon}")
+    equals.append(f"{indent_lvl1}}}")
+
+    return "\n".join(equals)
+
+
 def generate_hash_code(fields: List[Field]) -> str:
     hash_code = [
         "",

tests/test_method_generator.py

@@ -160,3 +160,31 @@ def test_generate_hash_code_invalid_name():
         attr = Field(name=name, type="String")
         with pytest.raises(ValueError):
             generate_hash_code([attr])
+
+
+def test_generate_equals():
+    class_name = "MyClass"
+    attr1 = field_exampleAttribute_int
+    attr2 = field_someName_String
+    attr3 = field_customData_CustomObject
+    attributes = [attr1, attr2, attr3]
+    expected = """
+    @Override
+    public boolean equals(Object obj) {
+        if (this == obj)
+            return true;
+        if (!(obj instanceof MyClass))
+            return false;
+        MyClass that = (MyClass) obj;
+        return Objects.equals(getExampleAttribute(), that.getExampleAttribute())
+                && Objects.equals(getSomeName(), that.getSomeName())
+                && Objects.equals(getCustomData(), that.getCustomData());
+    }"""
+    assert generate_equals(class_name, attributes) == expected
+
+
+def test_generate_hash_code_invalid_name():
+    for name in illegal_names:
+        attr = Field(name=name, type="String")
+        with pytest.raises(ValueError):
+            generate_equals("MyClass", [attr])