730 lines
22 KiB
TypeScript
730 lines
22 KiB
TypeScript
jest.mock('@/common', () => {
|
||
const actualCommon = jest.requireActual('@/common');
|
||
return {
|
||
...actualCommon,
|
||
ToolsService: actualCommon.ToolsService,
|
||
ensureSnowflakeId: jest.fn(),
|
||
setDictDecodeCache: jest.fn(),
|
||
throwVbenError: (message: string) => {
|
||
throw new Error(message);
|
||
},
|
||
};
|
||
});
|
||
|
||
import { ConfigService } from '@nestjs/config';
|
||
import { ToolsService } from '@/common';
|
||
import { QqbotNapcatContainerService } from '@/qqbot/napcat/qqbot-napcat-container.service';
|
||
|
||
describe('QqbotNapcatContainerService', () => {
|
||
it('removes previous account containers when binding a new primary container', async () => {
|
||
const bindingRepository = {
|
||
create: jest.fn((input) => input),
|
||
createQueryBuilder: jest.fn(() => ({
|
||
andWhere: jest.fn().mockReturnThis(),
|
||
getCount: jest.fn().mockResolvedValue(0),
|
||
where: jest.fn().mockReturnThis(),
|
||
})),
|
||
find: jest.fn().mockResolvedValue([
|
||
{
|
||
accountId: 'account-1',
|
||
containerId: 'container-old',
|
||
id: 'binding-old',
|
||
isDeleted: false,
|
||
},
|
||
]),
|
||
findOne: jest.fn().mockResolvedValue(null),
|
||
save: jest.fn(),
|
||
update: jest.fn(),
|
||
};
|
||
const containerRepository = {
|
||
findOne: jest.fn().mockResolvedValue({
|
||
id: 'container-old',
|
||
isDeleted: false,
|
||
name: 'napcat-old',
|
||
}),
|
||
update: jest.fn(),
|
||
};
|
||
const service = new QqbotNapcatContainerService(
|
||
{ get: jest.fn().mockReturnValue('') } as unknown as ConfigService,
|
||
containerRepository as any,
|
||
bindingRepository as any,
|
||
new ToolsService(),
|
||
);
|
||
jest.spyOn(service as any, 'getManagedMode').mockReturnValue('');
|
||
|
||
await service.bindAccount('account-1', 'container-new');
|
||
|
||
expect(bindingRepository.save).toHaveBeenCalledWith(
|
||
expect.objectContaining({
|
||
accountId: 'account-1',
|
||
bindStatus: 'bound',
|
||
containerId: 'container-new',
|
||
isPrimary: true,
|
||
}),
|
||
);
|
||
expect(containerRepository.update).toHaveBeenCalledWith(
|
||
{ id: 'container-old' },
|
||
expect.objectContaining({
|
||
isDeleted: true,
|
||
status: 'stopped',
|
||
}),
|
||
);
|
||
expect(bindingRepository.update).toHaveBeenCalledWith(
|
||
{ id: 'binding-old' },
|
||
expect.objectContaining({
|
||
bindStatus: 'disabled',
|
||
isDeleted: true,
|
||
isPrimary: false,
|
||
}),
|
||
);
|
||
});
|
||
|
||
it('uses the latest NapCat account status line as runtime offline truth', () => {
|
||
const service = new QqbotNapcatContainerService(
|
||
{ get: jest.fn() } as any,
|
||
{} as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
|
||
expect(
|
||
service.extractLoginState(`
|
||
06-11 08:39:19 [info] Mirror | 账号状态变更为离线
|
||
06-11 08:41:19 [info] Mirror | 账号状态变更为在线
|
||
`).offlineReason,
|
||
).toBeNull();
|
||
});
|
||
|
||
it('treats isOnline false as an offline runtime state', () => {
|
||
const service = new QqbotNapcatContainerService(
|
||
{ get: jest.fn() } as any,
|
||
{} as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
|
||
const result = service.extractLoginState(`
|
||
06-11 08:39:19 [info] Mirror | {"isOnline": false}
|
||
`);
|
||
|
||
expect(result).toEqual({
|
||
offlineReason: 'NapCat 账号状态变更为离线',
|
||
state: 'offline',
|
||
});
|
||
});
|
||
|
||
it('clears stale offline error when the latest NapCat account status is online', async () => {
|
||
const containerRepository = {
|
||
update: jest.fn(),
|
||
};
|
||
const service = new QqbotNapcatContainerService(
|
||
{
|
||
get: jest.fn((key: string) => {
|
||
const values: Record<string, string> = {
|
||
QQBOT_NAPCAT_CONTAINER_MODE: 'ssh',
|
||
QQBOT_NAPCAT_SSH_TARGET: 'nas',
|
||
};
|
||
return values[key] || '';
|
||
}),
|
||
} as any,
|
||
containerRepository as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
service.runProcess = jest.fn().mockResolvedValue({
|
||
stderr: '',
|
||
stdout: `
|
||
06-11 08:39:19 [info] Mirror | 账号状态变更为离线
|
||
06-11 08:41:19 [info] Mirror | 账号状态变更为在线
|
||
`,
|
||
});
|
||
|
||
const reason = await service.detectRuntimeOffline({
|
||
id: 'container-1',
|
||
name: 'kt-qqbot-napcat-1914728559',
|
||
});
|
||
|
||
expect(reason).toBeNull();
|
||
expect(containerRepository.update).toHaveBeenCalledWith(
|
||
{ id: 'container-1' },
|
||
expect.objectContaining({
|
||
lastCheckedAt: expect.any(Date),
|
||
lastError: null,
|
||
}),
|
||
);
|
||
});
|
||
|
||
it('extracts the latest NapCat offline reason from container logs', () => {
|
||
const service = new QqbotNapcatContainerService(
|
||
{ get: jest.fn() } as any,
|
||
{} as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
|
||
expect(
|
||
service.extractLoginState(`
|
||
06-11 08:39:19 [info] Mirror | [KickedOffLine] [下线通知] 您的账号已在另一台终端登录
|
||
06-11 08:40:49 [info] Mirror | 账号状态变更为离线
|
||
`).offlineReason,
|
||
).toBe('账号状态变更为离线');
|
||
});
|
||
|
||
it('uses a short timeout for runtime offline log detection', async () => {
|
||
const containerRepository = {
|
||
update: jest.fn(),
|
||
};
|
||
const service = new QqbotNapcatContainerService(
|
||
{
|
||
get: jest.fn((key: string) => {
|
||
const values: Record<string, string> = {
|
||
QQBOT_NAPCAT_CONTAINER_MODE: 'ssh',
|
||
QQBOT_NAPCAT_RUNTIME_CHECK_TIMEOUT_MS: '5000',
|
||
QQBOT_NAPCAT_SSH_TARGET: 'nas',
|
||
};
|
||
return values[key] || '';
|
||
}),
|
||
} as any,
|
||
containerRepository as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
service.runProcess = jest.fn().mockResolvedValue({
|
||
stderr: '',
|
||
stdout: '06-11 08:39:19 [info] Mirror | 账号状态变更为离线',
|
||
});
|
||
|
||
const reason = await service.detectRuntimeOffline({
|
||
id: 'container-1',
|
||
name: 'kt-qqbot-napcat-1914728559',
|
||
});
|
||
|
||
expect(reason).toBe('账号状态变更为离线');
|
||
expect(service.runProcess).toHaveBeenCalledWith(
|
||
'ssh',
|
||
expect.any(Array),
|
||
expect.any(String),
|
||
undefined,
|
||
5000,
|
||
);
|
||
});
|
||
|
||
it('truncates runtime offline reason before writing container lastError', async () => {
|
||
const containerRepository = {
|
||
update: jest.fn(),
|
||
};
|
||
const service = new QqbotNapcatContainerService(
|
||
{
|
||
get: jest.fn((key: string) => {
|
||
const values: Record<string, string> = {
|
||
QQBOT_NAPCAT_CONTAINER_MODE: 'ssh',
|
||
QQBOT_NAPCAT_SSH_TARGET: 'nas',
|
||
};
|
||
return values[key] || '';
|
||
}),
|
||
} as any,
|
||
containerRepository as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
service.runProcess = jest.fn().mockResolvedValue({
|
||
stderr: '',
|
||
stdout: `06-11 08:39:19 [info] Mirror | [KickedOffLine] [下线通知] ${'错误'.repeat(300)}`,
|
||
});
|
||
|
||
await service.detectRuntimeOffline({
|
||
id: 'container-1',
|
||
name: 'kt-qqbot-napcat-1914728559',
|
||
});
|
||
|
||
expect(containerRepository.update).toHaveBeenCalledWith(
|
||
{ id: 'container-1' },
|
||
expect.objectContaining({
|
||
lastError: `${'错误'.repeat(248)}错...`,
|
||
}),
|
||
);
|
||
});
|
||
|
||
it('truncates runtime check errors before writing container lastError', async () => {
|
||
const containerRepository = {
|
||
update: jest.fn(),
|
||
};
|
||
const service = new QqbotNapcatContainerService(
|
||
{
|
||
get: jest.fn((key: string) => {
|
||
const values: Record<string, string> = {
|
||
QQBOT_NAPCAT_CONTAINER_MODE: 'ssh',
|
||
QQBOT_NAPCAT_SSH_TARGET: 'nas',
|
||
};
|
||
return values[key] || '';
|
||
}),
|
||
} as any,
|
||
containerRepository as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
service.runProcess = jest
|
||
.fn()
|
||
.mockRejectedValue(new Error('错误'.repeat(300)));
|
||
|
||
await service.detectRuntimeOffline({
|
||
id: 'container-1',
|
||
name: 'kt-qqbot-napcat-1914728559',
|
||
});
|
||
|
||
expect(containerRepository.update).toHaveBeenCalledWith(
|
||
{ id: 'container-1' },
|
||
expect.objectContaining({
|
||
lastError: `${'错误'.repeat(248)}错...`,
|
||
}),
|
||
);
|
||
});
|
||
|
||
it('injects ACCOUNT and NAPCAT_QUICK_PASSWORD env only when provided', () => {
|
||
const service = new QqbotNapcatContainerService(
|
||
{ get: jest.fn().mockReturnValue('') } as any,
|
||
{} as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
|
||
const baseInput = {
|
||
dataDir: '/vol1/docker/kt-qqbot/napcat-instances/kt-qqbot-napcat-test',
|
||
image: 'mlikiowa/napcat-docker:latest',
|
||
name: 'kt-qqbot-napcat-test',
|
||
port: 6100,
|
||
reverseWsUrl: 'ws://127.0.0.1:48085/qqbot/onebot/reverse',
|
||
token: 'token-test',
|
||
};
|
||
|
||
const withAccount = service.buildRemoteCreateScript({
|
||
...baseInput,
|
||
account: '2354598417',
|
||
loginPassword: 'qq-password',
|
||
});
|
||
expect(withAccount).toContain("ACCOUNT='2354598417'");
|
||
expect(withAccount).toContain('-e ACCOUNT="$ACCOUNT"');
|
||
expect(withAccount).toContain("NAPCAT_QUICK_PASSWORD='qq-password'");
|
||
expect(withAccount).toContain(
|
||
'-e NAPCAT_QUICK_PASSWORD="$NAPCAT_QUICK_PASSWORD"',
|
||
);
|
||
|
||
const withoutAccount = service.buildRemoteCreateScript(baseInput);
|
||
expect(withoutAccount).not.toContain('-e ACCOUNT="$ACCOUNT"');
|
||
expect(withoutAccount).not.toMatch(/^ACCOUNT=/m);
|
||
expect(withoutAccount).not.toContain('NAPCAT_QUICK_PASSWORD');
|
||
});
|
||
|
||
it('skips docker pull when recreating in place for quick login', () => {
|
||
const service = new QqbotNapcatContainerService(
|
||
{ get: jest.fn().mockReturnValue('') } as any,
|
||
{} as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
|
||
const baseInput = {
|
||
dataDir: '/vol1/docker/kt-qqbot/napcat-instances/kt-qqbot-napcat-test',
|
||
image: 'mlikiowa/napcat-docker:latest',
|
||
name: 'kt-qqbot-napcat-test',
|
||
port: 6100,
|
||
reverseWsUrl: 'ws://127.0.0.1:48085/qqbot/onebot/reverse',
|
||
token: 'token-test',
|
||
};
|
||
|
||
expect(service.buildRemoteCreateScript(baseInput)).toContain('docker pull');
|
||
expect(
|
||
service.buildRemoteCreateScript({ ...baseInput, skipPull: true }),
|
||
).not.toContain('docker pull');
|
||
});
|
||
|
||
it('skips quick-login recreate when the container already carries ACCOUNT', async () => {
|
||
const containerRepository = {
|
||
createQueryBuilder: jest.fn(() => ({
|
||
addSelect: jest.fn().mockReturnThis(),
|
||
andWhere: jest.fn().mockReturnThis(),
|
||
getOne: jest.fn().mockResolvedValue({
|
||
dataDir: '/vol1/docker/kt-qqbot/napcat-instances/kt-qqbot-napcat-x',
|
||
id: 'container-1',
|
||
image: 'mlikiowa/napcat-docker:latest',
|
||
name: 'kt-qqbot-napcat-x',
|
||
reverseWsUrl: 'ws://127.0.0.1:48085/qqbot/onebot/reverse',
|
||
webuiPort: 6100,
|
||
webuiToken: 'token-x',
|
||
}),
|
||
where: jest.fn().mockReturnThis(),
|
||
})),
|
||
update: jest.fn(),
|
||
};
|
||
const service = new QqbotNapcatContainerService(
|
||
{
|
||
get: jest.fn((key: string) =>
|
||
key === 'QQBOT_NAPCAT_CONTAINER_MODE' ? 'ssh' : '',
|
||
),
|
||
} as any,
|
||
containerRepository as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
service.runProcess = jest
|
||
.fn()
|
||
.mockResolvedValue({ stderr: '', stdout: 'ACCOUNT=2354598417\nPATH=/x' });
|
||
|
||
const recreated = await service.ensureRuntimeQuickLogin(
|
||
{ id: 'container-1', name: 'kt-qqbot-napcat-x' },
|
||
'2354598417',
|
||
);
|
||
|
||
expect(recreated).toEqual({ changed: false, ok: true });
|
||
// 仅做了 inspect 检查，没有触发任何重建（rm/run）。
|
||
expect(service.runProcess).toHaveBeenCalledTimes(1);
|
||
expect(service.runProcess.mock.calls[0][1]).toEqual(
|
||
expect.arrayContaining(['sh -s']),
|
||
);
|
||
expect(service.runProcess.mock.calls[0][2]).toContain(
|
||
"docker inspect --format '{{range .Config.Env}}{{println .}}{{end}}' \"$NAME\"",
|
||
);
|
||
expect(containerRepository.update).not.toHaveBeenCalled();
|
||
});
|
||
|
||
it('recreates login env when docker env inspection fails', async () => {
|
||
const containerRepository = {
|
||
createQueryBuilder: jest.fn(() => ({
|
||
addSelect: jest.fn().mockReturnThis(),
|
||
andWhere: jest.fn().mockReturnThis(),
|
||
getOne: jest.fn().mockResolvedValue({
|
||
dataDir: '/vol1/docker/kt-qqbot/napcat-instances/kt-qqbot-napcat-x',
|
||
id: 'container-1',
|
||
image: 'mlikiowa/napcat-docker:latest',
|
||
name: 'kt-qqbot-napcat-x',
|
||
reverseWsUrl: 'ws://127.0.0.1:48085/qqbot/onebot/reverse',
|
||
webuiPort: 6100,
|
||
webuiToken: 'token-x',
|
||
}),
|
||
where: jest.fn().mockReturnThis(),
|
||
})),
|
||
update: jest.fn(),
|
||
};
|
||
const service = new QqbotNapcatContainerService(
|
||
{
|
||
get: jest.fn((key: string) =>
|
||
key === 'QQBOT_NAPCAT_CONTAINER_MODE' ? 'ssh' : '',
|
||
),
|
||
} as any,
|
||
containerRepository as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
service.runProcess = jest
|
||
.fn()
|
||
.mockRejectedValueOnce(new Error('inspect failed'))
|
||
.mockResolvedValueOnce({ stderr: '', stdout: '' });
|
||
|
||
const recreated = await service.ensureRuntimeLoginEnv(
|
||
{ id: 'container-1', name: 'kt-qqbot-napcat-x' },
|
||
{
|
||
loginPassword: 'qq-password',
|
||
selfId: '2354598417',
|
||
},
|
||
);
|
||
|
||
expect(recreated).toEqual({ changed: true, ok: true });
|
||
expect(service.runProcess).toHaveBeenCalledTimes(2);
|
||
expect(service.runProcess.mock.calls[1][2]).toContain(
|
||
"NAPCAT_QUICK_PASSWORD='qq-password'",
|
||
);
|
||
});
|
||
|
||
it('recreates quick-login runtime when password env is missing', async () => {
|
||
const containerRepository = {
|
||
createQueryBuilder: jest.fn(() => ({
|
||
addSelect: jest.fn().mockReturnThis(),
|
||
andWhere: jest.fn().mockReturnThis(),
|
||
getOne: jest.fn().mockResolvedValue({
|
||
dataDir: '/vol1/docker/kt-qqbot/napcat-instances/kt-qqbot-napcat-x',
|
||
id: 'container-1',
|
||
image: 'mlikiowa/napcat-docker:latest',
|
||
name: 'kt-qqbot-napcat-x',
|
||
reverseWsUrl: 'ws://127.0.0.1:48085/qqbot/onebot/reverse',
|
||
webuiPort: 6100,
|
||
webuiToken: 'token-x',
|
||
}),
|
||
where: jest.fn().mockReturnThis(),
|
||
})),
|
||
update: jest.fn(),
|
||
};
|
||
const service = new QqbotNapcatContainerService(
|
||
{
|
||
get: jest.fn((key: string) =>
|
||
key === 'QQBOT_NAPCAT_CONTAINER_MODE' ? 'ssh' : '',
|
||
),
|
||
} as any,
|
||
containerRepository as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
service.runProcess = jest
|
||
.fn()
|
||
.mockResolvedValueOnce({ stderr: '', stdout: 'ACCOUNT=2354598417\n' })
|
||
.mockResolvedValueOnce({ stderr: '', stdout: '' });
|
||
|
||
const recreated = await service.ensureRuntimeLoginEnv(
|
||
{ id: 'container-1', name: 'kt-qqbot-napcat-x' },
|
||
{
|
||
loginPassword: 'qq-password',
|
||
selfId: '2354598417',
|
||
},
|
||
);
|
||
|
||
expect(recreated).toEqual({ changed: true, ok: true });
|
||
expect(service.runProcess).toHaveBeenCalledTimes(2);
|
||
expect(service.runProcess.mock.calls[1][2]).toContain(
|
||
"NAPCAT_QUICK_PASSWORD='qq-password'",
|
||
);
|
||
expect(containerRepository.update).toHaveBeenCalledWith(
|
||
{ id: 'container-1' },
|
||
expect.objectContaining({
|
||
lastError: null,
|
||
status: 'running',
|
||
}),
|
||
);
|
||
});
|
||
|
||
it('removes password env for a quick-login-only runtime stage', async () => {
|
||
const containerRepository = {
|
||
createQueryBuilder: jest.fn(() => ({
|
||
addSelect: jest.fn().mockReturnThis(),
|
||
andWhere: jest.fn().mockReturnThis(),
|
||
getOne: jest.fn().mockResolvedValue({
|
||
dataDir: '/vol1/docker/kt-qqbot/napcat-instances/kt-qqbot-napcat-x',
|
||
id: 'container-1',
|
||
image: 'mlikiowa/napcat-docker:latest',
|
||
name: 'kt-qqbot-napcat-x',
|
||
reverseWsUrl: 'ws://127.0.0.1:48085/qqbot/onebot/reverse',
|
||
webuiPort: 6100,
|
||
webuiToken: 'token-x',
|
||
}),
|
||
where: jest.fn().mockReturnThis(),
|
||
})),
|
||
update: jest.fn(),
|
||
};
|
||
const service = new QqbotNapcatContainerService(
|
||
{
|
||
get: jest.fn((key: string) =>
|
||
key === 'QQBOT_NAPCAT_CONTAINER_MODE' ? 'ssh' : '',
|
||
),
|
||
} as any,
|
||
containerRepository as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
service.runProcess = jest
|
||
.fn()
|
||
.mockResolvedValueOnce({
|
||
stderr: '',
|
||
stdout: 'ACCOUNT=2354598417\nNAPCAT_QUICK_PASSWORD=old-password\n',
|
||
})
|
||
.mockResolvedValueOnce({ stderr: '', stdout: '' });
|
||
|
||
const recreated = await service.ensureRuntimeLoginEnv(
|
||
{ id: 'container-1', name: 'kt-qqbot-napcat-x' },
|
||
{
|
||
clearLoginPassword: true,
|
||
selfId: '2354598417',
|
||
},
|
||
);
|
||
|
||
expect(recreated).toEqual({ changed: true, ok: true });
|
||
expect(service.runProcess.mock.calls[1][2]).not.toContain(
|
||
'NAPCAT_QUICK_PASSWORD=',
|
||
);
|
||
});
|
||
|
||
it('does not report password auto-login success when password env cleanup fails', async () => {
|
||
const containerRepository = {
|
||
createQueryBuilder: jest.fn(() => ({
|
||
addSelect: jest.fn().mockReturnThis(),
|
||
andWhere: jest.fn().mockReturnThis(),
|
||
getOne: jest.fn().mockResolvedValue({
|
||
baseUrl: 'http://127.0.0.1:6100/',
|
||
id: 'container-1',
|
||
name: 'kt-qqbot-napcat-x',
|
||
webuiPort: 6100,
|
||
webuiToken: 'token-x',
|
||
}),
|
||
where: jest.fn().mockReturnThis(),
|
||
})),
|
||
};
|
||
const service = new QqbotNapcatContainerService(
|
||
{
|
||
get: jest.fn((key: string) =>
|
||
key === 'QQBOT_NAPCAT_CONTAINER_MODE' ? 'ssh' : '',
|
||
),
|
||
} as any,
|
||
containerRepository as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
jest
|
||
.spyOn(service, 'ensureRuntimeLoginEnv')
|
||
.mockResolvedValueOnce({ changed: false, ok: true })
|
||
.mockResolvedValueOnce({ changed: true, ok: true })
|
||
.mockResolvedValueOnce({ changed: false, ok: false });
|
||
jest
|
||
.spyOn(service, 'restartAndDetectLoginState')
|
||
.mockResolvedValueOnce({ offlineReason: '历史会话失效', state: 'offline' })
|
||
.mockResolvedValueOnce({ offlineReason: null, state: 'online' });
|
||
|
||
const result = await service.tryAutoLogin(
|
||
{ id: 'container-1' },
|
||
{
|
||
loginPassword: 'qq-password',
|
||
selfId: '2354598417',
|
||
},
|
||
);
|
||
|
||
expect(result).toEqual({
|
||
cleanupFailed: true,
|
||
success: false,
|
||
});
|
||
expect(service.ensureRuntimeLoginEnv).toHaveBeenNthCalledWith(
|
||
3,
|
||
expect.objectContaining({ id: 'container-1' }),
|
||
{
|
||
clearLoginPassword: true,
|
||
selfId: '2354598417',
|
||
},
|
||
);
|
||
});
|
||
|
||
it('does not leave password env untracked when password auto-login fails and cleanup fails', async () => {
|
||
const containerRepository = {
|
||
createQueryBuilder: jest.fn(() => ({
|
||
addSelect: jest.fn().mockReturnThis(),
|
||
andWhere: jest.fn().mockReturnThis(),
|
||
getOne: jest.fn().mockResolvedValue({
|
||
baseUrl: 'http://127.0.0.1:6100/',
|
||
id: 'container-1',
|
||
name: 'kt-qqbot-napcat-x',
|
||
webuiPort: 6100,
|
||
webuiToken: 'token-x',
|
||
}),
|
||
where: jest.fn().mockReturnThis(),
|
||
})),
|
||
};
|
||
const service = new QqbotNapcatContainerService(
|
||
{
|
||
get: jest.fn((key: string) =>
|
||
key === 'QQBOT_NAPCAT_CONTAINER_MODE' ? 'ssh' : '',
|
||
),
|
||
} as any,
|
||
containerRepository as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
jest
|
||
.spyOn(service, 'ensureRuntimeLoginEnv')
|
||
.mockResolvedValueOnce({ changed: false, ok: true })
|
||
.mockResolvedValueOnce({ changed: true, ok: true })
|
||
.mockResolvedValueOnce({ changed: false, ok: false });
|
||
jest
|
||
.spyOn(service, 'restartAndDetectLoginState')
|
||
.mockResolvedValueOnce({ offlineReason: '历史会话失效', state: 'offline' })
|
||
.mockResolvedValueOnce({ offlineReason: '密码登录失败', state: 'offline' });
|
||
|
||
const result = await service.tryAutoLogin(
|
||
{ id: 'container-1' },
|
||
{
|
||
loginPassword: 'qq-password',
|
||
selfId: '2354598417',
|
||
},
|
||
);
|
||
|
||
expect(result).toEqual({
|
||
cleanupFailed: true,
|
||
success: false,
|
||
});
|
||
expect(service.ensureRuntimeLoginEnv).toHaveBeenNthCalledWith(
|
||
3,
|
||
expect.objectContaining({ id: 'container-1' }),
|
||
{
|
||
clearLoginPassword: true,
|
||
selfId: '2354598417',
|
||
},
|
||
);
|
||
});
|
||
|
||
it('does not restart quick auto-login when password env cleanup fails first', async () => {
|
||
const containerRepository = {
|
||
createQueryBuilder: jest.fn(() => ({
|
||
addSelect: jest.fn().mockReturnThis(),
|
||
andWhere: jest.fn().mockReturnThis(),
|
||
getOne: jest.fn().mockResolvedValue({
|
||
baseUrl: 'http://127.0.0.1:6100/',
|
||
id: 'container-1',
|
||
name: 'kt-qqbot-napcat-x',
|
||
webuiPort: 6100,
|
||
webuiToken: 'token-x',
|
||
}),
|
||
where: jest.fn().mockReturnThis(),
|
||
})),
|
||
};
|
||
const service = new QqbotNapcatContainerService(
|
||
{
|
||
get: jest.fn((key: string) =>
|
||
key === 'QQBOT_NAPCAT_CONTAINER_MODE' ? 'ssh' : '',
|
||
),
|
||
} as any,
|
||
containerRepository as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
jest
|
||
.spyOn(service, 'ensureRuntimeLoginEnv')
|
||
.mockResolvedValueOnce({ changed: false, ok: false });
|
||
jest
|
||
.spyOn(service, 'restartAndDetectLoginState')
|
||
.mockResolvedValueOnce({ offlineReason: null, state: 'online' });
|
||
|
||
const result = await service.tryAutoLogin(
|
||
{ id: 'container-1' },
|
||
{
|
||
loginPassword: 'qq-password',
|
||
selfId: '2354598417',
|
||
},
|
||
);
|
||
|
||
expect(result).toEqual({
|
||
cleanupFailed: true,
|
||
success: false,
|
||
});
|
||
expect(service.restartAndDetectLoginState).not.toHaveBeenCalled();
|
||
});
|
||
|
||
it('requires an explicit NapCat image when creating a managed container', async () => {
|
||
const containerRepository = {
|
||
find: jest.fn().mockResolvedValue([]),
|
||
save: jest.fn(),
|
||
};
|
||
const service = new QqbotNapcatContainerService(
|
||
{
|
||
get: jest.fn((key: string) => {
|
||
const values: Record<string, string> = {
|
||
QQBOT_NAPCAT_CONTAINER_MODE: 'ssh',
|
||
QQBOT_NAPCAT_SSH_TARGET: 'nas',
|
||
};
|
||
return values[key] || '';
|
||
}),
|
||
} as any,
|
||
containerRepository as any,
|
||
{} as any,
|
||
new ToolsService(),
|
||
) as any;
|
||
|
||
await expect(service.createManagedContainer('1914728559')).rejects.toThrow(
|
||
'NapCat 镜像未配置，请先设置 QQBOT_NAPCAT_IMAGE',
|
||
);
|
||
expect(containerRepository.save).not.toHaveBeenCalled();
|
||
});
|
||
});
|