fix(tenant): include pagination options in middlewares (#17835)

tenant/middleware_bucket_logging.go

@@ -83,7 +83,7 @@ func (l *BucketLogger) FindBuckets(ctx context.Context, filter influxdb.BucketFi
 		}
 		l.logger.Debug("buckets find", dur)
 	}(time.Now())
-	return l.bucketService.FindBuckets(ctx, filter)
+	return l.bucketService.FindBuckets(ctx, filter, opt...)
 }
 
 func (l *BucketLogger) UpdateBucket(ctx context.Context, id influxdb.ID, upd influxdb.BucketUpdate) (u *influxdb.Bucket, err error) {

tenant/middleware_bucket_logging_test.go

@@ -0,0 +1,19 @@
+package tenant_test
+
+import (
+	"testing"
+
+	"github.com/influxdata/influxdb/v2"
+	"github.com/influxdata/influxdb/v2/tenant"
+	influxdbtesting "github.com/influxdata/influxdb/v2/testing"
+	"go.uber.org/zap/zaptest"
+)
+
+func TestBucketLoggingService(t *testing.T) {
+	influxdbtesting.BucketService(initBoltBucketLoggingService, t, influxdbtesting.WithoutHooks())
+}
+
+func initBoltBucketLoggingService(f influxdbtesting.BucketFields, t *testing.T) (influxdb.BucketService, string, func()) {
+	svc, s, closer := initBoltBucketService(f, t)
+	return tenant.NewBucketLogger(zaptest.NewLogger(t), svc), s, closer
+}

tenant/middleware_org_logging.go

@@ -71,7 +71,7 @@ func (l *OrgLogger) FindOrganizations(ctx context.Context, filter influxdb.Organ
 		}
 		l.logger.Debug("orgs find", dur)
 	}(time.Now())
-	return l.orgService.FindOrganizations(ctx, filter)
+	return l.orgService.FindOrganizations(ctx, filter, opt...)
 }
 
 func (l *OrgLogger) UpdateOrganization(ctx context.Context, id influxdb.ID, upd influxdb.OrganizationUpdate) (u *influxdb.Organization, err error) {

tenant/middleware_org_logging_test.go

@@ -0,0 +1,19 @@
+package tenant_test
+
+import (
+	"testing"
+
+	"github.com/influxdata/influxdb/v2"
+	"github.com/influxdata/influxdb/v2/tenant"
+	influxdbtesting "github.com/influxdata/influxdb/v2/testing"
+	"go.uber.org/zap/zaptest"
+)
+
+func TestOrganizationLoggingService(t *testing.T) {
+	influxdbtesting.OrganizationService(initBoltOrganizationLoggingService, t)
+}
+
+func initBoltOrganizationLoggingService(f influxdbtesting.OrganizationFields, t *testing.T) (influxdb.OrganizationService, string, func()) {
+	orgSvc, s, closer := initBoltOrganizationService(f, t)
+	return tenant.NewOrgLogger(zaptest.NewLogger(t), orgSvc), s, closer
+}

tenant/middleware_urm_logging.go

@@ -45,7 +45,7 @@ func (l *URMLogger) FindUserResourceMappings(ctx context.Context, filter influxd
 		}
 		l.logger.Debug("urm find", dur)
 	}(time.Now())
-	return l.urmService.FindUserResourceMappings(ctx, filter)
+	return l.urmService.FindUserResourceMappings(ctx, filter, opt...)
 }
 
 func (l *URMLogger) DeleteUserResourceMapping(ctx context.Context, resourceID, userID influxdb.ID) (err error) {

tenant/middleware_urm_logging_test.go

@@ -0,0 +1,19 @@
+package tenant_test
+
+import (
+	"testing"
+
+	"github.com/influxdata/influxdb/v2"
+	"github.com/influxdata/influxdb/v2/tenant"
+	influxdbtesting "github.com/influxdata/influxdb/v2/testing"
+	"go.uber.org/zap/zaptest"
+)
+
+func TestUserResourceMappingLoggingService(t *testing.T) {
+	influxdbtesting.UserResourceMappingService(initBoltUserResourceMappingLoggingService, t)
+}
+
+func initBoltUserResourceMappingLoggingService(f influxdbtesting.UserResourceFields, t *testing.T) (influxdb.UserResourceMappingService, func()) {
+	svc, closer := initBoltUserResourceMappingService(f, t)
+	return tenant.NewURMLogger(zaptest.NewLogger(t), svc), closer
+}

tenant/middleware_user_logging.go

@@ -71,7 +71,7 @@ func (l *UserLogger) FindUsers(ctx context.Context, filter influxdb.UserFilter,
 		}
 		l.logger.Debug("users find", dur)
 	}(time.Now())
-	return l.userService.FindUsers(ctx, filter)
+	return l.userService.FindUsers(ctx, filter, opt...)
 }
 
 func (l *UserLogger) UpdateUser(ctx context.Context, id influxdb.ID, upd influxdb.UserUpdate) (u *influxdb.User, err error) {

tenant/middleware_user_logging_test.go

@@ -0,0 +1,19 @@
+package tenant_test
+
+import (
+	"testing"
+
+	"github.com/influxdata/influxdb/v2"
+	"github.com/influxdata/influxdb/v2/tenant"
+	influxdbtesting "github.com/influxdata/influxdb/v2/testing"
+	"go.uber.org/zap/zaptest"
+)
+
+func TestUserLoggingService(t *testing.T) {
+	influxdbtesting.UserService(initBoltUserLoggingService, t)
+}
+
+func initBoltUserLoggingService(f influxdbtesting.UserFields, t *testing.T) (influxdb.UserService, string, func()) {
+	svc, s, closer := initBoltUserService(f, t)
+	return tenant.NewUserLogger(zaptest.NewLogger(t), svc), s, closer
+}