SnowLuma v2.4.2 source snapshot

packages/protocol/tests/oidb-services/highway/ntv2-upload-request.test.ts

@@ -0,0 +1,79 @@
+import { describe, expect, it, vi } from 'vitest';
+import { protobuf_encode } from '@snowluma/proton';
+import type { SendPacketResult } from '@snowluma/common/packet-sender';
+import type { OidbBase } from '@snowluma/proto-defs/oidb';
+import type { NTV2UploadRichMediaResp } from '@snowluma/proto-defs/highway';
+import { OidbError } from '../../../src/oidb-service';
+import { Ntv2UploadRequest } from '../../../src/oidb-services/highway/ntv2-upload-request';
+
+function packet(body: NTV2UploadRichMediaResp, errorCode = 0): SendPacketResult {
+    return {
+        success: true,
+        gotResponse: true,
+        errorCode: 0,
+        errorMessage: '',
+        responseData: Buffer.from(
+            protobuf_encode<OidbBase<NTV2UploadRichMediaResp>>({
+                command: 0x11c5,
+                subCommand: 100,
+                errorCode,
+                errorMsg: errorCode ? 'denied' : '',
+                body,
+            }),
+        ),
+    };
+}
+
+const params: Ntv2UploadRequest.Params = {
+    oidbCmd: 0x11c5,
+    isGroup: false,
+    targetIdOrUid: 'u_peer',
+    requestId: 1,
+    businessType: 1,
+    uploadInfo: [],
+    compatQmsgSceneType: 1,
+    extBizInfo: {},
+    tryFast: true,
+    clientRandomId: 1n,
+};
+
+describe('Ntv2UploadRequest', () => {
+    it.each([
+        { tempGroupId: 0 },
+        { tempGroupId: -1 },
+        { tempGroupId: 1.5 },
+        { tempGroupId: 700, isGroup: true },
+        { tempGroupId: 700, targetIdOrUid: '' },
+        { tempGroupId: 700, targetIdOrUid: 1001 },
+    ])('rejects invalid temp context before sending: %j', async (invalid) => {
+        const sendRawPacket = vi.fn();
+        await expect(
+            Ntv2UploadRequest.invoke({ sendRawPacket }, { ...params, ...invalid }),
+        ).rejects.toThrow('invalid temp-session image upload context');
+        expect(sendRawPacket).not.toHaveBeenCalled();
+    });
+
+    it('sends the resolved command on the default OIDB wire name', async () => {
+        const sendRawPacket = vi.fn(async () =>
+            packet({
+                upload: { msgInfo: { msgInfoBody: [] } },
+            }),
+        );
+        await Ntv2UploadRequest.invoke({ sendRawPacket }, params);
+        expect(sendRawPacket.mock.calls[0]![0]).toBe('OidbSvcTrpcTcp.0x11c5_100');
+    });
+
+    it('raises OidbError when the envelope is rejected', async () => {
+        const sendRawPacket = vi.fn(async () => packet({ upload: { msgInfo: {} } }, 42));
+        await expect(Ntv2UploadRequest.invoke({ sendRawPacket }, params)).rejects.toBeInstanceOf(
+            OidbError,
+        );
+    });
+
+    it('fails in deserialize when msgInfo is missing', async () => {
+        const sendRawPacket = vi.fn(async () => packet({ upload: { uKey: 'k' } }));
+        await expect(Ntv2UploadRequest.invoke({ sendRawPacket }, params)).rejects.toThrow(
+            'upload response missing msgInfo',
+        );
+    });
+});