fix(NODE-7160): migrate node-mongodb-native/test/unit/assorted tests

Author: PavelSafronov

test/unit/assorted/sessions_client.test.js

@@ -0,0 +1,122 @@
+import { expect } from 'chai';
+
+import { isHello, MongoClient } from '../../mongodb';
+import { ReplSetFixture } from '../../tools/common';
+import * as mock from '../../tools/mongodb-mock/index';
+import { type MockServer } from '../../tools/mongodb-mock/src/server';
+
+const test: { server: MockServer } = { server: undefined };
+describe('Sessions - client/unit', function () {
+  describe('Client', function () {
+    afterEach(async () => await mock.cleanup());
+
+    beforeEach(async () => {
+      const server: MockServer = await mock.createServer();
+      test.server = server;
+    });
+
+    it('should not throw a synchronous exception if sessions are not supported', async function () {
+      test.server.setMessageHandler(request => {
+        const doc = request.document;
+        if (isHello(doc)) {
+          request.reply(Object.assign({}, mock.HELLO));
+        } else if (doc.endSessions) {
+          request.reply({ ok: 1 });
+        }
+      });
+
+      const client = new MongoClient(`mongodb://${test.server.uri()}/test`);
+      await client.connect();
+      expect(() => client.startSession()).to.not.throw(
+        'Current topology does not support sessions'
+      );
+      await client.close();
+    });
+
+    it('should throw an exception if sessions are not supported on some servers', async function () {
+      const replicaSetMock = new ReplSetFixture();
+      let testClient;
+      try {
+        await replicaSetMock.setup({ doNotInitHandlers: true });
+        replicaSetMock.firstSecondaryServer.setMessageHandler(request => {
+          const doc = request.document;
+          if (isHello(doc)) {
+            const hello = replicaSetMock.firstSecondaryStates[0];
+            hello.logicalSessionTimeoutMinutes = 20;
+            request.reply(hello);
+          } else if (doc.endSessions) {
+            request.reply({ ok: 1 });
+          }
+        });
+
+        replicaSetMock.secondSecondaryServer.setMessageHandler(request => {
+          const doc = request.document;
+          if (isHello(doc)) {
+            const hello = replicaSetMock.secondSecondaryStates[0];
+            hello.logicalSessionTimeoutMinutes = 10;
+            request.reply(hello);
+          } else if (doc.endSessions) {
+            request.reply({ ok: 1 });
+          }
+        });
+
+        replicaSetMock.arbiterServer.setMessageHandler(request => {
+          const doc = request.document;
+          if (isHello(doc)) {
+            const hello = replicaSetMock.arbiterStates[0];
+            hello.logicalSessionTimeoutMinutes = 30;
+            request.reply(hello);
+          } else if (doc.endSessions) {
+            request.reply({ ok: 1 });
+          }
+        });
+
+        replicaSetMock.primaryServer.setMessageHandler(request => {
+          const doc = request.document;
+          if (isHello(doc)) {
+            const hello = replicaSetMock.primaryStates[0];
+            hello.logicalSessionTimeoutMinutes = null;
+            request.reply(hello);
+          } else if (doc.endSessions) {
+            request.reply({ ok: 1 });
+          }
+        });
+
+        const uri = replicaSetMock.uri();
+        testClient = new MongoClient(uri);
+        const client = await testClient.connect();
+        const session = client.startSession();
+        await client.db().collection('t').insertOne({ a: 1 }, { session });
+        expect.fail('Expected an error to be thrown about not supporting sessions');
+      } catch (error) {
+        expect(error.message).to.equal('Current topology does not support sessions');
+      } finally {
+        if (testClient) {
+          await testClient.close();
+        }
+      }
+    });
+
+    it('should return a client session when requested if the topology supports it', async function () {
+      test.server.setMessageHandler(request => {
+        const doc = request.document;
+        if (isHello(doc)) {
+          request.reply(
+            Object.assign({}, mock.HELLO, {
+              logicalSessionTimeoutMinutes: 10
+            })
+          );
+        } else if (doc.endSessions) {
+          request.reply({ ok: 1 });
+        }
+      });
+
+      const client = new MongoClient(`mongodb://${test.server.uri()}/test`);
+      await client.connect();
+      const session = client.startSession();
+      expect(session).to.exist;
+      session.endSession();
+      await client.close();
+    });
+  });
+});