mirror of
https://github.com/hanzoai/chat.git
synced 2026-09-30 20:51:17 +00:00
zeekay
32969de598
librechat-data-provider → @hanzochat/data-provider @librechat/data-schemas → @hanzochat/data-schemas @librechat/client → @hanzochat/client (@hanzochat/api already forked; external @librechat/agents fork follows) All imports, package.json names/deps, tsconfig path aliases, jest moduleNameMapper, and rollup/vite build aliases updated. The data-provider react-query subpath resolves via the exports map. Lockfiles regenerated in a follow-on once the canonical package manager is confirmed.
243 lines
7.2 KiB
JavaScript
243 lines
7.2 KiB
JavaScript
const { FileSources } = require('@hanzochat/data-provider');
|
|
const { handleExistingUser } = require('./process');
|
|
|
|
jest.mock('~/server/services/Files/strategies', () => ({
|
|
getStrategyFunctions: jest.fn(),
|
|
}));
|
|
|
|
jest.mock('~/server/services/Files/images/avatar', () => ({
|
|
resizeAvatar: jest.fn(),
|
|
}));
|
|
|
|
jest.mock('~/models', () => ({
|
|
updateUser: jest.fn(),
|
|
createUser: jest.fn(),
|
|
getUserById: jest.fn(),
|
|
}));
|
|
|
|
jest.mock('~/server/services/Config', () => ({
|
|
getAppConfig: jest.fn().mockResolvedValue({}),
|
|
}));
|
|
|
|
jest.mock('@hanzochat/api', () => ({
|
|
getBalanceConfig: jest.fn(() => ({
|
|
enabled: false,
|
|
})),
|
|
}));
|
|
|
|
const { getStrategyFunctions } = require('~/server/services/Files/strategies');
|
|
const { resizeAvatar } = require('~/server/services/Files/images/avatar');
|
|
const { updateUser } = require('~/models');
|
|
|
|
describe('handleExistingUser', () => {
|
|
beforeEach(() => {
|
|
jest.clearAllMocks();
|
|
process.env.CDN_PROVIDER = FileSources.local;
|
|
});
|
|
|
|
it('should handle null avatar without throwing error', async () => {
|
|
const oldUser = {
|
|
_id: 'user123',
|
|
avatar: null,
|
|
};
|
|
const avatarUrl = 'https://example.com/avatar.png';
|
|
|
|
await handleExistingUser(oldUser, avatarUrl);
|
|
|
|
expect(updateUser).toHaveBeenCalledWith('user123', { avatar: avatarUrl });
|
|
});
|
|
|
|
it('should handle undefined avatar without throwing error', async () => {
|
|
const oldUser = {
|
|
_id: 'user123',
|
|
// avatar is undefined
|
|
};
|
|
const avatarUrl = 'https://example.com/avatar.png';
|
|
|
|
await handleExistingUser(oldUser, avatarUrl);
|
|
|
|
expect(updateUser).toHaveBeenCalledWith('user123', { avatar: avatarUrl });
|
|
});
|
|
|
|
it('should not update avatar if it has manual=true flag', async () => {
|
|
const oldUser = {
|
|
_id: 'user123',
|
|
avatar: 'https://example.com/avatar.png?manual=true',
|
|
};
|
|
const avatarUrl = 'https://example.com/new-avatar.png';
|
|
|
|
await handleExistingUser(oldUser, avatarUrl);
|
|
|
|
expect(updateUser).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('should update avatar for local storage when avatar has no manual flag', async () => {
|
|
const oldUser = {
|
|
_id: 'user123',
|
|
avatar: 'https://example.com/old-avatar.png',
|
|
};
|
|
const avatarUrl = 'https://example.com/new-avatar.png';
|
|
|
|
await handleExistingUser(oldUser, avatarUrl);
|
|
|
|
expect(updateUser).toHaveBeenCalledWith('user123', { avatar: avatarUrl });
|
|
});
|
|
|
|
it('should process avatar for non-local storage', async () => {
|
|
process.env.CDN_PROVIDER = 's3';
|
|
|
|
const mockProcessAvatar = jest.fn().mockResolvedValue('processed-avatar-url');
|
|
getStrategyFunctions.mockReturnValue({ processAvatar: mockProcessAvatar });
|
|
resizeAvatar.mockResolvedValue(Buffer.from('resized-image'));
|
|
|
|
const oldUser = {
|
|
_id: 'user123',
|
|
avatar: null,
|
|
};
|
|
const avatarUrl = 'https://example.com/avatar.png';
|
|
|
|
await handleExistingUser(oldUser, avatarUrl);
|
|
|
|
expect(resizeAvatar).toHaveBeenCalledWith({
|
|
userId: 'user123',
|
|
input: avatarUrl,
|
|
});
|
|
expect(mockProcessAvatar).toHaveBeenCalledWith({
|
|
buffer: Buffer.from('resized-image'),
|
|
userId: 'user123',
|
|
manual: 'false',
|
|
});
|
|
expect(updateUser).toHaveBeenCalledWith('user123', { avatar: 'processed-avatar-url' });
|
|
});
|
|
|
|
it('should not update if avatar already has manual flag in non-local storage', async () => {
|
|
process.env.CDN_PROVIDER = 's3';
|
|
|
|
const oldUser = {
|
|
_id: 'user123',
|
|
avatar: 'https://cdn.example.com/avatar.png?manual=true',
|
|
};
|
|
const avatarUrl = 'https://example.com/new-avatar.png';
|
|
|
|
await handleExistingUser(oldUser, avatarUrl);
|
|
|
|
expect(resizeAvatar).not.toHaveBeenCalled();
|
|
expect(updateUser).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('should handle avatar with query parameters but without manual flag', async () => {
|
|
const oldUser = {
|
|
_id: 'user123',
|
|
avatar: 'https://example.com/avatar.png?size=large&format=webp',
|
|
};
|
|
const avatarUrl = 'https://example.com/new-avatar.png';
|
|
|
|
await handleExistingUser(oldUser, avatarUrl);
|
|
|
|
expect(updateUser).toHaveBeenCalledWith('user123', { avatar: avatarUrl });
|
|
});
|
|
|
|
it('should handle empty string avatar', async () => {
|
|
const oldUser = {
|
|
_id: 'user123',
|
|
avatar: '',
|
|
};
|
|
const avatarUrl = 'https://example.com/avatar.png';
|
|
|
|
await handleExistingUser(oldUser, avatarUrl);
|
|
|
|
expect(updateUser).toHaveBeenCalledWith('user123', { avatar: avatarUrl });
|
|
});
|
|
|
|
it('should handle avatar with manual=false parameter', async () => {
|
|
const oldUser = {
|
|
_id: 'user123',
|
|
avatar: 'https://example.com/avatar.png?manual=false',
|
|
};
|
|
const avatarUrl = 'https://example.com/new-avatar.png';
|
|
|
|
await handleExistingUser(oldUser, avatarUrl);
|
|
|
|
expect(updateUser).toHaveBeenCalledWith('user123', { avatar: avatarUrl });
|
|
});
|
|
|
|
it('should handle oldUser being null gracefully', async () => {
|
|
const avatarUrl = 'https://example.com/avatar.png';
|
|
|
|
// This should throw an error when trying to access oldUser._id
|
|
await expect(handleExistingUser(null, avatarUrl)).rejects.toThrow();
|
|
});
|
|
|
|
it('should update email when it has changed', async () => {
|
|
const oldUser = {
|
|
_id: 'user123',
|
|
email: '[email protected]',
|
|
avatar: 'https://example.com/avatar.png?manual=true',
|
|
};
|
|
const avatarUrl = 'https://example.com/avatar.png';
|
|
const newEmail = '[email protected]';
|
|
|
|
await handleExistingUser(oldUser, avatarUrl, {}, newEmail);
|
|
|
|
expect(updateUser).toHaveBeenCalledWith('user123', { email: '[email protected]' });
|
|
});
|
|
|
|
it('should update both avatar and email when both have changed', async () => {
|
|
const oldUser = {
|
|
_id: 'user123',
|
|
email: '[email protected]',
|
|
avatar: null,
|
|
};
|
|
const avatarUrl = 'https://example.com/new-avatar.png';
|
|
const newEmail = '[email protected]';
|
|
|
|
await handleExistingUser(oldUser, avatarUrl, {}, newEmail);
|
|
|
|
expect(updateUser).toHaveBeenCalledWith('user123', {
|
|
avatar: avatarUrl,
|
|
email: '[email protected]',
|
|
});
|
|
});
|
|
|
|
it('should not update email when it has not changed', async () => {
|
|
const oldUser = {
|
|
_id: 'user123',
|
|
email: '[email protected]',
|
|
avatar: 'https://example.com/avatar.png?manual=true',
|
|
};
|
|
const avatarUrl = 'https://example.com/avatar.png';
|
|
const sameEmail = '[email protected]';
|
|
|
|
await handleExistingUser(oldUser, avatarUrl, {}, sameEmail);
|
|
|
|
expect(updateUser).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('should trim email before comparison and update', async () => {
|
|
const oldUser = {
|
|
_id: 'user123',
|
|
email: '[email protected]',
|
|
avatar: 'https://example.com/avatar.png?manual=true',
|
|
};
|
|
const avatarUrl = 'https://example.com/avatar.png';
|
|
const newEmailWithSpaces = ' [email protected] ';
|
|
|
|
await handleExistingUser(oldUser, avatarUrl, {}, newEmailWithSpaces);
|
|
|
|
expect(updateUser).toHaveBeenCalledWith('user123', { email: '[email protected]' });
|
|
});
|
|
|
|
it('should not update when email parameter is not provided', async () => {
|
|
const oldUser = {
|
|
_id: 'user123',
|
|
email: '[email protected]',
|
|
avatar: 'https://example.com/avatar.png?manual=true',
|
|
};
|
|
const avatarUrl = 'https://example.com/avatar.png';
|
|
|
|
await handleExistingUser(oldUser, avatarUrl, {});
|
|
|
|
expect(updateUser).not.toHaveBeenCalled();
|
|
});
|
|
});
|