Merge pull request #5 from greggb/custom-attributes

Allow custom attributes 🚀

README.md

@@ -28,6 +28,19 @@ $ npm install babel-plugin-remove-test-ids --save
   ]
 }
 ```
+... you can also configure it with your attributes!
+```json
+{
+  "plugins": [
+    [
+      "remove-test-ids",
+      {
+        "attributes": ["data-test", "data-custom-test-attr"]
+      }
+    ]
+  ]
+}
+```
 
 ## Motivation
 

index.js

@@ -2,7 +2,7 @@
 
 module.exports = () => ({
   visitor: {
-    JSXAttribute: path =>
-      path.node.name.name === 'data-test-id' && path.remove()
+    JSXAttribute: (path, { opts: { attributes = ['data-test-id'] }}) => 
+      attributes.includes(path.node.name.name) && path.remove()
   }
 })

test.js

@@ -1,6 +1,12 @@
 const { transform } = require('babel-core')
 
 const conf = { plugins: ['syntax-jsx', './index.js'] }
+const confWithOptions = {
+  plugins: [
+    'syntax-jsx',
+    ['./index.js', { attributes: ['data-custom-test-attr'] }]
+  ]
+}
 
 describe('babel-plugin-remove-test-ids', () => {
   it('should remove the [data-test-id] attribute if present', () => {
@@ -10,6 +16,13 @@ describe('babel-plugin-remove-test-ids', () => {
     expect(code).toEqual(expected)
   })
 
+  it('should remove a custom test attribute', () => {
+    const jsx = '<Component data-custom-test-attr="test">Hello!</Component>;'
+    const expected = '<Component>Hello!</Component>;'
+    const { code } = transform(jsx, confWithOptions)
+    expect(code).toEqual(expected)
+  })
+
   it('should do nothing if [data-test-id] attribute is not present', () => {
     const jsx = '<Component content="something">Hello again!</Component>;'
     const { code } = transform(jsx, conf)