Add unit tests for parsing function parameters

leancodepl · Nov 9, 2023 · d2f4e42 · d2f4e42
1 parent 645fb64
commit d2f4e42
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 1 deletion.
diff --git a/src/test/dart_class_parser.test.ts b/src/test/dart_class_parser.test.ts
@@ -198,6 +198,45 @@ describe("parseLinesToClassFields", () => {
 
       expect(parseLinesToClassFields(lines)).toEqual([]);
     });
+
+    test("with fields that are functions, returns them", () => {
+      const lines = [
+        "class Button extends StatelessWidget {",
+        "  final void Function() fun1;",
+        "  final CustomClass Function(_) fun2;",
+        "  final CustomClass Function(CustomClass param1, CustomClass param2)? fun3;",
+        "  final (CustomClass, CustomClass2) Function(CustomClass param1, param2)? fun4;",
+        "",
+        "  const Button({",
+        "    super.key,",
+        "    required this.fun1,",
+        "    required this.fun2,",
+        "    this.fun3,",
+        "    this.fun4,",
+        "  });",
+        "",
+        "  @override",
+        "  Widget build(BuildContext context) {",
+        "    return const SizedBox.shrink();",
+        "  }",
+        "}",
+      ];
+
+      expect(parseLinesToClassFields(lines)).toEqual([
+        new DartClassField("fun1", "void Function()"),
+        new DartClassField("fun2", "CustomClass Function(_)"),
+        new DartClassField(
+          "fun3",
+          "CustomClass Function(CustomClass param1, CustomClass param2)",
+          true
+        ),
+        new DartClassField(
+          "fun4",
+          "(CustomClass, CustomClass2) Function(CustomClass param1, param2)",
+          true
+        ),
+      ]);
+    });
   });
   describe("when passed a StatefulWidget", () => {
     test("with fields after constructor, returns them", () => {

diff --git a/src/util/dart_class_parser.ts b/src/util/dart_class_parser.ts
@@ -53,7 +53,6 @@ function parseLinesToClassFields(input: Array<string>): Array<DartClassField> {
     .filter((line) => line !== "");
 
   return fieldListLines.map((line) => {
-    // FIXME Add unit tests for functions with multiple parameters
     const lineParts = line.trim().split(" ").slice(1);
     const name = lineParts.pop()?.replace(";", "") ?? "";
     const type = lineParts.join(" ");