transformEventsForSend test should use buffered produce client

jeremymeng · Jan 27, 2022 · 019f6c5 · 019f6c5
1 parent 3ae020b
commit 019f6c5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sdk/eventhub/event-hubs/test/internal/transformEventsForSend.spec.ts b/sdk/eventhub/event-hubs/test/internal/transformEventsForSend.spec.ts
@@ -6,7 +6,7 @@ import chai from "chai";
 import {
   EventData,
   EventDataBatch,
-  EventHubProducerClient,
+  EventHubBufferedProducerClient,
   PartitionPublishingProperties
 } from "../../src";
 import { EnvVarKeys, getEnvVars } from "../public/utils/testUtils";
@@ -47,7 +47,7 @@ describe("transformEventsForSend", function() {
     let batch: EventDataBatch;
 
     beforeEach("Populate EventDataBatch", async () => {
-      const producerClient = new EventHubProducerClient(service.connectionString, service.path);
+      const producerClient = new EventHubBufferedProducerClient(service.connectionString, service.path);
       batch = await producerClient.createBatch();
 
       for (let i = 1; i <= 10; i++) {
@@ -95,7 +95,7 @@ describe("transformEventsForSend", function() {
     let batch: EventDataBatch;
 
     beforeEach("Populate EventDataBatch", async () => {
-      const producerClient = new EventHubProducerClient(service.connectionString, service.path, {
+      const producerClient = new EventHubBufferedProducerClient(service.connectionString, service.path, {
         enableIdempotentPartitions: true
       });
       batch = await producerClient.createBatch({ partitionId: "0" });