fix transformConfig

bottenderjs · Aug 18, 2018 · 6f7ddde · 6f7ddde
1 parent 8d88cc8
commit 6f7ddde
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 3 deletions.
diff --git a/src/__tests__/transformConfig.spec.js b/src/__tests__/transformConfig.spec.js
@@ -12,7 +12,9 @@ it('should transform catch-all event', () => {
     },
   };
 
-  expect(transformConfig(config, { events: ['OTHER_A', 'OTHER_B'] })).toEqual({
+  expect(
+    transformConfig(config, { events: ['SOMETHING', 'OTHER_A', 'OTHER_B'] })
+  ).toEqual({
     states: {
       a: {
         on: {
@@ -65,13 +67,27 @@ it('should work with hierarchical machines', () => {
     },
   };
 
-  expect(transformConfig(config, { events: ['OTHER_A', 'OTHER_B'] })).toEqual({
+  expect(
+    transformConfig(config, {
+      events: [
+        'SOMETHING',
+        'TIMER',
+        'POWER_OUTAGE',
+        'PED_COUNTDOWN',
+        'OTHER_A',
+        'OTHER_B',
+      ],
+    })
+  ).toEqual({
     key: 'light',
     initial: 'green',
     states: {
       a: {
         on: {
           SOMETHING: 'somthing',
+          TIMER: 'other',
+          POWER_OUTAGE: 'other',
+          PED_COUNTDOWN: 'other',
           OTHER_A: 'other',
           OTHER_B: 'other',
         },
@@ -85,13 +101,19 @@ it('should work with hierarchical machines', () => {
         states: {
           walk: {
             on: {
+              SOMETHING: 'other',
+              TIMER: 'other',
+              POWER_OUTAGE: 'other',
               PED_COUNTDOWN: 'wait',
               OTHER_A: 'other',
               OTHER_B: 'other',
             },
           },
           wait: {
             on: {
+              SOMETHING: 'other',
+              TIMER: 'other',
+              POWER_OUTAGE: 'other',
               PED_COUNTDOWN: 'stop',
               OTHER_A: 'other',
               OTHER_B: 'other',

diff --git a/src/transformConfig.js b/src/transformConfig.js
@@ -7,14 +7,14 @@ function transformStateOn(on, { events }) {
     (result, value, key) => {
       if (key === '*') {
         return {
-          ...result,
           ...events.reduce(
             (acc, curr) => ({
               ...acc,
               [curr]: value,
             }),
             {}
           ),
+          ...result,
         };
       }