diff --git a/mocks/gomock/mocks.go b/mocks/gomock/mocks.go
index 225f4b8..9b38c4c 100644
--- a/mocks/gomock/mocks.go
+++ b/mocks/gomock/mocks.go
@@ -141,6 +141,18 @@ func (mr *MockChangeStreamMockRecorder) ResumeToken() *gomock.Call {
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResumeToken", reflect.TypeOf((*MockChangeStream)(nil).ResumeToken))
 }
 
+// SetBatchSize mocks base method.
+func (m *MockChangeStream) SetBatchSize(arg0 int32) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "SetBatchSize", arg0)
+}
+
+// SetBatchSize indicates an expected call of SetBatchSize.
+func (mr *MockChangeStreamMockRecorder) SetBatchSize(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetBatchSize", reflect.TypeOf((*MockChangeStream)(nil).SetBatchSize), arg0)
+}
+
 // TryNext mocks base method.
 func (m *MockChangeStream) TryNext(arg0 context.Context) bool {
 	m.ctrl.T.Helper()
@@ -983,6 +995,18 @@ func (mr *MockCursorMockRecorder) RemainingBatchLength() *gomock.Call {
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemainingBatchLength", reflect.TypeOf((*MockCursor)(nil).RemainingBatchLength))
 }
 
+// SetBatchSize mocks base method.
+func (m *MockCursor) SetBatchSize(arg0 int32) {
+	m.ctrl.T.Helper()
+	m.ctrl.Call(m, "SetBatchSize", arg0)
+}
+
+// SetBatchSize indicates an expected call of SetBatchSize.
+func (mr *MockCursorMockRecorder) SetBatchSize(arg0 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetBatchSize", reflect.TypeOf((*MockCursor)(nil).SetBatchSize), arg0)
+}
+
 // TryNext mocks base method.
 func (m *MockCursor) TryNext(arg0 context.Context) bool {
 	m.ctrl.T.Helper()
diff --git a/mocks/mockery/change_stream.go b/mocks/mockery/change_stream.go
index 4fe4b65..f226ed2 100644
--- a/mocks/mockery/change_stream.go
+++ b/mocks/mockery/change_stream.go
@@ -117,6 +117,11 @@ func (_m *ChangeStream) ResumeToken() bson.Raw {
 	return r0
 }
 
+// SetBatchSize provides a mock function with given fields: size
+func (_m *ChangeStream) SetBatchSize(size int32) {
+	_m.Called(size)
+}
+
 // TryNext provides a mock function with given fields: ctx
 func (_m *ChangeStream) TryNext(ctx context.Context) bool {
 	ret := _m.Called(ctx)
diff --git a/mocks/mockery/cursor.go b/mocks/mockery/cursor.go
index d4f5a1b..d559d3f 100644
--- a/mocks/mockery/cursor.go
+++ b/mocks/mockery/cursor.go
@@ -129,6 +129,11 @@ func (_m *Cursor) RemainingBatchLength() int {
 	return r0
 }
 
+// SetBatchSize provides a mock function with given fields: batchSize
+func (_m *Cursor) SetBatchSize(batchSize int32) {
+	_m.Called(batchSize)
+}
+
 // TryNext provides a mock function with given fields: ctx
 func (_m *Cursor) TryNext(ctx context.Context) bool {
 	ret := _m.Called(ctx)