mirror of
https://github.com/openclaw/openclaw.git
synced 2026-04-17 04:01:05 +00:00
46 lines
1.5 KiB
TypeScript
46 lines
1.5 KiB
TypeScript
type ClearableMock = {
|
|
mockClear(): unknown;
|
|
};
|
|
|
|
type ResettableMock = {
|
|
mockReset(): unknown;
|
|
};
|
|
|
|
type ResettableReturnMock = ResettableMock & {
|
|
mockReturnValue(value: unknown): unknown;
|
|
};
|
|
|
|
export type GenerationRuntimeMocks = {
|
|
createSubsystemLogger: ClearableMock;
|
|
describeFailoverError: ResettableMock;
|
|
getProvider: ResettableReturnMock;
|
|
getProviderEnvVars: ResettableReturnMock;
|
|
resolveProviderAuthEnvVarCandidates: ResettableReturnMock;
|
|
isFailoverError: ResettableReturnMock;
|
|
listProviders: ResettableReturnMock;
|
|
parseModelRef: ClearableMock;
|
|
resolveAgentModelFallbackValues: ResettableReturnMock;
|
|
resolveAgentModelPrimaryValue: ResettableReturnMock;
|
|
debug: ResettableMock;
|
|
};
|
|
|
|
export function resetGenerationRuntimeMocks(mocks: GenerationRuntimeMocks): void {
|
|
mocks.createSubsystemLogger.mockClear();
|
|
mocks.describeFailoverError.mockReset();
|
|
mocks.getProvider.mockReset();
|
|
mocks.getProviderEnvVars.mockReset();
|
|
mocks.getProviderEnvVars.mockReturnValue([]);
|
|
mocks.resolveProviderAuthEnvVarCandidates.mockReset();
|
|
mocks.resolveProviderAuthEnvVarCandidates.mockReturnValue({});
|
|
mocks.isFailoverError.mockReset();
|
|
mocks.isFailoverError.mockReturnValue(false);
|
|
mocks.listProviders.mockReset();
|
|
mocks.listProviders.mockReturnValue([]);
|
|
mocks.parseModelRef.mockClear();
|
|
mocks.resolveAgentModelFallbackValues.mockReset();
|
|
mocks.resolveAgentModelFallbackValues.mockReturnValue([]);
|
|
mocks.resolveAgentModelPrimaryValue.mockReset();
|
|
mocks.resolveAgentModelPrimaryValue.mockReturnValue(undefined);
|
|
mocks.debug.mockReset();
|
|
}
|