WIP

unexpectedjs · May 28, 2022 · 7b0eb42 · 7b0eb42
1 parent a0c52b0
commit 7b0eb42
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/lib/addAdditionalPromiseMethods.js b/lib/addAdditionalPromiseMethods.js
@@ -1,6 +1,6 @@
 const Context = require('./Context');
 
-function addAdditionalPromiseMethods(promise, expect, subject) {
+function addAdditionalPromiseMethods(promise, expect, subject, args) {
   promise.and = function (...args) {
     function executeAnd() {
       if (expect.findTypeOf(args[0]).is('expect.it')) {
@@ -32,6 +32,11 @@ function addAdditionalPromiseMethods(promise, expect, subject) {
             subject
           );
         } else {
+          return expect.execute(
+            shiftedSubject,
+            methodName.replace(/[A-Z]/g, ($0) => ` ${$0.toLowerCase()}`),
+            ...args
+          );
           return camelCaseMethods[methodName](shiftedSubject)(...args);
         }
       }

diff --git a/lib/createTopLevelExpect.js b/lib/createTopLevelExpect.js
@@ -1553,7 +1553,12 @@ expectPrototype._expect = function (context, args, forwardedFlags) {
     } else {
       result = makePromise.resolve(result);
     }
-    return addAdditionalPromiseMethods(result, this, subject);
+    return addAdditionalPromiseMethods(
+      result,
+      this,
+      subject,
+      Array.prototype.slice.call(args, 0, 2)
+    );
   } catch (e) {
     if (e && e._isUnexpected) {
       let newError = e;