fix: applied changes to test based on comments

microsoft · Jan 17, 2025 · 7e02b32 · 7e02b32
1 parent 19c0ac2
commit 7e02b32
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/packages/runtime/container-runtime/src/test/opLifecycle/remoteMessageProcessor.spec.ts b/packages/runtime/container-runtime/src/test/opLifecycle/remoteMessageProcessor.spec.ts
@@ -23,6 +23,7 @@ import {
 	ensureContentsDeserialized,
 	type IBatch,
 	type InboundMessageResult,
+	OpCompressor,
 	OpDecompressor,
 	OpGroupingManager,
 	OpSplitter,
@@ -136,7 +137,12 @@ describe("RemoteMessageProcessor", () => {
 			let leadingChunkCount = 0;
 			const outboundMessages: IBatchMessage[] = [];
 			if (option.compressionAndChunking.compression) {
-				batch = compressMultipleMessageBatch(batch);
+				if (batch.messages.length === 1) {
+					const compressor = new OpCompressor(mockLogger);
+					batch = compressor.compressBatch(batch);
+				} else {
+					batch = compressMultipleMessageBatch(batch);
+				}
 
 				if (option.compressionAndChunking.chunking) {
 					const splitter = new OpSplitter(