/**
* Test a manager that has one cursor running inside of a session.
*/
TEST_F(CursorManagerTestCustomOpCtx, OneCursorWithASession) {
// Add a cursor with a session to the cursor manager.
auto lsid = makeLogicalSessionIdForTest();
auto opCtx = _queryServiceContext->makeOperationContext(lsid, boost::none);
auto pinned = makeCursor(opCtx.get());
// Retrieve all sessions active in manager - set should contain just lsid.
LogicalSessionIdSet lsids;
useCursorManager()->appendActiveSessions(&lsids);
ASSERT_EQ(lsids.size(), size_t(1));
ASSERT(lsids.find(lsid) != lsids.end());
// Retrieve all cursors for this lsid - should be just ours.
auto cursors = useCursorManager()->getCursorsForSession(lsid);
ASSERT_EQ(cursors.size(), size_t(1));
auto cursorId = pinned.getCursor()->cursorid();
ASSERT(cursors.find(cursorId) != cursors.end());
// Remove the cursor from the manager.
pinned.release();
ASSERT_OK(useCursorManager()->killCursor(opCtx.get(), cursorId, false));
// There should be no more cursor entries by session id.
LogicalSessionIdSet sessions;
useCursorManager()->appendActiveSessions(&sessions);
ASSERT(sessions.empty());
ASSERT(useCursorManager()->getCursorsForSession(lsid).empty());
}
/**
* Test a manager with multiple cursors running inside of different sessions.
*/
TEST_F(CursorManagerTestCustomOpCtx, MultipleCursorsMultipleSessions) {
auto lsid1 = makeLogicalSessionIdForTest();
auto lsid2 = makeLogicalSessionIdForTest();
CursorId cursor1;
CursorId cursor2;
// Cursor with session 1.
{
auto opCtx1 = _queryServiceContext->makeOperationContext(lsid1, boost::none);
cursor1 = makeCursor(opCtx1.get()).getCursor()->cursorid();
}
// Cursor with session 2.
{
auto opCtx2 = _queryServiceContext->makeOperationContext(lsid2, boost::none);
cursor2 = makeCursor(opCtx2.get()).getCursor()->cursorid();
}
// Cursor with no session.
{
auto opCtx3 = _queryServiceContext->makeOperationContext();
makeCursor(opCtx3.get()).getCursor();
}
// Retrieve all sessions - should be both lsids.
LogicalSessionIdSet lsids;
useCursorManager()->appendActiveSessions(&lsids);
ASSERT_EQ(lsids.size(), size_t(2));
ASSERT(lsids.find(lsid1) != lsids.end());
ASSERT(lsids.find(lsid2) != lsids.end());
// Retrieve cursors for each session - should be just one.
auto cursors1 = useCursorManager()->getCursorsForSession(lsid1);
ASSERT_EQ(cursors1.size(), size_t(1));
ASSERT(cursors1.find(cursor1) != cursors1.end());
auto cursors2 = useCursorManager()->getCursorsForSession(lsid2);
ASSERT_EQ(cursors2.size(), size_t(1));
ASSERT(cursors2.find(cursor2) != cursors2.end());
}
