fix mock: support math functions with arity greater than 1 (#640)

software-mansion · Mar 13, 2020 · 383d2b5 · 383d2b5
1 parent 9f5f224
commit 383d2b5
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/mock.js b/mock.js
@@ -91,11 +91,17 @@ const Reanimated = {
     IDENTITY: 'identity',
   },
 
-  add: (a, b) => new AnimatedValue(getValue(a) + getValue(b)),
-  sub: (a, b) => new AnimatedValue(getValue(a) - getValue(b)),
-  multiply: (a, b) => new AnimatedValue(getValue(a) * getValue(b)),
-  divide: (a, b) => new AnimatedValue(getValue(a) / getValue(b)),
-  pow: (a, b) => new AnimatedValue(getValue(a) ** getValue(b)),
+
+  add: (...vals) =>
+    new AnimatedValue(vals.map(v => getValue(v)).reduce((acc, v) => acc + v)),
+  sub: (...vals) =>
+    new AnimatedValue(vals.map(v => getValue(v)).reduce((acc, v) => acc - v)),
+  divide: (...vals) =>
+    new AnimatedValue(vals.map(v => getValue(v)).reduce((acc, v) => acc / v)),
+  multiply: (...vals) =>
+    new AnimatedValue(vals.map(v => getValue(v)).reduce((acc, v) => acc * v)),
+  pow: (...vals) =>
+    new AnimatedValue(vals.map(v => getValue(v)).reduce((acc, v) => acc ** v)),
   modulo: (a, b) => new AnimatedValue(getValue(a) % getValue(b)),
   sqrt: a => new AnimatedValue(Math.sqrt(getValue(a))),
   log: a => new AnimatedValue(Math.log(getValue(a))),