style: run prettier on src directory

This commit is contained in:
Ashwin Bhat
2024-12-20 15:45:55 -08:00
parent c2c2043d05
commit 586c497740
15 changed files with 715 additions and 615 deletions

View File

@@ -1,85 +1,91 @@
import { describe, it, expect, vi } from 'vitest'
import { render, screen, fireEvent } from '@testing-library/react'
import SamplingTab from '../../components/SamplingTab'
import { Tabs } from '@/components/ui/tabs'
import type { CreateMessageRequest } from '@modelcontextprotocol/sdk/types.js'
import { describe, it, expect, vi } from "vitest";
import { render, screen, fireEvent } from "@testing-library/react";
import SamplingTab from "../../components/SamplingTab";
import { Tabs } from "@/components/ui/tabs";
import type { CreateMessageRequest } from "@modelcontextprotocol/sdk/types.js";
describe('SamplingTab', () => {
describe("SamplingTab", () => {
const mockRequest: CreateMessageRequest = {
model: 'test-model',
role: 'user',
model: "test-model",
role: "user",
content: {
type: 'text',
text: 'Test message'
}
}
type: "text",
text: "Test message",
},
};
const mockPendingRequests = [
{ id: 1, request: mockRequest },
{ id: 2, request: { ...mockRequest, content: { type: 'text', text: 'Another test' } } }
]
{
id: 2,
request: {
...mockRequest,
content: { type: "text", text: "Another test" },
},
},
];
const defaultProps = {
pendingRequests: mockPendingRequests,
onApprove: vi.fn(),
onReject: vi.fn()
}
onReject: vi.fn(),
};
const renderWithTabs = (component: React.ReactElement) => {
return render(
<Tabs defaultValue="sampling">
{component}
</Tabs>
)
}
return render(<Tabs defaultValue="sampling">{component}</Tabs>);
};
it('renders empty state when no requests', () => {
renderWithTabs(<SamplingTab {...defaultProps} pendingRequests={[]} />)
expect(screen.getByText('No pending requests')).toBeInTheDocument()
})
it("renders empty state when no requests", () => {
renderWithTabs(<SamplingTab {...defaultProps} pendingRequests={[]} />);
expect(screen.getByText("No pending requests")).toBeInTheDocument();
});
it('renders list of pending requests', () => {
renderWithTabs(<SamplingTab {...defaultProps} />)
expect(screen.getByText(/Test message/)).toBeInTheDocument()
expect(screen.getByText(/Another test/)).toBeInTheDocument()
})
it("renders list of pending requests", () => {
renderWithTabs(<SamplingTab {...defaultProps} />);
expect(screen.getByText(/Test message/)).toBeInTheDocument();
expect(screen.getByText(/Another test/)).toBeInTheDocument();
});
it('shows request details in JSON format', () => {
renderWithTabs(<SamplingTab {...defaultProps} />)
const requestJson = screen.getAllByText((content) => content.includes('"model": "test-model"'))
expect(requestJson).toHaveLength(2)
})
it("shows request details in JSON format", () => {
renderWithTabs(<SamplingTab {...defaultProps} />);
const requestJson = screen.getAllByText((content) =>
content.includes('"model": "test-model"'),
);
expect(requestJson).toHaveLength(2);
});
it("calls onApprove with stub response when Approve is clicked", () => {
const onApprove = vi.fn();
renderWithTabs(<SamplingTab {...defaultProps} onApprove={onApprove} />);
const approveButtons = screen.getAllByText("Approve");
fireEvent.click(approveButtons[0]);
it('calls onApprove with stub response when Approve is clicked', () => {
const onApprove = vi.fn()
renderWithTabs(<SamplingTab {...defaultProps} onApprove={onApprove} />)
const approveButtons = screen.getAllByText('Approve')
fireEvent.click(approveButtons[0])
expect(onApprove).toHaveBeenCalledWith(1, {
model: 'stub-model',
stopReason: 'endTurn',
role: 'assistant',
model: "stub-model",
stopReason: "endTurn",
role: "assistant",
content: {
type: 'text',
text: 'This is a stub response.'
}
})
})
type: "text",
text: "This is a stub response.",
},
});
});
it('calls onReject when Reject is clicked', () => {
const onReject = vi.fn()
renderWithTabs(<SamplingTab {...defaultProps} onReject={onReject} />)
const rejectButtons = screen.getAllByText('Reject')
fireEvent.click(rejectButtons[0])
expect(onReject).toHaveBeenCalledWith(1)
})
it("calls onReject when Reject is clicked", () => {
const onReject = vi.fn();
renderWithTabs(<SamplingTab {...defaultProps} onReject={onReject} />);
it('shows informational alert about sampling requests', () => {
renderWithTabs(<SamplingTab {...defaultProps} />)
expect(screen.getByText(/When the server requests LLM sampling/)).toBeInTheDocument()
})
})
const rejectButtons = screen.getAllByText("Reject");
fireEvent.click(rejectButtons[0]);
expect(onReject).toHaveBeenCalledWith(1);
});
it("shows informational alert about sampling requests", () => {
renderWithTabs(<SamplingTab {...defaultProps} />);
expect(
screen.getByText(/When the server requests LLM sampling/),
).toBeInTheDocument();
});
});