azure-mock 2.35.0 → 2.36.0

package/dist/index.js

@@ -1,6 +1,6 @@
+import { RestError, createHttpHeaders, createPipelineRequest } from "@azure/core-rest-pipeline";
 import { toHttpHeadersLike } from "@azure/core-http-compat";
-import { ID_SEPARATOR, InvalidOperationError, Operation, exhaustiveGuard, getOrCreate, getResultAsync, html, noop, streamToText, takeOne } from "@esposter/shared";
-import { createHttpHeaders, createPipelineRequest } from "@azure/core-rest-pipeline";
+import { ID_SEPARATOR, InvalidOperationError, Operation, exhaustiveGuard, getOrCreate, getResult, html, noop, streamToText, takeOne } from "@esposter/shared";
 import { AnonymousCredential } from "@azure/storage-blob";
 import { Readable } from "node:stream";
 import { MAX_QUEUE_VISIBILITY_TIMEOUT_MS, deserializeClause, deserializeKey, getTableNullClause, serializeClauses } from "@esposter/db";
@@ -22,7 +22,7 @@ var __copyProps = (to, from, except, desc) => {
 	}
 	return to;
 };
-var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", {
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(isNodeMode || !mod || !mod.__esModule || !__hasOwnProp.call(mod, "default") ? __defProp(target, "default", {
 	value: mod,
 	enumerable: true
 }) : target, mod));
@@ -30,12 +30,10 @@ const MOCK_BLOB_BASE_URL = "https://mockaccount.blob.core.windows.net";
 const MOCK_QUEUE_BASE_URL = "https://mockaccount.queue.core.windows.net";
 const MOCK_SEARCH_BASE_URL = "https://mockaccount.search.windows.net";
 const MOCK_TABLE_BASE_URL = "https://mockaccount.table.core.windows.net";
-var MockRestError = class extends Error {
-	statusCode;
+var MockRestError = class extends RestError {
 	constructor(message, statusCode) {
-		super(message);
+		super(message, { statusCode });
 		this.name = "MockRestError";
-		this.statusCode = statusCode;
 	}
 };
 const toWebResourceLike = (request) => ({
@@ -70,7 +68,16 @@ const createMockResponse = (status, url = "") => ({
 	status
 });
 const getMockSasUrl = (url, permissions, resourceType) => `${url}?sv=2025-11-05&${resourceType ? `sr=${resourceType}&` : ""}sig=mock-signature&st=1970-01-01T00:00:00Z&se=2099-12-31T23:59:59Z&sp=${permissions?.toString() ?? "r"}`;
-const MockContainerDatabase = /* @__PURE__ */ new Map();
+const MockContainerCreatedOnDatabase = /* @__PURE__ */ new Map();
+const MOCK_BLOB_SEEDED_CREATED_ON = /* @__PURE__ */ new Date(-864e13);
+const getMockContainerCreatedOnKey = (containerName, blobName) => `${containerName}/${blobName}`;
+var MockContainerDatabaseMap = class extends Map {
+	clear() {
+		super.clear();
+		MockContainerCreatedOnDatabase.clear();
+	}
+};
+const MockContainerDatabase = new MockContainerDatabaseMap();
 const MockEventGridDatabase = /* @__PURE__ */ new Map();
 const MockQueueDatabase = /* @__PURE__ */ new Map();
 const MockSearchDatabase = /* @__PURE__ */ new Map();
@@ -87,6 +94,17 @@ const getAzureErrorXml = (errorCode, errorMessage) => html`<?xml version="1.0" e
       <code>${errorCode}</code>
       <Message>${errorMessage}</Message>
     </Error>`;
+const decodeSegment = (segment) => getResult(() => decodeURIComponent(segment)).unwrapOr(segment);
+const getBlobUrlParts = (url) => {
+	const parsedUrl = getResult(() => new URL(url)).unwrapOr(void 0);
+	if (!parsedUrl) return void 0;
+	const pathSegments = parsedUrl.pathname.split("/").filter(Boolean);
+	if (pathSegments.length < 2) return void 0;
+	return {
+		blobName: decodeSegment(pathSegments.slice(1).join("/")),
+		containerName: decodeSegment(takeOne(pathSegments))
+	};
+};
 var MockBlobBatchClient = class {
 	url;
 	constructor(url) {
@@ -103,8 +121,8 @@ var MockBlobBatchClient = class {
 		let subResponsesSucceededCount = 0;
 		let subResponsesFailedCount = 0;
 		for (const url of urls) {
-			const pathSegments = new URL(url).pathname.split("/").filter(Boolean);
-			if (pathSegments.length < 2) {
+			const urlParts = getBlobUrlParts(url);
+			if (!urlParts) {
 				const errorCode = "InvalidUri";
 				const statusMessage = "Invalid blob URL format.";
 				subResponses.push({
@@ -121,11 +139,11 @@ var MockBlobBatchClient = class {
 				subResponsesFailedCount++;
 				continue;
 			}
-			const containerName = takeOne(pathSegments);
-			const blobName = pathSegments.slice(1).join("/");
+			const { blobName, containerName } = urlParts;
 			const container = this.getContainer(containerName);
 			if (container.has(blobName)) {
 				container.delete(blobName);
+				MockContainerCreatedOnDatabase.delete(getMockContainerCreatedOnKey(containerName, blobName));
 				subResponses.push({
 					_request: {
 						credential,
@@ -170,6 +188,7 @@ var MockBlobBatchClient = class {
 		return getOrCreate(MockContainerDatabase, containerName, () => /* @__PURE__ */ new Map());
 	}
 };
+const getBlobUrl = (containerName, blobName) => `${MOCK_BLOB_BASE_URL}/${containerName}/${blobName.split("/").map((segment) => encodeURIComponent(segment)).join("/")}`;
 var MockBlobClient = class {
 	connectionString;
 	containerName;
@@ -183,21 +202,21 @@ var MockBlobClient = class {
 		this.connectionString = connectionString;
 		this.containerName = containerName;
 		this.name = blobName;
-		this.url = `${MOCK_BLOB_BASE_URL}/${this.containerName}/${this.name}`;
+		this.url = getBlobUrl(this.containerName, this.name);
 	}
 	abortCopyFromURL() {
 		throw new Error("Method not implemented.");
 	}
 	beginCopyFromURL(copySource) {
-		const pathSegments = new URL(copySource).pathname.split("/").filter(Boolean);
-		if (pathSegments.length < 2) throw new MockRestError("Invalid copy source URL format", 400);
-		const sourceContainerName = takeOne(pathSegments);
-		const sourceBlobName = pathSegments.slice(1).join("/");
+		const sourceParts = getBlobUrlParts(copySource);
+		if (!sourceParts) throw new MockRestError("Invalid copy source URL format", 400);
+		const { blobName: sourceBlobName, containerName: sourceContainerName } = sourceParts;
 		const sourceContainer = MockContainerDatabase.get(sourceContainerName);
 		if (!sourceContainer) throw new MockRestError("Source container not found", 404);
 		const sourceData = sourceContainer.get(sourceBlobName);
 		if (!sourceData) throw new MockRestError("Source blob not found", 404);
 		this.container.set(this.name, Buffer.from(sourceData));
+		MockContainerCreatedOnDatabase.set(getMockContainerCreatedOnKey(this.containerName, this.name), /* @__PURE__ */ new Date());
 		const response = { _response: createMockResponse(202, `${this.url}?comp=copy`) };
 		return Promise.resolve({
 			cancelOperation: () => Promise.resolve(),
@@ -220,11 +239,15 @@ var MockBlobClient = class {
 	delete() {
 		if (!this.container.has(this.name)) throw new MockRestError("The specified blob does not exist.", 404);
 		this.container.delete(this.name);
+		MockContainerCreatedOnDatabase.delete(getMockContainerCreatedOnKey(this.containerName, this.name));
 		return Promise.resolve({ _response: createMockResponse(200) });
 	}
 	deleteIfExists() {
 		const succeeded = this.container.has(this.name);
-		if (succeeded) this.container.delete(this.name);
+		if (succeeded) {
+			this.container.delete(this.name);
+			MockContainerCreatedOnDatabase.delete(getMockContainerCreatedOnKey(this.containerName, this.name));
+		}
 		return Promise.resolve({
 			_response: createMockResponse(succeeded ? 200 : 404),
 			succeeded
@@ -249,7 +272,7 @@ var MockBlobClient = class {
 		throw new Error("Method not implemented.");
 	}
 	exists() {
-		throw new Error("Method not implemented.");
+		return Promise.resolve(this.container.has(this.name));
 	}
 	generateSasStringToSign() {
 		throw new Error("Method not implemented.");
@@ -362,8 +385,11 @@ var MockBlockBlobClient = class extends MockBlobClient {
 	syncUploadFromURL() {
 		throw new Error("Method not implemented.");
 	}
-	async upload(body, _contentLength) {
-		this.container.set(this.name, await bodyToBuffer(body));
+	async upload(body, _contentLength, options) {
+		const buffer = await bodyToBuffer(body);
+		if (options?.conditions?.ifNoneMatch === "*" && this.container.has(this.name)) throw new MockRestError("The specified blob already exists.", 409);
+		this.container.set(this.name, buffer);
+		MockContainerCreatedOnDatabase.set(getMockContainerCreatedOnKey(this.containerName, this.name), /* @__PURE__ */ new Date());
 		return { _response: createMockResponse(201) };
 	}
 	uploadBrowserData() {
@@ -438,7 +464,8 @@ var MockContainerClient = class {
 	deleteBlob(blobName) {
 		if (!this.container.has(blobName)) throw new MockRestError("The specified blob does not exist.", 404);
 		this.container.delete(blobName);
-		return Promise.resolve({ _response: createMockResponse(200, `${this.url}/${blobName}`) });
+		MockContainerCreatedOnDatabase.delete(getMockContainerCreatedOnKey(this.containerName, blobName));
+		return Promise.resolve({ _response: createMockResponse(200, getBlobUrl(this.containerName, blobName)) });
 	}
 	deleteIfExists() {
 		throw new Error("Method not implemented.");
@@ -528,8 +555,8 @@ var MockContainerClient = class {
 			}
 		};
 	}
-	listBlobsFlat() {
-		const blobItemIterator = this.#getBlobItemIterator();
+	listBlobsFlat(options) {
+		const blobItemIterator = this.#getBlobItemIterator(options);
 		return {
 			byPage: () => async function* () {
 				const allBlobItems = [];
@@ -545,14 +572,14 @@ var MockContainerClient = class {
 						parsedBody: {
 							containerName: this.containerName,
 							marker: "",
-							prefix: "",
+							prefix: options?.prefix ?? "",
 							segment: { blobItems: allBlobItems },
 							serviceEndpoint: ""
 						}
 					},
 					containerName: this.containerName,
 					marker: "",
-					prefix: "",
+					prefix: options?.prefix ?? "",
 					segment: { blobItems: allBlobItems },
 					serviceEndpoint: ""
 				});
@@ -591,6 +618,7 @@ var MockContainerClient = class {
 					blobType: "BlockBlob",
 					contentLength: buffer.length,
 					contentType: "application/octet-stream",
+					createdOn: MockContainerCreatedOnDatabase.get(getMockContainerCreatedOnKey(this.containerName, name)) ?? MOCK_BLOB_SEEDED_CREATED_ON,
 					etag: `"${crypto.randomUUID()}"`,
 					lastModified: /* @__PURE__ */ new Date(),
 					leaseState: "available",
@@ -612,21 +640,26 @@ var MockContainerClient = class {
 			...blobItem
 		});
 	}
-	async *#getBlobItemIterator() {
-		for (const [name, buffer] of this.container.entries()) yield await Promise.resolve({
-			deleted: false,
-			name,
-			properties: {
-				blobType: "BlockBlob",
-				contentLength: buffer.length,
-				contentType: "application/octet-stream",
-				etag: `"${crypto.randomUUID()}"`,
-				lastModified: /* @__PURE__ */ new Date(),
-				leaseState: "available",
-				leaseStatus: "unlocked"
-			},
-			snapshot: ""
-		});
+	async *#getBlobItemIterator(options) {
+		const prefix = options?.prefix ?? "";
+		for (const [name, buffer] of this.container.entries()) {
+			if (!name.startsWith(prefix)) continue;
+			yield await Promise.resolve({
+				deleted: false,
+				name,
+				properties: {
+					blobType: "BlockBlob",
+					contentLength: buffer.length,
+					contentType: "application/octet-stream",
+					createdOn: MockContainerCreatedOnDatabase.get(getMockContainerCreatedOnKey(this.containerName, name)) ?? MOCK_BLOB_SEEDED_CREATED_ON,
+					etag: `"${crypto.randomUUID()}"`,
+					lastModified: /* @__PURE__ */ new Date(),
+					leaseState: "available",
+					leaseStatus: "unlocked"
+				},
+				snapshot: ""
+			});
+		}
 	}
 };
 /**
@@ -837,7 +870,8 @@ var import_fast_deep_equal = /* @__PURE__ */ __toESM((/* @__PURE__ */ __commonJS
 	};
 })))(), 1);
 const isNullClause = (clause) => {
-	return (0, import_fast_deep_equal.default)(clause, getTableNullClause(clause.key));
+	const tableNullClause = getTableNullClause(clause.key);
+	return (0, import_fast_deep_equal.default)(clause, tableNullClause);
 };
 const createFilterPredicate = (filter) => {
 	const orGroups = filter.replaceAll(String.raw`(`, " ").replaceAll(String.raw`)`, "").split(/\s+and\s+/iu).filter(Boolean).map((group) => group.split(/\s+or\s+/iu).filter(Boolean));
@@ -1004,6 +1038,11 @@ var MockServiceBusSender = class {
 		return Promise.resolve();
 	}
 };
+const compareByCompositeKey = (a, b) => {
+	if (a.partitionKey !== b.partitionKey) return a.partitionKey < b.partitionKey ? -1 : 1;
+	else if (a.rowKey === b.rowKey) return 0;
+	else return a.rowKey < b.rowKey ? -1 : 1;
+};
 /**
 * An in-memory mock of the Azure TableClient.
 * It uses a Map to simulate table storage and correctly implements the TableClient interface.
@@ -1024,9 +1063,7 @@ var MockTableClient = class {
 		this.url = `${MOCK_TABLE_BASE_URL}/${this.tableName}`;
 	}
 	createEntity(entity) {
-		const key = this.#getCompositeKey(entity.partitionKey, entity.rowKey);
-		if (this.table.has(key)) throw new MockRestError("The specified entity already exists.", 409);
-		this.table.set(key, entity);
+		this.#applyCreate(entity);
 		return Promise.resolve({
 			date: /* @__PURE__ */ new Date(),
 			etag: this.#getEtag()
@@ -1036,9 +1073,7 @@ var MockTableClient = class {
 		throw new Error("Method not implemented.");
 	}
 	deleteEntity(partitionKey, rowKey) {
-		const key = this.#getCompositeKey(partitionKey, rowKey);
-		if (!this.table.has(key)) throw new MockRestError("The specified resource does not exist.", 404);
-		this.table.delete(key);
+		this.#applyDelete(partitionKey, rowKey);
 		return Promise.resolve({});
 	}
 	deleteTable() {
@@ -1053,15 +1088,18 @@ var MockTableClient = class {
 		if (!entity) throw new MockRestError("The specified resource does not exist.", 404);
 		return Promise.resolve({
 			...entity,
-			etag: this.#getEtag()
+			etag: entity.etag ?? this.#getEtag()
 		});
 	}
 	listEntities(options) {
 		const withMetadata = this.#withMetadata.bind(this);
 		const filter = options?.queryOptions?.filter;
-		const tableEntities = [...this.table.values()];
+		const tableEntities = [...this.table.values()].toSorted(compareByCompositeKey);
 		const predicate = filter ? createFilterPredicate(filter) : void 0;
 		const resultTableEntities = predicate ? tableEntities.filter((e) => predicate(e)) : tableEntities;
+		const entityIterator = (async function* (entities) {
+			for (const entity of entities) yield await Promise.resolve(withMetadata(entity));
+		})(resultTableEntities);
 		return {
 			byPage: ({ maxPageSize } = {}) => (async function* (entities) {
 				if (maxPageSize !== void 0 && maxPageSize <= 0) throw new RangeError("maxPageSize must be greater than 0.");
@@ -1074,9 +1112,7 @@ var MockTableClient = class {
 				}
 				for (let i = 0; i < allEntitiesWithMetadata.length; i += maxPageSize) yield await Promise.resolve(allEntitiesWithMetadata.slice(i, i + maxPageSize));
 			})(resultTableEntities),
-			next: () => (async function* (entities) {
-				for (const entity of entities) yield await Promise.resolve(withMetadata(entity));
-			})(resultTableEntities).next(),
+			next: () => entityIterator.next(),
 			[Symbol.asyncIterator]() {
 				return this;
 			}
@@ -1085,24 +1121,24 @@ var MockTableClient = class {
 	setAccessPolicy() {
 		throw new Error("Method not implemented.");
 	}
-	async submitTransaction(actions) {
+	submitTransaction(actions) {
 		let partitionKey;
 		for (const [, entity] of actions) if (partitionKey === void 0) partitionKey = entity.partitionKey;
 		else if (entity.partitionKey !== partitionKey) throw new MockRestError("All transaction actions must target the same partitionKey.", 400);
 		const snapshot = new Map(this.table);
-		await getResultAsync(async () => {
+		getResult(() => {
 			for (const [type, entity, updateMode] of actions) switch (type) {
 				case "create":
-					await this.createEntity(entity);
+					this.#applyCreate(entity);
 					break;
 				case "delete":
-					await this.deleteEntity(entity.partitionKey, entity.rowKey);
+					this.#applyDelete(entity.partitionKey, entity.rowKey);
 					break;
 				case "update":
-					await this.updateEntity(entity, updateMode);
+					this.#applyUpdate(entity, updateMode);
 					break;
 				case "upsert":
-					await this.upsertEntity(entity, updateMode);
+					this.#applyUpsert(entity, updateMode);
 					break;
 				default: exhaustiveGuard(type);
 			}
@@ -1112,57 +1148,77 @@ var MockTableClient = class {
 		}).match(noop, (error) => {
 			throw error;
 		});
-		return {
-			getResponseForEntity: noop,
+		return Promise.resolve({
+			getResponseForEntity: () => void 0,
 			status: 202,
 			subResponses: []
-		};
+		});
+	}
+	updateEntity(entity, mode = "Merge", options) {
+		this.#applyUpdate(entity, mode, options?.etag);
+		return Promise.resolve({
+			date: /* @__PURE__ */ new Date(),
+			etag: this.#getEtag()
+		});
 	}
-	updateEntity(entity, mode = "Merge") {
+	upsertEntity(entity, mode = "Merge") {
+		this.#applyUpsert(entity, mode);
+		return Promise.resolve({
+			date: /* @__PURE__ */ new Date(),
+			etag: this.#getEtag()
+		});
+	}
+	#applyCreate(entity) {
+		const key = this.#getCompositeKey(entity.partitionKey, entity.rowKey);
+		if (this.table.has(key)) throw new MockRestError("The specified entity already exists.", 409);
+		this.table.set(key, {
+			...entity,
+			etag: this.#getEtag()
+		});
+	}
+	#applyDelete(partitionKey, rowKey) {
+		const key = this.#getCompositeKey(partitionKey, rowKey);
+		if (!this.table.has(key)) throw new MockRestError("The specified resource does not exist.", 404);
+		this.table.delete(key);
+	}
+	#applyUpdate(entity, mode = "Merge", etag) {
 		const key = this.#getCompositeKey(entity.partitionKey, entity.rowKey);
 		const existingEntity = this.table.get(key);
 		if (!existingEntity) throw new MockRestError("The specified resource does not exist.", 404);
-		else if (mode === "Merge") return this.#mergeEntity(key, existingEntity, entity);
-		else {
-			this.table.set(key, entity);
-			return Promise.resolve({
-				date: /* @__PURE__ */ new Date(),
-				etag: this.#getEtag()
-			});
-		}
+		else if (etag !== void 0 && etag !== "*" && etag !== existingEntity.etag) throw new MockRestError("The update condition specified in the request was not satisfied.", 412);
+		else if (mode === "Merge") this.table.set(key, {
+			...existingEntity,
+			...entity,
+			etag: this.#getEtag()
+		});
+		else this.table.set(key, {
+			...entity,
+			etag: this.#getEtag()
+		});
 	}
-	upsertEntity(entity, mode = "Merge") {
+	#applyUpsert(entity, mode = "Merge") {
 		const key = this.#getCompositeKey(entity.partitionKey, entity.rowKey);
 		const existingEntity = this.table.get(key);
-		if (existingEntity && mode === "Merge") return this.#mergeEntity(key, existingEntity, entity);
-		else {
-			this.table.set(key, entity);
-			return Promise.resolve({
-				date: /* @__PURE__ */ new Date(),
-				etag: this.#getEtag()
-			});
-		}
+		if (existingEntity && mode === "Merge") this.table.set(key, {
+			...existingEntity,
+			...entity,
+			etag: this.#getEtag()
+		});
+		else this.table.set(key, {
+			...entity,
+			etag: this.#getEtag()
+		});
 	}
 	#getCompositeKey(partitionKey, rowKey) {
 		return `${partitionKey}${ID_SEPARATOR}${rowKey}`;
 	}
 	#getEtag() {
-		return `W/"datetime'${(/* @__PURE__ */ new Date()).toISOString()}'"`;
-	}
-	#mergeEntity(key, entity, entityToMerge) {
-		this.table.set(key, {
-			...entity,
-			...entityToMerge
-		});
-		return Promise.resolve({
-			date: /* @__PURE__ */ new Date(),
-			etag: this.#getEtag()
-		});
+		return `W/"${crypto.randomUUID()}"`;
 	}
 	#withMetadata(entity) {
 		return {
 			...entity,
-			etag: this.#getEtag()
+			etag: entity.etag ?? this.#getEtag()
 		};
 	}
 };
@@ -1179,4 +1235,4 @@ const applyFilter = (documents, clauses) => {
 	const predicate = createFilterPredicate(serializeClauses(clauses));
 	return documents.filter((document) => predicate(document));
 };
-export { MOCK_BLOB_BASE_URL, MOCK_QUEUE_BASE_URL, MOCK_SEARCH_BASE_URL, MOCK_TABLE_BASE_URL, MockBlobBatchClient, MockBlobClient, MockBlockBlobClient, MockContainerClient, MockContainerDatabase, MockEventGridDatabase, MockEventGridPublisherClient, MockQueueClient, MockQueueDatabase, MockRestError, MockSearchClient, MockSearchDatabase, MockServiceBusDatabase, MockServiceBusSender, MockTableClient, MockTableDatabase, MockWebPubSubServiceClient, applyFilter, bodyToBuffer, compare, createFilterPredicate, createMockResponse, getAzureErrorXml, getBlobItemXml, getBlobPrefixXml, getListBlobsXml, getMockSasUrl, isNullClause, isReadableStream, toWebResourceLike };
+export { MOCK_BLOB_BASE_URL, MOCK_BLOB_SEEDED_CREATED_ON, MOCK_QUEUE_BASE_URL, MOCK_SEARCH_BASE_URL, MOCK_TABLE_BASE_URL, MockBlobBatchClient, MockBlobClient, MockBlockBlobClient, MockContainerClient, MockContainerCreatedOnDatabase, MockContainerDatabase, MockEventGridDatabase, MockEventGridPublisherClient, MockQueueClient, MockQueueDatabase, MockRestError, MockSearchClient, MockSearchDatabase, MockServiceBusDatabase, MockServiceBusSender, MockTableClient, MockTableDatabase, MockWebPubSubServiceClient, applyFilter, bodyToBuffer, compare, compareByCompositeKey, createFilterPredicate, createMockResponse, getAzureErrorXml, getBlobItemXml, getBlobPrefixXml, getBlobUrl, getBlobUrlParts, getListBlobsXml, getMockContainerCreatedOnKey, getMockSasUrl, isNullClause, isReadableStream, toWebResourceLike };