fix(action-code): get parent of regex

EdJoPaTo · Sep 20, 2018 · dbb123c · dbb123c
1 parent 51443fb
commit dbb123c
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/action-code.js b/action-code.js
@@ -58,11 +58,19 @@ class ActionCode {
   }
 
   parent() {
-    const parts = this.code.split(':')
+    const isRegex = this.code instanceof RegExp
+    const content = isRegex ? this.code.source : this.code
+    const parts = content.split(':')
     // Remove current
     parts.pop()
     const parent = parts.join(':')
-    return new ActionCode(parent || 'main')
+    let newCode
+    if (parent) {
+      newCode = isRegex ? new RegExp(parent) : parent
+    } else {
+      newCode = 'main'
+    }
+    return new ActionCode(newCode)
   }
 }
 

diff --git a/action-code.test.js b/action-code.test.js
@@ -48,6 +48,11 @@ test('regex', t => {
   t.deepEqual(new ActionCode(/(.+)/).get(), /^(.+)$/)
 })
 
+test('regex parent', t => {
+  t.deepEqual(new ActionCode(/b:(.+)/).parent().get(), /^b$/)
+  t.deepEqual(new ActionCode(/b-(.+)/).parent().get(), 'main')
+})
+
 test('concat string with regex', t => {
   t.deepEqual(new ActionCode('b').concat(/(.+)/).get(), /^b:(.+)$/)
 })