improve generate_hash_code function

Author: MrMaydo

src/method_generator.py

@@ -138,17 +138,25 @@ def generate_hash_code(fields: List[Field]) -> str:
     hash_code = [
         "",
         f"{indent_lvl1}@Override",
-        f"{indent_lvl1}public int hashCode() {{",
-        f"{indent_lvl2}return Objects.hash("
+        f"{indent_lvl1}public int hashCode() {{"
     ]
 
     for i, field in enumerate(fields):
         _validate_java_identifier(field.name)
         getter_name = "get" + field.name[0].upper() + field.name[1:]
         comma = "," if i < (len(fields) - 1) else ""
-        hash_code.append(f"{indent_lvl2}        {getter_name}(){comma}")
 
-    hash_code.append(f"{indent_lvl2});")
+        if len(fields) == 1:
+            hash_code.append(f"{indent_lvl2}return Objects.hash({getter_name}());")
+        elif len(fields) > 1 and i == 0:
+            hash_code.append(f"{indent_lvl2}return Objects.hash(")
+            hash_code.append(f"{indent_lvl2}        {getter_name}(){comma}")
+        elif len(fields) > 1 and i == (len(fields) - 1):
+            hash_code.append(f"{indent_lvl2}        {getter_name}(){comma}")
+            hash_code.append(f"{indent_lvl2});")
+        else:
+            hash_code.append(f"{indent_lvl2}        {getter_name}(){comma}")
+
     hash_code.append(f"{indent_lvl1}}}")
 
     return "\n".join(hash_code)

tests/test_method_generator.py

@@ -155,6 +155,16 @@ def test_generate_hash_code():
     assert generate_hash_code(attributes) == expected
 
 
+def test_generate_hash_code_one_field():
+    attr = field_exampleAttribute_int
+    expected = """
+    @Override
+    public int hashCode() {
+        return Objects.hash(getExampleAttribute());
+    }"""
+    assert generate_hash_code([attr]) == expected
+
+
 def test_generate_hash_code_invalid_name():
     for name in illegal_names:
         attr = Field(name=name, type="String")