test: fix session pinning assertions

mongodb · Jul 29, 2021 · 453fc4c · 453fc4c
1 parent 6b928bc
commit 453fc4c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/functional/unified-spec-runner/operations.ts b/test/functional/unified-spec-runner/operations.ts
@@ -127,12 +127,12 @@ operations.set('assertSessionNotDirty', async ({ entities, operation }) => {
 });
 
 operations.set('assertSessionPinned', async ({ entities, operation }) => {
-  const session = entities.getEntity('session', operation.arguments.session);
+  const session = operation.arguments.session;
   expect(session.isPinned, 'session should be pinned').to.be.true;
 });
 
 operations.set('assertSessionUnpinned', async ({ entities, operation }) => {
-  const session = entities.getEntity('session', operation.arguments.session);
+  const session = operation.arguments.session;
   expect(session.isPinned, 'session should be unpinned').to.be.false;
 });