fix: pass typescript plugin to parse source (#38)

package.json

@@ -36,7 +36,7 @@
   },
   "dependencies": {
     "@actions/core": "^1.10.1",
-    "@babel/core": "^7.22.17",
+    "@babel/core": "^7.23.0",
     "@babel/parser": "^7.23.0",
     "@babel/plugin-syntax-jsx": "^7.22.5",
     "arg": "^5.0.2",

src/index.js

@@ -57,12 +57,12 @@ function findTestAttributes(source, options = {}) {
 
   try {
     ast = babel.parse(source, {
-      plugins: ['jsx'],
+      plugins: ['jsx', 'typescript'],
       sourceType: 'script',
     })
   } catch (e) {
     ast = babel.parse(source, {
-      plugins: ['jsx'],
+      plugins: ['jsx', 'typescript'],
       sourceType: 'module',
     })
   }

tests/fixtures/address.tsx

@@ -0,0 +1,7 @@
+type Address = {
+  street?: string
+}
+
+export function Address() {
+  return <div data-cy="street">Main</div>
+}

tests/jsx-basic.js

@@ -46,3 +46,15 @@ test('finds data-testid attribute', (t) => {
   const found = findTestAttributes(source)
   t.deepEqual(found, ['greeting'])
 })
+
+test('handles optional ? operator', (t) => {
+  const source = stripIndent`
+    const person = {}
+    console.log(person?.name)
+    export const Greeting = () => {
+      return <div data-testid="greeting">Hello</div>
+    }
+  `
+  const found = findTestAttributes(source)
+  t.deepEqual(found, ['greeting'])
+})

tests/tsx-file.js

@@ -0,0 +1,9 @@
+const test = require('ava')
+const path = require('path')
+const { findTestAttributesInFile } = require('../src')
+
+test('finds test ids in TSX file', (t) => {
+  const filename = path.join(__dirname, 'fixtures', 'address.tsx')
+  const found = findTestAttributesInFile(filename)
+  t.deepEqual(found, ['street'])
+})