From a41c9741bb1c82c010d848fe0797a13dab11fcad Mon Sep 17 00:00:00 2001
From: Rafi Adnan <rafi.adnan@optimizely.com>
Date: Wed, 11 Jun 2025 10:55:05 -0400
Subject: [PATCH 1/2] fix: ignore flowNode check if one does not exist

---
 src/__tests__/data/MultipleAllWithDisplayName.tsx | 7 +++----
 src/parser.ts                                     | 3 ++-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/__tests__/data/MultipleAllWithDisplayName.tsx b/src/__tests__/data/MultipleAllWithDisplayName.tsx
index 8c89283..767458c 100644
--- a/src/__tests__/data/MultipleAllWithDisplayName.tsx
+++ b/src/__tests__/data/MultipleAllWithDisplayName.tsx
@@ -3,10 +3,9 @@ import * as React from 'react';
 interface ButtonProps
   extends Omit<React.ButtonHTMLAttributes<HTMLButtonElement>, 'type'> {}
 
-export const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
-  (props, ref) => <button {...props} ref={ref} type="button" />
-);
-
+export function Button(props: ButtonProps) {
+  return <button {...props} type="button" />;
+}
 Button.displayName = 'First';
 
 export const SubmitButton = React.forwardRef<HTMLButtonElement, ButtonProps>(
diff --git a/src/parser.ts b/src/parser.ts
index 4d5ed4d..5351938 100644
--- a/src/parser.ts
+++ b/src/parser.ts
@@ -1340,7 +1340,8 @@ function getTextValueOfFunctionProperty(
         (expr.left as ts.PropertyAccessExpression).name &&
         (expr.left as ts.PropertyAccessExpression).name.escapedText ===
           propertyName &&
-        flowNodeNameEscapedText === exp.escapedName
+        (!flowNodeNameEscapedText ||
+          flowNodeNameEscapedText === exp.escapedName)
       );
     })
     .filter(statement => {

From 569fb6927cf767c13c12d1517ab52f27a3f52f31 Mon Sep 17 00:00:00 2001
From: Rafi Adnan <rafi.adnan@optimizely.com>
Date: Thu, 12 Jun 2025 09:46:02 -0400
Subject: [PATCH 2/2] add test for function expression with custom displayName

---
 .../StatelessDisplayNameFunctionExpression.tsx     | 14 ++++++++++++++
 src/__tests__/parser.ts                            | 10 ++++++++++
 2 files changed, 24 insertions(+)
 create mode 100644 src/__tests__/data/StatelessDisplayNameFunctionExpression.tsx

diff --git a/src/__tests__/data/StatelessDisplayNameFunctionExpression.tsx b/src/__tests__/data/StatelessDisplayNameFunctionExpression.tsx
new file mode 100644
index 0000000..bd89da2
--- /dev/null
+++ b/src/__tests__/data/StatelessDisplayNameFunctionExpression.tsx
@@ -0,0 +1,14 @@
+export interface StatelessFunctionExpressionProps {
+  /** myProp description */
+  myProp: string;
+}
+
+/** StatelessFunctionExpression description */
+export function StatelessFunctionExpression(
+  props: StatelessFunctionExpressionProps
+) {
+  return <div>My Property = {props.myProp}</div>;
+}
+
+StatelessFunctionExpression.displayName =
+  'StatelessDisplayNameFunctionExpression';
diff --git a/src/__tests__/parser.ts b/src/__tests__/parser.ts
index ac5afcd..11eaba5 100644
--- a/src/__tests__/parser.ts
+++ b/src/__tests__/parser.ts
@@ -1023,6 +1023,16 @@ describe('parser', () => {
       assert.equal(parsed.displayName, 'StatelessDisplayName');
     });
 
+    it('should be taken from stateless component `displayName` property (using named export and function expression)', () => {
+      const [parsed] = parse(
+        fixturePath('StatelessDisplayNameFunctionExpression')
+      );
+      assert.equal(
+        parsed.displayName,
+        'StatelessDisplayNameFunctionExpression'
+      );
+    });
+
     it('should be taken from stateful component `displayName` property (using named export)', () => {
       const [parsed] = parse(fixturePath('StatefulDisplayName'));
       assert.equal(parsed.displayName, 'StatefulDisplayName');