feat(action-code): concat ActionCodes together

EdJoPaTo · Sep 12, 2018 · 5b0b4a7 · 5b0b4a7
1 parent 867a83c
commit 5b0b4a7
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/action-code.js b/action-code.js
@@ -30,6 +30,12 @@ class ActionCode {
   }
 
   static concat(prefix, action) {
+    if (prefix instanceof ActionCode) {
+      prefix = prefix.code
+    }
+    if (action instanceof ActionCode) {
+      action = action.code
+    }
     if (prefix instanceof RegExp) {
       throw new TypeError('concat to an RegExp is currently not supported. Open an Issue for it.')
     }

diff --git a/action-code.test.js b/action-code.test.js
@@ -42,6 +42,11 @@ test('static concat', t => {
   t.is(ActionCode.concat('a:b', 'c:d').get(), 'a:b:c:d')
 })
 
+test('concat an ActionCode', t => {
+  t.is(ActionCode.concat('main', new ActionCode('a')).get(), 'a')
+  t.deepEqual(ActionCode.concat('b', new ActionCode(/(.+)/i)).get(), /^b:(.+)$/i)
+})
+
 test('regex', t => {
   t.deepEqual(new ActionCode(/(.+)/).get(), /^(.+)$/)
   t.deepEqual(new ActionCode(/(.+)/i).get(), /^(.+)$/i)