Add tests

tests/pylint/test_enforce_type_hints.py

@@ -313,7 +313,9 @@ def test_invalid_config_flow_step(
     linter: UnittestLinter, type_hint_checker: BaseChecker
 ) -> None:
     """Ensure invalid hints are rejected for ConfigFlow step."""
-    class_node, func_node, arg_node = astroid.extract_node(
+    type_hint_checker.linter.config.ignore_missing_annotations = True
+
+    class_node, func_node, arg_node, func_node2 = astroid.extract_node(
         """
     class FlowHandler():
         pass
@@ -329,6 +331,12 @@ async def async_step_zeroconf( #@
             device_config: dict #@
         ):
             pass
+
+        async def async_step_custom( #@
+            self,
+            user_input
+        ):
+            pass
     """,
         "homeassistant.components.pylint_test.config_flow",
     )
@@ -354,6 +362,15 @@ async def async_step_zeroconf( #@
             end_line=11,
             end_col_offset=33,
         ),
+        pylint.testutils.MessageTest(
+            msg_id="hass-return-type",
+            node=func_node2,
+            args=("ConfigFlowResult", "async_step_custom"),
+            line=17,
+            col_offset=4,
+            end_line=17,
+            end_col_offset=31,
+        ),
     ):
         type_hint_checker.visit_classdef(class_node)