Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
73 changes: 73 additions & 0 deletions gui/src/pages/config/sections/ModelsSection.test.tsx
Original file line number Diff line number Diff line change
@@ -0,0 +1,73 @@
import { act, fireEvent, render, screen } from "@testing-library/react";
import { Provider } from "react-redux";
import { beforeEach, describe, expect, it, vi } from "vitest";
import { AuthProvider } from "../../../context/Auth";
import { IdeMessengerContext } from "../../../context/IdeMessenger";
import { MockIdeMessenger } from "../../../context/MockIdeMessenger";
import { createMockStore } from "../../../util/test/mockStore";
import { ModelsSection } from "./ModelsSection";

describe("ModelsSection", () => {
const renderComponent = async (mockMessenger?: MockIdeMessenger) => {
const { mockIdeMessenger, ...store } = createMockStore({}, mockMessenger);

const result = await act(async () =>
render(
<Provider store={store}>
<IdeMessengerContext.Provider value={mockIdeMessenger}>
<AuthProvider>
<ModelsSection />
</AuthProvider>
</IdeMessengerContext.Provider>
</Provider>,
),
);

return { ...result, mockIdeMessenger };
};

beforeEach(() => {
vi.clearAllMocks();
});

it("renders current docs links for chat, autocomplete, and edit", async () => {
await renderComponent();

const learnMoreLinks = screen.getAllByRole("link", { name: "Learn more" });
const hrefs = learnMoreLinks.map((link) => link.getAttribute("href"));

expect(hrefs).toEqual([
"https://docs.continue.dev/ide-extensions/chat/quick-start",
"https://docs.continue.dev/ide-extensions/autocomplete/quick-start",
"https://docs.continue.dev/ide-extensions/edit/quick-start",
]);
});

it("opens current setup docs URLs when setup buttons are clicked", async () => {
const { mockIdeMessenger } = await renderComponent(new MockIdeMessenger());
const postSpy = vi.spyOn(mockIdeMessenger, "post");

fireEvent.click(screen.getByRole("button", { name: "Setup Chat model" }));
fireEvent.click(
screen.getByRole("button", { name: "Setup Autocomplete model" }),
);
fireEvent.click(screen.getByRole("button", { name: "Setup Edit model" }));

expect(postSpy).toHaveBeenCalledTimes(3);
expect(postSpy).toHaveBeenNthCalledWith(
1,
"openUrl",
"https://docs.continue.dev/ide-extensions/chat/model-setup",
);
expect(postSpy).toHaveBeenNthCalledWith(
2,
"openUrl",
"https://docs.continue.dev/ide-extensions/autocomplete/model-setup",
);
expect(postSpy).toHaveBeenNthCalledWith(
3,
"openUrl",
"https://docs.continue.dev/ide-extensions/edit/model-setup",
);
});
});
12 changes: 6 additions & 6 deletions gui/src/pages/config/sections/ModelsSection.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ export function ModelsSection() {
<span>
Used in Chat, Plan, Agent mode (
<a
href="https://docs.continue.dev/features/chat/quick-start"
href="https://docs.continue.dev/ide-extensions/chat/quick-start"
target="_blank"
rel="noopener noreferrer"
className="text-inherit underline hover:brightness-125"
Expand All @@ -97,7 +97,7 @@ export function ModelsSection() {
selectedModel={config.selectedModelByRole.chat ?? undefined}
onSelect={(model) => handleRoleUpdate("chat", model)}
onConfigure={handleConfigureModel}
setupURL="https://docs.continue.dev/chat/model-setup"
setupURL="https://docs.continue.dev/ide-extensions/chat/model-setup"
/>

<Divider />
Expand All @@ -109,7 +109,7 @@ export function ModelsSection() {
<span>
Used in inline code completions as you type (
<a
href="https://docs.continue.dev/features/autocomplete/quick-start"
href="https://docs.continue.dev/ide-extensions/autocomplete/quick-start"
target="_blank"
rel="noopener noreferrer"
className="text-inherit underline hover:brightness-125"
Expand All @@ -123,7 +123,7 @@ export function ModelsSection() {
selectedModel={config.selectedModelByRole.autocomplete ?? undefined}
onSelect={(model) => handleRoleUpdate("autocomplete", model)}
onConfigure={handleConfigureModel}
setupURL="https://docs.continue.dev/autocomplete/model-setup"
setupURL="https://docs.continue.dev/ide-extensions/autocomplete/model-setup"
/>

{/* Jetbrains has a model selector inline */}
Expand All @@ -142,7 +142,7 @@ export function ModelsSection() {
<span>
Used to transform a selected section of code (
<a
href="https://docs.continue.dev/features/edit/quick-start"
href="https://docs.continue.dev/ide-extensions/edit/quick-start"
target="_blank"
rel="noopener noreferrer"
className="text-inherit underline hover:brightness-125"
Expand All @@ -156,7 +156,7 @@ export function ModelsSection() {
selectedModel={config.selectedModelByRole.edit ?? undefined}
onSelect={(model) => handleRoleUpdate("edit", model)}
onConfigure={handleConfigureModel}
setupURL="https://docs.continue.dev/edit/model-setup"
setupURL="https://docs.continue.dev/ide-extensions/edit/model-setup"
/>
</>
)}
Expand Down
Loading