Engage!

1 files changed, 81 insertions, 0 deletions

diff --git a/llama.cpp/tools/server/webui/tests/stories/fixtures/storybook-mocks.ts b/llama.cpp/tools/server/webui/tests/stories/fixtures/storybook-mocks.ts
new file mode 100644
index 0000000..c40a746
--- /dev/null
+++ b/llama.cpp/tools/server/webui/tests/stories/fixtures/storybook-mocks.ts
@@ -0,0 +1,81 @@
+import { serverStore } from '$lib/stores/server.svelte';
+import { modelsStore } from '$lib/stores/models.svelte';
+
+/**
+ * Mock server properties for Storybook testing
+ * This utility allows setting mock server configurations without polluting production code
+ */
+export function mockServerProps(props: Partial<ApiLlamaCppServerProps>): void {
+        // Reset any pointer-events from previous tests (dropdown cleanup)
+        const body = document.querySelector('body');
+        if (body) body.style.pointerEvents = '';
+
+        // Directly set the props for testing purposes
+        (serverStore as unknown as { props: ApiLlamaCppServerProps }).props = {
+                model_path: props.model_path || 'test-model',
+                modalities: {
+                        vision: props.modalities?.vision ?? false,
+                        audio: props.modalities?.audio ?? false
+                },
+                ...props
+        } as ApiLlamaCppServerProps;
+
+        // Set router mode role so activeModelId can be set
+        (serverStore as unknown as { props: ApiLlamaCppServerProps }).props.role = 'ROUTER';
+
+        // Also mock modelsStore methods for modality checking
+        const vision = props.modalities?.vision ?? false;
+        const audio = props.modalities?.audio ?? false;
+
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        (modelsStore as any).modelSupportsVision = () => vision;
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        (modelsStore as any).modelSupportsAudio = () => audio;
+
+        // Mock models list with a test model so activeModelId can be resolved
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        (modelsStore as any).models = [
+                {
+                        id: 'test-model',
+                        name: 'Test Model',
+                        model: 'test-model'
+                }
+        ];
+
+        // Mock selectedModelId
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        (modelsStore as any).selectedModelId = 'test-model';
+}
+
+/**
+ * Reset server store to clean state for testing
+ */
+export function resetServerStore(): void {
+        (serverStore as unknown as { props: ApiLlamaCppServerProps }).props = {
+                model_path: '',
+                modalities: {
+                        vision: false,
+                        audio: false
+                }
+        } as ApiLlamaCppServerProps;
+        (serverStore as unknown as { error: string }).error = '';
+        (serverStore as unknown as { loading: boolean }).loading = false;
+}
+
+/**
+ * Common mock configurations for Storybook stories
+ */
+export const mockConfigs = {
+        visionOnly: {
+                modalities: { vision: true, audio: false }
+        },
+        audioOnly: {
+                modalities: { vision: false, audio: true }
+        },
+        bothModalities: {
+                modalities: { vision: true, audio: true }
+        },
+        noModalities: {
+                modalities: { vision: false, audio: false }
+        }
+} as const;