mirrors/bun.sh
1
0
Fork 0
mirror of https://github.com/oven-sh/bun synced 2026-02-14 12:51:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(mock): add support for rejected values in JSMockFunction (#21489)

Browse Source
This commit is contained in:
fuyou
2025-07-31 12:45:38 +08:00
committed by Jarred Sumner
parent 2d6ce0ee61
commit 5f010c65d8
2 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/bun.js/bindings/JSMockFunction.cpp
View File

@@ -148,6 +148,7 @@ public:
Call,
ReturnValue,
ReturnThis,
RejectedValue,
};
static JSMockImplementation* create(JSC::JSGlobalObject* globalObject, JSC::Structure* structure, Kind kind, JSC::JSValue heldValue, bool isOnce)
@@ -962,6 +963,11 @@ JSC_DEFINE_HOST_FUNCTION(jsMockFunctionCall, (JSGlobalObject * lexicalGlobalObje
setReturnValue(createMockResult(vm, globalObject, "return"_s, thisValue));
return JSValue::encode(thisValue);
}
case JSMockImplementation::Kind::RejectedValue: {
JSValue rejectedPromise = JSC::JSPromise::rejectedPromise(globalObject, impl->underlyingValue.get());
setReturnValue(createMockResult(vm, globalObject, "return"_s, rejectedPromise));
return JSValue::encode(rejectedPromise);
}
default: {
RELEASE_ASSERT_NOT_REACHED();
}
@@ -1235,7 +1241,7 @@ JSC_DEFINE_HOST_FUNCTION(jsMockFunctionMockRejectedValue, (JSC::JSGlobalObject *
auto scope = DECLARE_THROW_SCOPE(vm);
CHECK_IS_MOCK_FUNCTION(thisValue);
pushImpl(thisObject, globalObject, JSMockImplementation::Kind::ReturnValue, JSC::JSPromise::rejectedPromise(globalObject, callframe->argument(0)));
pushImpl(thisObject, globalObject, JSMockImplementation::Kind::RejectedValue, callframe->argument(0));
RELEASE_AND_RETURN(scope, JSValue::encode(thisObject));
}
@@ -1248,7 +1254,7 @@ JSC_DEFINE_HOST_FUNCTION(jsMockFunctionMockRejectedValueOnce, (JSC::JSGlobalObje
auto scope = DECLARE_THROW_SCOPE(vm);
CHECK_IS_MOCK_FUNCTION(thisValue);
pushImplOnce(thisObject, globalObject, JSMockImplementation::Kind::ReturnValue, JSC::JSPromise::rejectedPromise(globalObject, callframe->argument(0)));
pushImplOnce(thisObject, globalObject, JSMockImplementation::Kind::RejectedValue, callframe->argument(0));
RELEASE_AND_RETURN(scope, JSValue::encode(thisObject));
}

5
test/js/bun/test/mock-fn.test.js
View File

@@ -599,6 +599,11 @@ describe("mock()", () => {
expect(await expectRejects(fn())).toBe(44);
expect(await expectRejects(fn())).toBe(42);
});
test("mockRejectedValue doesn't throw when never called", () => {
const fn = jest.fn().mockRejectedValue(new Error("Test error"));
expect(fn).toBeDefined();
expect(typeof fn).toBe("function");
});
test("withImplementation (sync)", () => {
const fn = jest.fn(() => "1");
expect(fn()).toBe("1");