mirror of https://github.com/open-webui/open-webui
107 lines
3.9 KiB
TypeScript
107 lines
3.9 KiB
TypeScript
// eslint-disable-next-line @typescript-eslint/triple-slash-reference
|
|
/// <reference path="../support/index.d.ts" />
|
|
|
|
// These tests run through the chat flow.
|
|
describe('Settings', () => {
|
|
// Wait for 2 seconds after all tests to fix an issue with Cypress's video recording missing the last few frames
|
|
after(() => {
|
|
// eslint-disable-next-line cypress/no-unnecessary-waiting
|
|
cy.wait(2000);
|
|
});
|
|
|
|
beforeEach(() => {
|
|
// Login as the admin user
|
|
cy.loginAdmin();
|
|
// Visit the home page
|
|
cy.visit('/');
|
|
});
|
|
|
|
context('Ollama', () => {
|
|
it('user can select a model', () => {
|
|
// Click on the model selector
|
|
cy.get('button[aria-label="Select a model"]').click();
|
|
// Select the first model
|
|
cy.get('button[aria-label="model-item"]').first().click();
|
|
});
|
|
|
|
it('user can perform text chat', () => {
|
|
// Click on the model selector
|
|
cy.get('button[aria-label="Select a model"]').click();
|
|
// Select the first model
|
|
cy.get('button[aria-label="model-item"]').first().click();
|
|
// Type a message
|
|
cy.get('#chat-textarea').type('Hi, what can you do? A single sentence only please.', {
|
|
force: true
|
|
});
|
|
// Send the message
|
|
cy.get('button[type="submit"]').click();
|
|
// User's message should be visible
|
|
cy.get('.chat-user').should('exist');
|
|
// Wait for the response
|
|
// .chat-assistant is created after the first token is received
|
|
cy.get('.chat-assistant', { timeout: 10_000 }).should('exist');
|
|
// Generation Info is created after the stop token is received
|
|
cy.get('div[aria-label="Generation Info"]', { timeout: 120_000 }).should('exist');
|
|
});
|
|
|
|
it('user can share chat', () => {
|
|
// Click on the model selector
|
|
cy.get('button[aria-label="Select a model"]').click();
|
|
// Select the first model
|
|
cy.get('button[aria-label="model-item"]').first().click();
|
|
// Type a message
|
|
cy.get('#chat-textarea').type('Hi, what can you do? A single sentence only please.', {
|
|
force: true
|
|
});
|
|
// Send the message
|
|
cy.get('button[type="submit"]').click();
|
|
// User's message should be visible
|
|
cy.get('.chat-user').should('exist');
|
|
// Wait for the response
|
|
// .chat-assistant is created after the first token is received
|
|
cy.get('.chat-assistant', { timeout: 10_000 }).should('exist');
|
|
// Generation Info is created after the stop token is received
|
|
cy.get('div[aria-label="Generation Info"]', { timeout: 120_000 }).should('exist');
|
|
// spy on requests
|
|
const spy = cy.spy();
|
|
cy.intercept('POST', '/api/v1/chats/**/share', spy);
|
|
// Open context menu
|
|
cy.get('#chat-context-menu-button').click();
|
|
// Click share button
|
|
cy.get('#chat-share-button').click();
|
|
// Check if the share dialog is visible
|
|
cy.get('#copy-and-share-chat-button').should('exist');
|
|
// Click the copy button
|
|
cy.get('#copy-and-share-chat-button').click();
|
|
cy.wrap({}, { timeout: 5_000 }).should(() => {
|
|
// Check if the share request was made
|
|
expect(spy).to.be.callCount(1);
|
|
});
|
|
});
|
|
|
|
it('user can generate image', () => {
|
|
// Click on the model selector
|
|
cy.get('button[aria-label="Select a model"]').click();
|
|
// Select the first model
|
|
cy.get('button[aria-label="model-item"]').first().click();
|
|
// Type a message
|
|
cy.get('#chat-textarea').type('Hi, what can you do? A single sentence only please.', {
|
|
force: true
|
|
});
|
|
// Send the message
|
|
cy.get('button[type="submit"]').click();
|
|
// User's message should be visible
|
|
cy.get('.chat-user').should('exist');
|
|
// Wait for the response
|
|
// .chat-assistant is created after the first token is received
|
|
cy.get('.chat-assistant', { timeout: 10_000 }).should('exist');
|
|
// Generation Info is created after the stop token is received
|
|
cy.get('div[aria-label="Generation Info"]', { timeout: 120_000 }).should('exist');
|
|
// Click on the generate image button
|
|
cy.get('[aria-label="Generate Image"]').click();
|
|
// Wait for image to be visible
|
|
cy.get('img[data-cy="image"]', { timeout: 60_000 }).should('be.visible');
|
|
});
|
|
});
|
|
});
|