mirror of
https://github.com/openclaw/openclaw.git
synced 2026-05-18 18:14:46 +00:00
test: tighten image provider lookup assertions
This commit is contained in:
@@ -34,6 +34,15 @@ async function loadProviderRegistry() {
|
||||
return await import("./provider-registry.js");
|
||||
}
|
||||
|
||||
function requireImageProvider(id: string): ImageGenerationProviderPlugin {
|
||||
const provider = getImageGenerationProvider(id);
|
||||
expect(provider).toBeDefined();
|
||||
if (!provider) {
|
||||
throw new Error(`expected image generation provider ${id}`);
|
||||
}
|
||||
return provider;
|
||||
}
|
||||
|
||||
describe("image-generation provider registry", () => {
|
||||
beforeEach(async () => {
|
||||
resolvePluginCapabilityProvidersMock.mockReset();
|
||||
@@ -56,7 +65,7 @@ describe("image-generation provider registry", () => {
|
||||
|
||||
const provider = getImageGenerationProvider("custom-image");
|
||||
|
||||
expect(provider?.id).toBe("custom-image");
|
||||
expect(provider).toMatchObject({ id: "custom-image" });
|
||||
expect(resolvePluginCapabilityProvidersMock).toHaveBeenCalledWith({
|
||||
key: "imageGenerationProviders",
|
||||
cfg: undefined,
|
||||
@@ -72,6 +81,6 @@ describe("image-generation provider registry", () => {
|
||||
expect(listImageGenerationProviders().map((provider) => provider.id)).toEqual(["safe-image"]);
|
||||
expect(getImageGenerationProvider("__proto__")).toBeUndefined();
|
||||
expect(getImageGenerationProvider("constructor")).toBeUndefined();
|
||||
expect(getImageGenerationProvider("safe-alias")?.id).toBe("safe-image");
|
||||
expect(requireImageProvider("safe-alias")).toMatchObject({ id: "safe-image" });
|
||||
});
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user