azure-mock 2.34.1 → 2.35.0

package/dist/index.d.ts

@@ -1,9 +1,10 @@
+import { HttpHeadersLike, WebResourceLike } from "@azure/core-http-compat";
 import { MapValue } from "@esposter/shared";
-import { WebResourceLike } from "@azure/core-http-compat";
 import { PipelineRequest } from "@azure/core-rest-pipeline";
 import { AnonymousCredential, AppendBlobClient, BlobAbortCopyFromURLResponse, BlobBatchClient, BlobBatchDeleteBlobsResponse, BlobBeginCopyFromURLResponse, BlobClient, BlobCopyFromURLResponse, BlobCreateSnapshotResponse, BlobDeleteIfExistsResponse, BlobDeleteImmutabilityPolicyResponse, BlobDeleteOptions, BlobDeleteResponse, BlobDownloadResponseModel, BlobDownloadResponseParsed, BlobGenerateSasUrlOptions, BlobGetAccountInfoResponse, BlobGetPropertiesResponse, BlobGetTagsResponse, BlobItem, BlobLeaseClient, BlobPrefix, BlobSetHTTPHeadersResponse, BlobSetImmutabilityPolicyResponse, BlobSetLegalHoldResponse, BlobSetMetadataResponse, BlobSetTagsResponse, BlobSetTierResponse, BlobUndeleteResponse, BlobUploadCommonResponse, BlockBlobClient, BlockBlobCommitBlockListResponse, BlockBlobGetBlockListResponse, BlockBlobPutBlobFromUrlResponse, BlockBlobStageBlockFromURLResponse, BlockBlobStageBlockResponse, BlockBlobUploadResponse, ContainerClient, ContainerCreateIfNotExistsResponse, ContainerCreateResponse, ContainerDeleteIfExistsResponse, ContainerDeleteResponse, ContainerFindBlobsByTagsSegmentResponse, ContainerGenerateSasUrlOptions, ContainerGetAccessPolicyResponse, ContainerGetAccountInfoResponse, ContainerGetPropertiesResponse, ContainerListBlobFlatSegmentResponse, ContainerListBlobHierarchySegmentResponse, ContainerListBlobsOptions, ContainerSetAccessPolicyResponse, ContainerSetMetadataResponse, FilterBlobItem, HttpRequestBody, PageBlobClient, PollOperationState, PollerLikeWithCancellation, StorageSharedKeyCredential } from "@azure/storage-blob";
 import { BinaryOperator, Clause } from "@esposter/db-schema";
 import { EventGridEvent, EventGridPublisherClient } from "@azure/eventgrid";
+import { ServiceBusMessage, ServiceBusMessageBatch, ServiceBusSender } from "@azure/service-bus";
 import { CreateTableEntityResponse, GetAccessPolicyResponse, GetTableEntityResponse, ListTableEntitiesOptions, TableClient, TableDeleteEntityHeaders, TableEntity, TableEntityResult, TableEntityResultPage, TableMergeEntityHeaders, TableSetAccessPolicyHeaders, TableTransactionResponse, UpdateMode } from "@azure/data-tables";
 import { QueueClearMessagesResponse, QueueClient, QueueCreateIfNotExistsResponse, QueueCreateOptions, QueueCreateResponse, QueueDeleteIfExistsResponse, QueueDeleteMessageResponse, QueueDeleteResponse, QueueGenerateSasUrlOptions, QueueGetAccessPolicyResponse, QueueGetPropertiesResponse, QueueItem, QueuePeekMessagesOptions, QueuePeekMessagesResponse, QueueReceiveMessageOptions, QueueReceiveMessageResponse, QueueSendMessageOptions, QueueSendMessageResponse, QueueServiceProperties, QueueSetAccessPolicyResponse, QueueSetMetadataResponse, QueueUpdateMessageResponse, SignedIdentifier } from "@azure/storage-queue";
 import { AutocompleteResult, CountDocumentsOptions, DeleteDocumentsOptions, GetDocumentOptions, IndexDocumentsBatch, IndexDocumentsOptions, IndexDocumentsResult, MergeDocumentsOptions, MergeOrUploadDocumentsOptions, NarrowedModel, SearchClient, SearchDocumentsResult, SearchOptions, SelectFields, SuggestDocumentsResult, SuggestOptions, UploadDocumentsOptions } from "@azure/search-documents";
@@ -11,6 +12,12 @@ declare const MOCK_BLOB_BASE_URL = "https://mockaccount.blob.core.windows.net";
 declare const MOCK_QUEUE_BASE_URL = "https://mockaccount.queue.core.windows.net";
 declare const MOCK_SEARCH_BASE_URL = "https://mockaccount.search.windows.net";
 declare const MOCK_TABLE_BASE_URL = "https://mockaccount.table.core.windows.net";
+interface MockResponse {
+  headers: HttpHeadersLike;
+  parsedHeaders: Record<string, never>;
+  request: WebResourceLike;
+  status: number;
+}
 declare class MockRestError extends Error {
   statusCode: number;
   constructor(message: string, statusCode: number);
@@ -45,10 +52,15 @@ interface PagedAsyncIterableIterator<TElement, TPage = TElement[], TPageSettings
    */
   next: () => Promise<IteratorResult<TElement>>;
 }
+declare const createMockResponse: (status: number, url?: string) => MockResponse;
+declare const getMockSasUrl: (url: string, permissions: undefined | {
+  toString: () => string;
+}, resourceType?: string) => string;
 declare const MockContainerDatabase: Map<string, Map<string, Buffer>>;
 declare const MockEventGridDatabase: Map<string, EventGridEvent<unknown>[]>;
 declare const MockQueueDatabase: Map<string, string[]>;
 declare const MockSearchDatabase: Map<string, object[]>;
+declare const MockServiceBusDatabase: Map<string, ServiceBusMessage[]>;
 declare const MockTableDatabase: Map<string, Map<string, TableEntity>>;
 type BlobHierarchyItem = (BlobItem & {
   kind: "blob";
@@ -171,8 +183,8 @@ const update2: UpdateOperation<User> = {
 @category Utilities
 */
 type OptionalKeysOf<Type extends object> = Type extends unknown // For distributing `Type`
-? (keyof { [Key in keyof Type as IsOptionalKeyOf<Type, Key> extends false ? never : Key]: never }) & keyof Type // Intersect with `keyof Type` to ensure result of `OptionalKeysOf<Type>` is always assignable to `keyof Type`
-: never;
+ ? (keyof { [Key in keyof Type as IsOptionalKeyOf<Type, Key> extends false ? never : Key]: never; }) & keyof Type // Intersect with `keyof Type` to ensure result of `OptionalKeysOf<Type>` is always assignable to `keyof Type`
+ : never; // Should never happen
 /**
 Extract all required keys from the given type.
 
@@ -204,7 +216,7 @@ const validator3 = createValidation<User>('luckyNumber', value => value > 0);
 @category Utilities
 */
 type RequiredKeysOf<Type extends object> = Type extends unknown // For distributing `Type`
-? Exclude<keyof Type, OptionalKeysOf<Type>> : never;
+ ? Exclude<keyof Type, OptionalKeysOf<Type>> : never; // Should never happen
 /**
 Returns a boolean for whether the given type is `never`.
 
@@ -410,7 +422,7 @@ fn(someInterface as Simplify<SomeInterface>); // Good: transform an `interface`
 @see {@link SimplifyDeep}
 @category Object
 */
-type Simplify<T> = { [KeyType in keyof T]: T[KeyType] } & {};
+type Simplify<T> = { [KeyType in keyof T]: T[KeyType]; } & {};
 /**
 Returns a boolean for whether the two given types are equal.
 
@@ -531,7 +543,7 @@ type ExampleWithoutIndexSignatures = OmitIndexSignature<Example>;
 @see {@link PickIndexSignature}
 @category Object
 */
-type OmitIndexSignature<ObjectType> = { [KeyType in keyof ObjectType as {} extends Record<KeyType, unknown> ? never : KeyType]: ObjectType[KeyType] };
+type OmitIndexSignature<ObjectType> = { [KeyType in keyof ObjectType as {} extends Record<KeyType, unknown> ? never : KeyType]: ObjectType[KeyType]; };
 /**
 Pick only index signatures from the given object type, leaving out all explicitly defined properties.
 
@@ -577,9 +589,9 @@ type ExampleIndexSignature = PickIndexSignature<Example>;
 @see {@link OmitIndexSignature}
 @category Object
 */
-type PickIndexSignature<ObjectType> = { [KeyType in keyof ObjectType as {} extends Record<KeyType, unknown> ? KeyType : never]: ObjectType[KeyType] };
+type PickIndexSignature<ObjectType> = { [KeyType in keyof ObjectType as {} extends Record<KeyType, unknown> ? KeyType : never]: ObjectType[KeyType]; };
 // Merges two objects without worrying about index signatures.
-type SimpleMerge<Destination, Source> = Simplify<{ [Key in keyof Destination as Key extends keyof Source ? never : Key]: Destination[Key] } & Source>;
+type SimpleMerge<Destination, Source> = Simplify<{ [Key in keyof Destination as Key extends keyof Source ? never : Key]: Destination[Key]; } & Source>;
 /**
 Merge two types into a new type. Keys of the second type overrides keys of the first type.
 
@@ -643,10 +655,9 @@ Note: If you want a merge type that more accurately reflects the runtime behavio
 @category Object
 */
 type Merge<Destination, Source> = Destination extends unknown // For distributing `Destination`
-? Source extends unknown // For distributing `Source`
-? If<IsEqual<Destination, Source>, Destination, _Merge<Destination, Source>> : never // Should never happen
-: never;
-// Should never happen
+ ? Source extends unknown // For distributing `Source`
+ ? If<IsEqual<Destination, Source>, Destination, _Merge<Destination, Source>> : never // Should never happen
+ : never; // Should never happen
 type _Merge<Destination, Source> = Simplify<SimpleMerge<PickIndexSignature<Destination>, PickIndexSignature<Source>> & SimpleMerge<OmitIndexSignature<Destination>, OmitIndexSignature<Source>>>;
 /**
 Merges user specified options with default options.
@@ -700,36 +711,9 @@ type Result = ApplyDefaultOptions<PathsOptions, DefaultPathsOptions, SpecifiedOp
 // Types of property 'leavesOnly' are incompatible. Type 'string' is not assignable to type 'boolean'.
 ```
 */
-type ApplyDefaultOptions<Options extends object, Defaults extends Simplify<Omit<Required<Options>, RequiredKeysOf<Options>> & Partial<Record<RequiredKeysOf<Options>, never>>>, SpecifiedOptions extends Options> = _ApplyDefaultOptions<Options, Defaults, SpecifiedOptions> extends infer Result extends Required<Options> // `extends Required<Options>` ensures that `ApplyDefaultOptions<SomeOption, ...>` is always assignable to `Required<SomeOption>`
-? Result : never;
-type _ApplyDefaultOptions<Options, Defaults, SpecifiedOptions> = If<IsAny<SpecifiedOptions>, Defaults, If<IsNever<SpecifiedOptions>, Defaults, Merge<Defaults, { [Key in keyof SpecifiedOptions as undefined extends Required<Options>[Key & keyof Options] ? Key : undefined extends SpecifiedOptions[Key] ? never : Key]: SpecifiedOptions[Key] }>>>;
-/**
-Collapses literal types in a union into their corresponding primitive types, when possible. For example, `CollapseLiterals<'foo' | 'bar' | (string & {})>` returns `string`.
-
-Note: This doesn't collapse literals within tagged types. For example, `CollapseLiterals<Tagged<'foo' | (string & {}), 'Tag'>>` returns `("foo" & Tag<"Tag", never>) | (string & Tag<"Tag", never>)` and not `string & Tag<"Tag", never>`.
-
-Use-case: For collapsing unions created using {@link LiteralUnion}.
-
-@example
-```
-import type {LiteralUnion} from 'type-fest';
-
-type A = CollapseLiterals<'foo' | 'bar' | (string & {})>;
-//=> string
-
-type B = CollapseLiterals<LiteralUnion<1 | 2 | 3, number>>;
-//=> number
-
-type C = CollapseLiterals<LiteralUnion<'onClick' | 'onChange', `on${string}`>>;
-//=> `on${string}`
-
-type D = CollapseLiterals<'click' | 'change' | (`on${string}` & {})>;
-//=> 'click' | 'change' | `on${string}`
-
-type E = CollapseLiterals<LiteralUnion<'foo' | 'bar', string> | null | undefined>;
-//=> string | null | undefined
-```
-*/
+type ApplyDefaultOptions<Options extends object, Defaults extends Simplify<Omit<Required<Options>, RequiredKeysOf<Options>> & Partial<Record<RequiredKeysOf<Options>, never>>>, SpecifiedOptions extends Options> = _ApplyDefaultOptions<Options, Defaults, SpecifiedOptions> extends (infer Result extends Required<Options> // `extends Required<Options>` ensures that `ApplyDefaultOptions<SomeOption, ...>` is always assignable to `Required<SomeOption>`
+) ? Result : never;
+type _ApplyDefaultOptions<Options, Defaults, SpecifiedOptions> = If<IsAny<SpecifiedOptions>, Defaults, If<IsNever<SpecifiedOptions>, Defaults, Merge<Defaults, { [Key in keyof SpecifiedOptions as undefined extends Required<Options>[Key & keyof Options] ? Key : undefined extends SpecifiedOptions[Key] ? never : Key]: SpecifiedOptions[Key]; }>>>;
 /**
 Filter out keys from an object.
 
@@ -760,10 +744,12 @@ type Filtered = Filter<'bar', 'foo'>;
 type Filter<KeyType, ExcludeType> = IsEqual<KeyType, ExcludeType> extends true ? never : (KeyType extends ExcludeType ? never : KeyType);
 type ExceptOptions = {
   /**
-  Disallow assigning non-specified properties.
-  	Note that any omitted properties in the resulting type will be present in autocomplete as `undefined`.
-  	@default false
-  */
+	Disallow assigning non-specified properties.
+
+	Note that any omitted properties in the resulting type will be present in autocomplete as `undefined`.
+
+	@default false
+	*/
   requireExactProps?: boolean;
 };
 type DefaultExceptOptions = {
@@ -825,7 +811,7 @@ type PostPayloadFixed = Except<UserData, 'email'>;
 @category Object
 */
 type Except<ObjectType, KeysType extends keyof ObjectType, Options extends ExceptOptions = {}> = _Except<ObjectType, KeysType, ApplyDefaultOptions<ExceptOptions, DefaultExceptOptions, Options>>;
-type _Except<ObjectType, KeysType extends keyof ObjectType, Options extends Required<ExceptOptions>> = { [KeyType in keyof ObjectType as Filter<KeyType, KeysType>]: ObjectType[KeyType] } & (Options['requireExactProps'] extends true ? Partial<Record<KeysType, never>> : {});
+type _Except<ObjectType, KeysType extends keyof ObjectType, Options extends Required<ExceptOptions>> = { [KeyType in keyof ObjectType as Filter<KeyType, KeysType>]: ObjectType[KeyType]; } & (Options['requireExactProps'] extends true ? Partial<Record<KeysType, never>> : {});
 declare class MockBlobClient implements Except<BlobClient, "accountName"> {
   connectionString: string;
   containerName: string;
@@ -1004,6 +990,26 @@ declare class MockSearchClient<TModel extends object = Record<string, unknown>>
   suggest<TFields extends SelectFields<TModel> = never>(_searchText: string, _suggesterName: string, _options?: SuggestOptions<TModel, TFields>): Promise<SuggestDocumentsResult<TModel, TFields>>;
   uploadDocuments(documents: TModel[], _options?: UploadDocumentsOptions): Promise<IndexDocumentsResult>;
 }
+/**
+ * An in-memory mock of the Azure ServiceBusSender.
+ * It uses a Map to simulate a Service Bus queue and correctly implements the ServiceBusSender interface.
+ *
+ * @example
+ * const mockServiceBusSender = new MockServiceBusSender("hello world");
+ * await mockServiceBusSender.scheduleMessages({ body: "hello world" }, new Date());
+ */
+declare class MockServiceBusSender implements ServiceBusSender {
+  entityPath: string;
+  identifier: string;
+  isClosed: boolean;
+  get queue(): MapValue<typeof MockServiceBusDatabase>;
+  constructor(queueName: string);
+  cancelScheduledMessages(): Promise<void>;
+  close(): Promise<void>;
+  createMessageBatch(): Promise<ServiceBusMessageBatch>;
+  scheduleMessages(messages: ServiceBusMessage | ServiceBusMessage[], scheduledEnqueueTimeUtc: Date): Promise<never[]>;
+  sendMessages(messages: ServiceBusMessage | ServiceBusMessage[] | ServiceBusMessageBatch): Promise<void>;
+}
 /**
  * An in-memory mock of the Azure TableClient.
  * It uses a Map to simulate table storage and correctly implements the TableClient interface.
@@ -1049,10 +1055,7 @@ declare const bodyToBuffer: (body: HttpRequestBody) => Promise<Buffer>;
  * @returns A formatted XML string.
  */
 declare const getAzureErrorXml: (errorCode: string, errorMessage: string) => string;
-declare const getBlobItemXml: ({
-  name,
-  properties
-}: BlobItem) => string;
+declare const getBlobItemXml: ({ name, properties }: BlobItem) => string;
 declare const getBlobPrefixXml: (name: string) => string;
 /**
  * Generates a standard Azure Storage list blobs XML response body.
@@ -1067,4 +1070,4 @@ declare const applyFilter: <T extends Record<string, unknown>>(documents: T[], c
 declare const compare: <T>(operator: BinaryOperator, leftHandSide: T, rightHandSide: null | T) => boolean;
 declare const createFilterPredicate: (filter: string) => ((document: Record<string, unknown>) => boolean);
 declare const isNullClause: (clause: Clause<Record<string, unknown>>) => boolean;
-export { BlobHierarchyItem, 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, MockTableClient, MockTableDatabase, MockWebPubSubServiceClient, PageSettings, PagedAsyncIterableIterator, applyFilter, bodyToBuffer, compare, createFilterPredicate, getAzureErrorXml, getBlobItemXml, getBlobPrefixXml, getListBlobsXml, isNullClause, isReadableStream, toWebResourceLike };
+export { BlobHierarchyItem, 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, MockResponse, MockRestError, MockSearchClient, MockSearchDatabase, MockServiceBusDatabase, MockServiceBusSender, MockTableClient, MockTableDatabase, MockWebPubSubServiceClient, PageSettings, PagedAsyncIterableIterator, applyFilter, bodyToBuffer, compare, createFilterPredicate, createMockResponse, getAzureErrorXml, getBlobItemXml, getBlobPrefixXml, getListBlobsXml, getMockSasUrl, isNullClause, isReadableStream, toWebResourceLike };

package/dist/index.js

@@ -1,10 +1,10 @@
-import { ID_SEPARATOR, InvalidOperationError, Operation, exhaustiveGuard, getResultAsync, html, noop, streamToText, takeOne } from "@esposter/shared";
 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 { AnonymousCredential } from "@azure/storage-blob";
 import { Readable } from "node:stream";
 import { MAX_QUEUE_VISIBILITY_TIMEOUT_MS, deserializeClause, deserializeKey, getTableNullClause, serializeClauses } from "@esposter/db";
-import { BinaryOperator } from "@esposter/db-schema";
+import { AZURE_MAX_PAGE_SIZE, BinaryOperator } from "@esposter/db-schema";
 var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
@@ -38,22 +38,6 @@ var MockRestError = class extends Error {
 		this.statusCode = statusCode;
 	}
 };
-const MockContainerDatabase = /* @__PURE__ */ new Map();
-const MockEventGridDatabase = /* @__PURE__ */ new Map();
-const MockQueueDatabase = /* @__PURE__ */ new Map();
-const MockSearchDatabase = /* @__PURE__ */ new Map();
-const MockTableDatabase = /* @__PURE__ */ new Map();
-/**
-* Generates a standard Azure Storage error XML response body.
-* @param errorCode The official Azure error code (e.g., "BlobNotFound").
-* @param errorMessage The user-friendly error message.
-* @returns A formatted XML string.
-*/
-const getAzureErrorXml = (errorCode, errorMessage) => html`<?xml version="1.0" encoding="utf-8"?>
-    <Error>
-      <code>${errorCode}</code>
-      <Message>${errorMessage}</Message>
-    </Error>`;
 const toWebResourceLike = (request) => ({
 	abortSignal: request.abortSignal,
 	agent: request.agent,
@@ -79,6 +63,30 @@ const toWebResourceLike = (request) => ({
 	validateRequestProperties: noop,
 	withCredentials: request.withCredentials
 });
+const createMockResponse = (status, url = "") => ({
+	headers: toHttpHeadersLike(createHttpHeaders()),
+	parsedHeaders: {},
+	request: toWebResourceLike(createPipelineRequest({ 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 MockEventGridDatabase = /* @__PURE__ */ new Map();
+const MockQueueDatabase = /* @__PURE__ */ new Map();
+const MockSearchDatabase = /* @__PURE__ */ new Map();
+const MockServiceBusDatabase = /* @__PURE__ */ new Map();
+const MockTableDatabase = /* @__PURE__ */ new Map();
+/**
+* Generates a standard Azure Storage error XML response body.
+* @param errorCode The official Azure error code (e.g., "BlobNotFound").
+* @param errorMessage The user-friendly error message.
+* @returns A formatted XML string.
+*/
+const getAzureErrorXml = (errorCode, errorMessage) => html`<?xml version="1.0" encoding="utf-8"?>
+    <Error>
+      <code>${errorCode}</code>
+      <Message>${errorMessage}</Message>
+    </Error>`;
 var MockBlobBatchClient = class {
 	url;
 	constructor(url) {
@@ -111,6 +119,7 @@ var MockBlobBatchClient = class {
 					statusMessage
 				});
 				subResponsesFailedCount++;
+				continue;
 			}
 			const containerName = takeOne(pathSegments);
 			const blobName = pathSegments.slice(1).join("/");
@@ -146,13 +155,11 @@ var MockBlobBatchClient = class {
 		}
 		return Promise.resolve({
 			_response: {
+				...createMockResponse(202, this.url),
 				headers: toHttpHeadersLike(createHttpHeaders({
 					"content-type": "multipart/mixed",
 					"x-ms-request-id": crypto.randomUUID()
-				})),
-				parsedHeaders: {},
-				request: toWebResourceLike(createPipelineRequest({ url: this.url })),
-				status: 202
+				}))
 			},
 			subResponses,
 			subResponsesFailedCount,
@@ -160,12 +167,7 @@ var MockBlobBatchClient = class {
 		});
 	}
 	getContainer(containerName) {
-		let container = MockContainerDatabase.get(containerName);
-		if (!container) {
-			container = /* @__PURE__ */ new Map();
-			MockContainerDatabase.set(containerName, container);
-		}
-		return container;
+		return getOrCreate(MockContainerDatabase, containerName, () => /* @__PURE__ */ new Map());
 	}
 };
 var MockBlobClient = class {
@@ -175,12 +177,7 @@ var MockBlobClient = class {
 	name;
 	url;
 	get container() {
-		let container = MockContainerDatabase.get(this.containerName);
-		if (!container) {
-			container = /* @__PURE__ */ new Map();
-			MockContainerDatabase.set(this.containerName, container);
-		}
-		return container;
+		return getOrCreate(MockContainerDatabase, this.containerName, () => /* @__PURE__ */ new Map());
 	}
 	constructor(connectionString, containerName, blobName) {
 		this.connectionString = connectionString;
@@ -201,12 +198,7 @@ var MockBlobClient = class {
 		const sourceData = sourceContainer.get(sourceBlobName);
 		if (!sourceData) throw new MockRestError("Source blob not found", 404);
 		this.container.set(this.name, Buffer.from(sourceData));
-		const response = { _response: {
-			headers: toHttpHeadersLike(createHttpHeaders()),
-			parsedHeaders: {},
-			request: toWebResourceLike(createPipelineRequest({ url: `${this.url}?comp=copy` })),
-			status: 202
-		} };
+		const response = { _response: createMockResponse(202, `${this.url}?comp=copy`) };
 		return Promise.resolve({
 			cancelOperation: () => Promise.resolve(),
 			getOperationState: () => ({
@@ -228,23 +220,13 @@ 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);
-		return Promise.resolve({ _response: {
-			headers: toHttpHeadersLike(createHttpHeaders()),
-			parsedHeaders: {},
-			request: toWebResourceLike(createPipelineRequest({ url: "" })),
-			status: 200
-		} });
+		return Promise.resolve({ _response: createMockResponse(200) });
 	}
 	deleteIfExists() {
 		const succeeded = this.container.has(this.name);
 		if (succeeded) this.container.delete(this.name);
 		return Promise.resolve({
-			_response: {
-				headers: toHttpHeadersLike(createHttpHeaders()),
-				parsedHeaders: {},
-				request: toWebResourceLike(createPipelineRequest({ url: "" })),
-				status: succeeded ? 200 : 404
-			},
+			_response: createMockResponse(succeeded ? 200 : 404),
 			succeeded
 		});
 	}
@@ -254,12 +236,7 @@ var MockBlobClient = class {
 	download() {
 		const buffer = this.container.get(this.name);
 		return Promise.resolve({
-			_response: {
-				headers: toHttpHeadersLike(createHttpHeaders()),
-				parsedHeaders: {},
-				request: toWebResourceLike(createPipelineRequest({ url: "" })),
-				status: buffer ? 200 : 404
-			},
+			_response: createMockResponse(buffer ? 200 : 404),
 			readableStreamBody: buffer ? Readable.from(buffer) : void 0
 		});
 	}
@@ -278,8 +255,7 @@ var MockBlobClient = class {
 		throw new Error("Method not implemented.");
 	}
 	generateSasUrl(options) {
-		const sp = options.permissions?.toString() ?? "r";
-		return Promise.resolve(`${MOCK_BLOB_BASE_URL}/${this.containerName}/${this.name}?sv=2025-11-05&sr=b&sig=mock-signature&st=1970-01-01T00:00:00Z&se=2099-12-31T23:59:59Z&sp=${sp}`);
+		return Promise.resolve(getMockSasUrl(this.url, options.permissions, "b"));
 	}
 	generateUserDelegationSasStringToSign() {
 		throw new Error("Method not implemented.");
@@ -388,12 +364,7 @@ var MockBlockBlobClient = class extends MockBlobClient {
 	}
 	async upload(body, _contentLength) {
 		this.container.set(this.name, await bodyToBuffer(body));
-		return { _response: {
-			headers: toHttpHeadersLike(createHttpHeaders()),
-			parsedHeaders: {},
-			request: toWebResourceLike(createPipelineRequest({ url: "" })),
-			status: 201
-		} };
+		return { _response: createMockResponse(201) };
 	}
 	uploadBrowserData() {
 		throw new Error("Method not implemented.");
@@ -448,12 +419,7 @@ var MockContainerClient = class {
 	credential = new AnonymousCredential();
 	url;
 	get container() {
-		let container = MockContainerDatabase.get(this.containerName);
-		if (!container) {
-			container = /* @__PURE__ */ new Map();
-			MockContainerDatabase.set(this.containerName, container);
-		}
-		return container;
+		return getOrCreate(MockContainerDatabase, this.containerName, () => /* @__PURE__ */ new Map());
 	}
 	constructor(connectionString, containerName) {
 		this.connectionString = connectionString;
@@ -472,12 +438,7 @@ 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: {
-			headers: toHttpHeadersLike(createHttpHeaders()),
-			parsedHeaders: {},
-			request: toWebResourceLike(createPipelineRequest({ url: `${this.url}/${blobName}` })),
-			status: 200
-		} });
+		return Promise.resolve({ _response: createMockResponse(200, `${this.url}/${blobName}`) });
 	}
 	deleteIfExists() {
 		throw new Error("Method not implemented.");
@@ -492,8 +453,7 @@ var MockContainerClient = class {
 		throw new Error("Method not implemented.");
 	}
 	generateSasUrl(options) {
-		const sp = options.permissions?.toString() ?? "r";
-		return Promise.resolve(`${MOCK_BLOB_BASE_URL}/${this.containerName}?sv=2025-11-05&sr=c&sig=mock-signature&st=1970-01-01T00:00:00Z&se=2099-12-31T23:59:59Z&sp=${sp}`);
+		return Promise.resolve(getMockSasUrl(this.url, options.permissions, "c"));
 	}
 	generateUserDelegationSasStringToSign() {
 		throw new Error("Method not implemented.");
@@ -545,18 +505,15 @@ var MockContainerClient = class {
 				}
 				if (allBlobItems.length > 0 || allBlobPrefixes.length > 0) yield await Promise.resolve({
 					_response: {
+						...createMockResponse(200),
 						bodyAsText: getListBlobsXml(this.containerName, `${allBlobItemXml.join("")}${allBlobPrefixXml.join("")}`),
-						headers: toHttpHeadersLike(createHttpHeaders()),
 						parsedBody: {
 							containerName: this.containerName,
 							marker: "",
 							prefix: options?.prefix ?? "",
 							segment: { blobItems: allBlobItems },
 							serviceEndpoint: ""
-						},
-						parsedHeaders: {},
-						request: toWebResourceLike(createPipelineRequest({ url: "" })),
-						status: 200
+						}
 					},
 					containerName: this.containerName,
 					marker: "",
@@ -583,18 +540,15 @@ var MockContainerClient = class {
 				}
 				if (allBlobItems.length > 0) yield await Promise.resolve({
 					_response: {
+						...createMockResponse(200),
 						bodyAsText: getListBlobsXml(this.containerName, allBlobItemXml.join("")),
-						headers: toHttpHeadersLike(createHttpHeaders()),
 						parsedBody: {
 							containerName: this.containerName,
 							marker: "",
 							prefix: "",
 							segment: { blobItems: allBlobItems },
 							serviceEndpoint: ""
-						},
-						parsedHeaders: {},
-						request: toWebResourceLike(createPipelineRequest({ url: "" })),
-						status: 200
+						}
 					},
 					containerName: this.containerName,
 					marker: "",
@@ -690,9 +644,7 @@ var MockEventGridPublisherClient = class {
 		this.inputSchema = topicType;
 	}
 	send(newEvents) {
-		const events = MockEventGridDatabase.get(this.endpointUrl) ?? [];
-		events.push(...newEvents);
-		MockEventGridDatabase.set(this.endpointUrl, events);
+		getOrCreate(MockEventGridDatabase, this.endpointUrl, () => []).push(...newEvents);
 		return Promise.resolve();
 	}
 };
@@ -711,12 +663,7 @@ var MockQueueClient = class {
 	name;
 	url;
 	get queue() {
-		let messages = MockQueueDatabase.get(this.name);
-		if (!messages) {
-			messages = [];
-			MockQueueDatabase.set(this.name, messages);
-		}
-		return messages;
+		return getOrCreate(MockQueueDatabase, this.name, () => []);
 	}
 	constructor(connectionString, queueName) {
 		this.connectionString = connectionString;
@@ -748,8 +695,7 @@ var MockQueueClient = class {
 		throw new Error("Method not implemented.");
 	}
 	generateSasUrl(options) {
-		const sp = options.permissions?.toString() ?? "r";
-		return `${MOCK_QUEUE_BASE_URL}/${this.name}?sv=2025-11-05&sig=mock-signature&st=1970-01-01T00:00:00Z&se=2099-12-31T23:59:59Z&sp=${sp}`;
+		return getMockSasUrl(this.url, options.permissions);
 	}
 	generateUserDelegationSasUrl() {
 		throw new Error("Method not implemented.");
@@ -779,12 +725,9 @@ var MockQueueClient = class {
 		}));
 		return Promise.resolve({
 			_response: {
+				...createMockResponse(200, this.url),
 				bodyAsText: "",
-				headers: toHttpHeadersLike(createHttpHeaders()),
-				parsedBody: peekedMessageItems,
-				parsedHeaders: {},
-				request: toWebResourceLike(createPipelineRequest({ url: this.url })),
-				status: 200
+				parsedBody: peekedMessageItems
 			},
 			peekedMessageItems
 		});
@@ -801,12 +744,9 @@ var MockQueueClient = class {
 		}));
 		return Promise.resolve({
 			_response: {
+				...createMockResponse(200, this.url),
 				bodyAsText: "",
-				headers: toHttpHeadersLike(createHttpHeaders()),
-				parsedBody: receivedMessageItems,
-				parsedHeaders: {},
-				request: toWebResourceLike(createPipelineRequest({ url: this.url })),
-				status: 200
+				parsedBody: receivedMessageItems
 			},
 			receivedMessageItems
 		});
@@ -828,12 +768,9 @@ var MockQueueClient = class {
 		}];
 		return Promise.resolve({
 			_response: {
+				...createMockResponse(200, this.url),
 				bodyAsText: "",
-				headers: toHttpHeadersLike(createHttpHeaders()),
-				parsedBody: enqueuedMessages,
-				parsedHeaders: {},
-				request: toWebResourceLike(createPipelineRequest({ url: this.url })),
-				status: 200
+				parsedBody: enqueuedMessages
 			},
 			expiresOn,
 			insertedOn,
@@ -973,12 +910,7 @@ var MockSearchClient = class {
 	indexName;
 	serviceVersion = "";
 	get documents() {
-		let documents = MockSearchDatabase.get(this.indexName);
-		if (!documents) {
-			documents = [];
-			MockSearchDatabase.set(this.indexName, documents);
-		}
-		return documents;
+		return getOrCreate(MockSearchDatabase, this.indexName, () => []);
 	}
 	constructor(indexName) {
 		this.indexName = indexName;
@@ -1031,6 +963,48 @@ var MockSearchClient = class {
 	}
 };
 /**
+* An in-memory mock of the Azure ServiceBusSender.
+* It uses a Map to simulate a Service Bus queue and correctly implements the ServiceBusSender interface.
+*
+* @example
+* const mockServiceBusSender = new MockServiceBusSender("hello world");
+* await mockServiceBusSender.scheduleMessages({ body: "hello world" }, new Date());
+*/
+var MockServiceBusSender = class {
+	entityPath;
+	identifier;
+	isClosed = false;
+	get queue() {
+		return getOrCreate(MockServiceBusDatabase, this.entityPath, () => []);
+	}
+	constructor(queueName) {
+		this.entityPath = queueName;
+		this.identifier = queueName;
+	}
+	cancelScheduledMessages() {
+		throw new Error("Method not implemented.");
+	}
+	close() {
+		this.isClosed = true;
+		return Promise.resolve();
+	}
+	createMessageBatch() {
+		throw new Error("Method not implemented.");
+	}
+	scheduleMessages(messages, scheduledEnqueueTimeUtc) {
+		for (const message of Array.isArray(messages) ? messages : [messages]) this.queue.push({
+			...message,
+			scheduledEnqueueTimeUtc
+		});
+		return Promise.resolve([]);
+	}
+	sendMessages(messages) {
+		if (!Array.isArray(messages) && "tryAddMessage" in messages) throw new Error("Method not implemented.");
+		this.queue.push(...Array.isArray(messages) ? messages : [messages]);
+		return Promise.resolve();
+	}
+};
+/**
 * An in-memory mock of the Azure TableClient.
 * It uses a Map to simulate table storage and correctly implements the TableClient interface.
 *
@@ -1043,12 +1017,7 @@ var MockTableClient = class {
 	tableName;
 	url;
 	get table() {
-		let table = MockTableDatabase.get(this.tableName);
-		if (!table) {
-			table = /* @__PURE__ */ new Map();
-			MockTableDatabase.set(this.tableName, table);
-		}
-		return table;
+		return getOrCreate(MockTableDatabase, this.tableName, () => /* @__PURE__ */ new Map());
 	}
 	constructor(_url, tableName) {
 		this.tableName = tableName;
@@ -1096,6 +1065,7 @@ var MockTableClient = class {
 		return {
 			byPage: ({ maxPageSize } = {}) => (async function* (entities) {
 				if (maxPageSize !== void 0 && maxPageSize <= 0) throw new RangeError("maxPageSize must be greater than 0.");
+				else if (maxPageSize !== void 0 && maxPageSize > AZURE_MAX_PAGE_SIZE) throw new MockRestError("One of the request inputs is not valid.", 400);
 				const allEntitiesWithMetadata = entities.map((e) => withMetadata(e));
 				if (allEntitiesWithMetadata.length === 0) return;
 				else if (!maxPageSize) {
@@ -1209,4 +1179,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, MockTableClient, MockTableDatabase, MockWebPubSubServiceClient, applyFilter, bodyToBuffer, compare, createFilterPredicate, getAzureErrorXml, getBlobItemXml, getBlobPrefixXml, getListBlobsXml, isNullClause, isReadableStream, toWebResourceLike };
+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 };