@qnsp/storage-sdk 0.2.1 → 0.3.1

package/src/index.test.ts

@@ -1,9 +1,34 @@
+import { clearActivationCache } from "@qnsp/sdk-activation";
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 import { StorageClient, StorageEventsClient } from "./index.js";
 
-const mockTenantId = "tenant-123";
+const mockTenantId = "a1b2c3d4-e5f6-7890-abcd-ef1234567890";
 const baseUrl = "https://storage.qnsp.example/";
 
+// Valid UUID test constants
+const mockUploadId = "11111111-1111-4111-a111-111111111111";
+const mockUploadId2 = "22222222-2222-4222-a222-222222222222";
+const mockUploadIdErr = "33333333-3333-4333-a333-333333333333";
+const mockDocumentId = "44444444-4444-4444-a444-444444444444";
+
+const MOCK_ACTIVATION_RESPONSE = {
+	activated: true,
+	tenantId: "a1b2c3d4-e5f6-4789-8abc-def012345678",
+	tier: "dev-pro",
+	activationToken: "tok_test",
+	expiresInSeconds: 3600,
+	activatedAt: new Date().toISOString(),
+	limits: {
+		storageGB: 50,
+		apiCalls: 100_000,
+		enclavesEnabled: false,
+		aiTrainingEnabled: false,
+		aiInferenceEnabled: true,
+		sseEnabled: true,
+		vaultEnabled: true,
+	},
+};
+
 function jsonResponse(payload: unknown, init?: ResponseInit): Response {
 	return new Response(JSON.stringify(payload), {
 		status: 200,
@@ -43,6 +68,7 @@ async function collectStream(stream: ReadableStream<Uint8Array>): Promise<Uint8A
 
 describe("StorageClient", () => {
 	beforeEach(() => {
+		clearActivationCache();
 		vi.stubGlobal("fetch", vi.fn());
 	});
 
@@ -60,8 +86,8 @@ describe("StorageClient", () => {
 		});
 
 		const expectedResponse = {
-			uploadId: "upload-1",
-			documentId: "document-1",
+			uploadId: mockUploadId,
+			documentId: mockDocumentId,
 			tenantId: mockTenantId,
 			chunkSizeBytes: 16_777_216,
 			totalSizeBytes: 42_000_000,
@@ -76,9 +102,9 @@ describe("StorageClient", () => {
 			},
 		};
 
-		(globalThis.fetch as unknown as ReturnType<typeof vi.fn>).mockResolvedValue(
-			jsonResponse(expectedResponse),
-		);
+		(globalThis.fetch as unknown as ReturnType<typeof vi.fn>)
+			.mockResolvedValueOnce(jsonResponse(MOCK_ACTIVATION_RESPONSE))
+			.mockResolvedValueOnce(jsonResponse(expectedResponse));
 
 		const result = await client.initiateUpload({
 			name: "contract.pdf",
@@ -88,8 +114,8 @@ describe("StorageClient", () => {
 		});
 
 		expect(result).toEqual(expectedResponse);
-		expect(globalThis.fetch).toHaveBeenCalledTimes(1);
-		const call = (globalThis.fetch as unknown as ReturnType<typeof vi.fn>).mock.calls.at(0);
+		expect(globalThis.fetch).toHaveBeenCalledTimes(2);
+		const call = (globalThis.fetch as unknown as ReturnType<typeof vi.fn>).mock.calls.at(1);
 		expect(call).toBeDefined();
 		if (!call) throw new Error("fetch not invoked");
 		const [url, init] = call;
@@ -118,7 +144,7 @@ describe("StorageClient", () => {
 		});
 
 		const expectedPayload = {
-			uploadId: "upload-1",
+			uploadId: mockUploadId,
 			partId: 1,
 			status: "uploaded",
 			sizeBytes: 3,
@@ -131,16 +157,16 @@ describe("StorageClient", () => {
 			resumeToken: null,
 		};
 
-		(globalThis.fetch as unknown as ReturnType<typeof vi.fn>).mockResolvedValue(
-			jsonResponse(expectedPayload),
-		);
+		(globalThis.fetch as unknown as ReturnType<typeof vi.fn>)
+			.mockResolvedValueOnce(jsonResponse(MOCK_ACTIVATION_RESPONSE))
+			.mockResolvedValueOnce(jsonResponse(expectedPayload));
 
 		const source = Buffer.from([0xde, 0xad, 0xbe]);
 
-		const result = await client.uploadPart("upload-1", 1, source);
+		const result = await client.uploadPart(mockUploadId, 1, source);
 		expect(result).toEqual(expectedPayload);
-		expect(globalThis.fetch).toHaveBeenCalledTimes(1);
-		const uploadCall = (globalThis.fetch as unknown as ReturnType<typeof vi.fn>).mock.calls.at(0);
+		expect(globalThis.fetch).toHaveBeenCalledTimes(2);
+		const uploadCall = (globalThis.fetch as unknown as ReturnType<typeof vi.fn>).mock.calls.at(1);
 		expect(uploadCall).toBeDefined();
 		if (!uploadCall) throw new Error("fetch not invoked");
 		const [, init] = uploadCall;
@@ -158,6 +184,7 @@ describe("StorageClient", () => {
 	it("parses ranged download responses and surfaces stream metadata", async () => {
 		const client = new StorageClient({
 			baseUrl,
+			apiKey: "test-api-key",
 			tenantId: mockTenantId,
 		});
 
@@ -177,14 +204,16 @@ describe("StorageClient", () => {
 			},
 		});
 
-		(globalThis.fetch as unknown as ReturnType<typeof vi.fn>).mockResolvedValue(response);
+		(globalThis.fetch as unknown as ReturnType<typeof vi.fn>)
+			.mockResolvedValueOnce(jsonResponse(MOCK_ACTIVATION_RESPONSE))
+			.mockResolvedValueOnce(response);
 
-		const result = await client.downloadStream("document-1", 2, {
+		const result = await client.downloadStream(mockDocumentId, 2, {
 			range: "bytes=0-2",
 		});
 
 		expect(globalThis.fetch).toHaveBeenCalledWith(
-			"https://storage.qnsp.example/storage/v1/documents/document-1/versions/2/content?tenantId=tenant-123&range=bytes%3D0-2",
+			`https://storage.qnsp.example/storage/v1/documents/${mockDocumentId}/versions/2/content?tenantId=${mockTenantId}&range=bytes%3D0-2`,
 			expect.objectContaining({
 				method: "GET",
 				headers: expect.objectContaining({
@@ -207,13 +236,14 @@ describe("StorageClient", () => {
 		const telemetry = { record: vi.fn() };
 		const client = new StorageClient({
 			baseUrl,
+			apiKey: "test-api-key",
 			tenantId: mockTenantId,
 			telemetry,
 		});
 
 		const statusResponse = {
-			uploadId: "upload-123",
-			documentId: "doc-1",
+			uploadId: mockUploadId2,
+			documentId: mockDocumentId,
 			tenantId: mockTenantId,
 			status: "pending",
 			chunkSizeBytes: 4,
@@ -228,11 +258,11 @@ describe("StorageClient", () => {
 			lastPartNumber: 1,
 		};
 
-		(globalThis.fetch as unknown as ReturnType<typeof vi.fn>).mockResolvedValue(
-			jsonResponse(statusResponse),
-		);
+		(globalThis.fetch as unknown as ReturnType<typeof vi.fn>)
+			.mockResolvedValueOnce(jsonResponse(MOCK_ACTIVATION_RESPONSE))
+			.mockResolvedValueOnce(jsonResponse(statusResponse));
 
-		await client.getUploadStatus("upload-123");
+		await client.getUploadStatus(mockUploadId2);
 
 		expect(telemetry.record).toHaveBeenCalled();
 		const event = (telemetry.record as ReturnType<typeof vi.fn>).mock.calls.at(-1)?.[0];
@@ -248,15 +278,16 @@ describe("StorageClient", () => {
 		const telemetry = { record: vi.fn() };
 		const client = new StorageClient({
 			baseUrl,
+			apiKey: "test-api-key",
 			tenantId: mockTenantId,
 			telemetry,
 		});
 
-		(globalThis.fetch as unknown as ReturnType<typeof vi.fn>).mockResolvedValue(
-			new Response("boom", { status: 500, statusText: "error" }),
-		);
+		(globalThis.fetch as unknown as ReturnType<typeof vi.fn>)
+			.mockResolvedValueOnce(jsonResponse(MOCK_ACTIVATION_RESPONSE))
+			.mockResolvedValueOnce(new Response("boom", { status: 500, statusText: "error" }));
 
-		await expect(client.completeUpload("upload-err")).rejects.toThrow(/Storage API error/);
+		await expect(client.completeUpload(mockUploadIdErr)).rejects.toThrow(/Storage API error/);
 
 		const event = (telemetry.record as ReturnType<typeof vi.fn>).mock.calls.at(-1)?.[0];
 		expect(event).toBeDefined();
@@ -271,12 +302,13 @@ describe("StorageClient", () => {
 		const telemetry = { record: vi.fn() };
 		const client = new StorageClient({
 			baseUrl,
+			apiKey: "test-api-key",
 			tenantId: mockTenantId,
 			telemetry,
 		});
 
 		const expectedPayload = {
-			uploadId: "upload-1",
+			uploadId: mockUploadId,
 			partId: 1,
 			status: "uploaded",
 			sizeBytes: 3,
@@ -289,11 +321,11 @@ describe("StorageClient", () => {
 			resumeToken: null,
 		};
 
-		(globalThis.fetch as unknown as ReturnType<typeof vi.fn>).mockResolvedValue(
-			jsonResponse(expectedPayload),
-		);
+		(globalThis.fetch as unknown as ReturnType<typeof vi.fn>)
+			.mockResolvedValueOnce(jsonResponse(MOCK_ACTIVATION_RESPONSE))
+			.mockResolvedValueOnce(jsonResponse(expectedPayload));
 
-		await client.uploadPart("upload-1", 1, Buffer.from([1, 2, 3]));
+		await client.uploadPart(mockUploadId, 1, Buffer.from([1, 2, 3]));
 
 		const event = (telemetry.record as ReturnType<typeof vi.fn>).mock.calls.at(-1)?.[0];
 		expect(event).toBeDefined();
@@ -312,10 +344,12 @@ describe("StorageClient", () => {
 		});
 
 		(globalThis.fetch as unknown as ReturnType<typeof vi.fn>)
+			// activation
+			.mockResolvedValueOnce(jsonResponse(MOCK_ACTIVATION_RESPONSE))
 			// PATCH response
 			.mockResolvedValueOnce(
 				jsonResponse({
-					documentId: "doc-1",
+					documentId: mockDocumentId,
 					tenantId: mockTenantId,
 					compliance: {
 						retentionMode: "compliance",
@@ -328,7 +362,7 @@ describe("StorageClient", () => {
 			// GET response
 			.mockResolvedValueOnce(
 				jsonResponse({
-					documentId: "doc-1",
+					documentId: mockDocumentId,
 					tenantId: mockTenantId,
 					compliance: {
 						retentionMode: "compliance",
@@ -344,26 +378,30 @@ describe("StorageClient", () => {
 				}),
 			);
 
-		const updated = await client.updateDocumentPolicies("doc-1", {
+		const updated = await client.updateDocumentPolicies(mockDocumentId, {
 			retentionMode: "compliance",
 			retainUntil: "2026-01-01T00:00:00Z",
 			legalHolds: ["hold-1"],
 		});
-		expect(updated.documentId).toBe("doc-1");
+		expect(updated.documentId).toBe(mockDocumentId);
 		const patchCall = (globalThis.fetch as unknown as ReturnType<typeof vi.fn>).mock
-			.calls[0] as unknown as [string, RequestInit];
-		expect(patchCall[0]).toBe("https://storage.qnsp.example/storage/v1/documents/doc-1/policies");
+			.calls[1] as unknown as [string, RequestInit];
+		expect(patchCall[0]).toBe(
+			`https://storage.qnsp.example/storage/v1/documents/${mockDocumentId}/policies`,
+		);
 		expect(patchCall[1]?.headers).toMatchObject({
 			"Content-Type": "application/json",
 			Authorization: "Bearer key",
 			"x-tenant-id": mockTenantId,
 		});
 
-		const policies = await client.getDocumentPolicies("doc-1");
+		const policies = await client.getDocumentPolicies(mockDocumentId);
 		expect(policies.tenantId).toBe(mockTenantId);
 		const getCall = (globalThis.fetch as unknown as ReturnType<typeof vi.fn>).mock
-			.calls[1] as unknown as [string, RequestInit];
-		expect(getCall[0]).toBe("https://storage.qnsp.example/storage/v1/documents/doc-1/policies");
+			.calls[2] as unknown as [string, RequestInit];
+		expect(getCall[0]).toBe(
+			`https://storage.qnsp.example/storage/v1/documents/${mockDocumentId}/policies`,
+		);
 		expect(getCall[1]?.headers).toMatchObject({
 			"Content-Type": "application/json",
 			Authorization: "Bearer key",
@@ -379,10 +417,12 @@ describe("StorageClient", () => {
 		});
 
 		(globalThis.fetch as unknown as ReturnType<typeof vi.fn>)
+			// activation
+			.mockResolvedValueOnce(jsonResponse(MOCK_ACTIVATION_RESPONSE))
 			// apply
 			.mockResolvedValueOnce(
 				jsonResponse({
-					documentId: "doc-1",
+					documentId: mockDocumentId,
 					tenantId: mockTenantId,
 					legalHolds: ["hold-9"],
 				}),
@@ -390,21 +430,23 @@ describe("StorageClient", () => {
 			// release (204)
 			.mockResolvedValueOnce(new Response(null, { status: 204 }));
 
-		const applied = await client.applyLegalHold("doc-1", { holdId: "hold-9" });
+		const applied = await client.applyLegalHold(mockDocumentId, { holdId: "hold-9" });
 		expect(applied.legalHolds).toContain("hold-9");
 		const postCall = (globalThis.fetch as unknown as ReturnType<typeof vi.fn>).mock
-			.calls[0] as unknown as [string, RequestInit];
-		expect(postCall[0]).toBe("https://storage.qnsp.example/storage/v1/documents/doc-1/legal-holds");
+			.calls[1] as unknown as [string, RequestInit];
+		expect(postCall[0]).toBe(
+			`https://storage.qnsp.example/storage/v1/documents/${mockDocumentId}/legal-holds`,
+		);
 		expect(postCall[1]?.method).toBe("POST");
 		expect(postCall[1]?.headers).toMatchObject({
 			"x-tenant-id": mockTenantId,
 		});
 
-		await client.releaseLegalHold("doc-1", "hold-9");
+		await client.releaseLegalHold(mockDocumentId, "hold-9");
 		const delCall = (globalThis.fetch as unknown as ReturnType<typeof vi.fn>).mock
-			.calls[1] as unknown as [string, RequestInit];
+			.calls[2] as unknown as [string, RequestInit];
 		expect(delCall[0]).toBe(
-			"https://storage.qnsp.example/storage/v1/documents/doc-1/legal-holds/hold-9",
+			`https://storage.qnsp.example/storage/v1/documents/${mockDocumentId}/legal-holds/hold-9`,
 		);
 		expect(delCall[1]?.method).toBe("DELETE");
 		expect(delCall[1]?.headers).toMatchObject({
@@ -427,6 +469,7 @@ describe("StorageEventsClient", () => {
 		const telemetry = { record: vi.fn() };
 		const client = new StorageEventsClient({
 			baseUrl,
+			apiKey: "test-api-key",
 			telemetry,
 		});