test for batchWrite

googleapis · May 31, 2024 · ac03ebb · ac03ebb
1 parent ce3bc03
commit ac03ebb
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/test/database.ts b/test/database.ts
@@ -603,7 +603,6 @@ describe('Database', () => {
     let fakeDataStream: Transform;
     let fakeMutationGroup: FakeTransaction;
     let fakeMutationGroup2: FakeTransaction;
-
     let getSessionStub: sinon.SinonStub;
     let requestStreamStub: sinon.SinonStub;
     let batchWriteStub: sinon.SinonStub;
@@ -614,7 +613,6 @@ describe('Database', () => {
     fakeMutationGroup2 = new FakeTransaction(
       {} as google.spanner.v1.BatchWriteRequest.MutationGroup
     );
-    let fakeStream: Transform;
 
     beforeEach(() => {
       fakePool = database.pool_;
@@ -631,7 +629,7 @@ describe('Database', () => {
 
       batchWriteStub = sandbox
       .stub(fakeMutationGroup, 'batchWrite')
-      .returns(fakeStream);
+      .returns(fakeDataStream);
     });
 
     it('should get a session via `getSession`', () => {
@@ -660,6 +658,14 @@ describe('Database', () => {
         });
     });
 
+    it('should return data when passing a valid list of mutationGroups', done => {
+      database.batchWrite(mutationGroups, fakeOptions)
+        .on('data', data => {
+          console.error('Data:', data);
+          done();
+        });
+    });
+
     it('should call `requestStream` with correct arguments', () => {
       const expectedReqOpts = {
         session: fakeSession ? fakeSession.formattedName_! : undefined,