LibreChat/packages/data-provider/specs/api-endpoints.spec.ts

/**
 * @jest-environment jsdom
 */
import { buildLoginRedirectUrl } from '../src/api-endpoints';

describe('buildLoginRedirectUrl', () => {
  let savedLocation: Location;

  beforeEach(() => {
    savedLocation = window.location;
    Object.defineProperty(window, 'location', {
      value: { pathname: '/c/abc123', search: '?model=gpt-4', hash: '#msg-5' },
      writable: true,
    });
  });

  afterEach(() => {
    Object.defineProperty(window, 'location', { value: savedLocation, writable: true });
  });

  it('builds a login URL from explicit args', () => {
    const result = buildLoginRedirectUrl('/c/new', '?q=hello', '');
    expect(result).toBe('/login?redirect_to=%2Fc%2Fnew%3Fq%3Dhello');
  });

  it('encodes complex paths with query and hash', () => {
    const result = buildLoginRedirectUrl('/c/new', '?q=hello&submit=true', '#section');
    expect(result).toContain('redirect_to=');
    const encoded = result.split('redirect_to=')[1];
    expect(decodeURIComponent(encoded)).toBe('/c/new?q=hello&submit=true#section');
  });

  it('falls back to window.location when no args provided', () => {
    const result = buildLoginRedirectUrl();
    const encoded = result.split('redirect_to=')[1];
    expect(decodeURIComponent(encoded)).toBe('/c/abc123?model=gpt-4#msg-5');
  });

  it('falls back to "/" when all location parts are empty', () => {
    Object.defineProperty(window, 'location', {
      value: { pathname: '', search: '', hash: '' },
      writable: true,
    });
    const result = buildLoginRedirectUrl();
    expect(result).toBe('/login?redirect_to=%2F');
  });

  it('returns plain /login when pathname is /login (prevents recursive redirect)', () => {
    const result = buildLoginRedirectUrl('/login', '?redirect_to=%2Fc%2Fnew', '');
    expect(result).toBe('/login');
  });

  it('returns plain /login when window.location is already /login', () => {
    Object.defineProperty(window, 'location', {
      value: { pathname: '/login', search: '?redirect_to=%2Fc%2Fabc', hash: '' },
      writable: true,
    });
    const result = buildLoginRedirectUrl();
    expect(result).toBe('/login');
  });

  it('returns plain /login for /login sub-paths', () => {
    const result = buildLoginRedirectUrl('/login/2fa', '', '');
    expect(result).toBe('/login');
  });

  it('returns plain /login for basename-prefixed /login (e.g. /librechat/login)', () => {
    Object.defineProperty(window, 'location', {
      value: { pathname: '/librechat/login', search: '?redirect_to=%2Fc%2Fabc', hash: '' },
      writable: true,
    });
    const result = buildLoginRedirectUrl();
    expect(result).toBe('/login');
  });

  it('returns plain /login for basename-prefixed /login sub-paths', () => {
    const result = buildLoginRedirectUrl('/librechat/login/2fa', '', '');
    expect(result).toBe('/login');
  });

  it('does NOT match paths where "login" is a substring of a segment', () => {
    const result = buildLoginRedirectUrl('/c/loginhistory', '', '');
    expect(result).toContain('redirect_to=');
    expect(decodeURIComponent(result.split('redirect_to=')[1])).toBe('/c/loginhistory');
  });
});