diff --git a/packages/server/src/middleware/tests/usageQuota.spec.js b/packages/server/src/middleware/tests/usageQuota.spec.js index d828f2ca60..7a64b1a039 100644 --- a/packages/server/src/middleware/tests/usageQuota.spec.js +++ b/packages/server/src/middleware/tests/usageQuota.spec.js @@ -6,6 +6,9 @@ jest.mock("../../environment", () => ({ isDev: () => true, _set: () => {}, })) +jest.mock("@budibase/auth/tenancy", () => ({ + getTenantId: () => "testing123" +})) const usageQuotaMiddleware = require("../usageQuota") const usageQuota = require("../../utilities/usageQuota") diff --git a/packages/server/src/middleware/usageQuota.js b/packages/server/src/middleware/usageQuota.js index 3a244ef5bc..de54c93200 100644 --- a/packages/server/src/middleware/usageQuota.js +++ b/packages/server/src/middleware/usageQuota.js @@ -1,6 +1,10 @@ const CouchDB = require("../db") const usageQuota = require("../utilities/usageQuota") const env = require("../environment") +const { getTenantId } = require("@budibase/auth/tenancy") + +// tenants without limits +const EXCLUDED_TENANTS = ["bb", "default", "bbtest", "bbstaging"] // currently only counting new writes and deletes const METHOD_MAP = { @@ -28,8 +32,10 @@ function getProperty(url) { } module.exports = async (ctx, next) => { + const tenantId = getTenantId() + // if in development or a self hosted cloud usage quotas should not be executed - if (env.isDev() || env.SELF_HOSTED) { + if (env.isDev() || env.SELF_HOSTED || EXCLUDED_TENANTS.includes(tenantId)) { return next() }