Register rate limiter component in convex tests

Rhys Sullivan committed 8 months ago

Commit 8536cf21 parent eac4b6c6 files · +106 −4Browse files

Unified · Split

6 files

 import { api } from "./_generated/api";
 import type { Id } from "./_generated/dataModel.d.ts";
 import schema from "./schema";
+import { registerRateLimiterComponent } from "./testHelpers";
 // ---------------------------------------------------------------------------
 // Helpers
 // ---------------------------------------------------------------------------
 function setup() {
-  return convexTest(schema, {
+  const t = convexTest(schema, {
     "./workspace.ts": () => import("./workspace"),
     "./workspaces.ts": () => import("./workspaces"),
     "./organizations.ts": () => import("./organizations"),
     "./toolRegistry.ts": () => import("./toolRegistry"),
     "./_generated/api.js": () => import("./_generated/api.js"),
   });
+  registerRateLimiterComponent(t);
+  return t;
 }
 /**

 import { internal } from "./_generated/api";
 import type { Id } from "./_generated/dataModel.d.ts";
 import schema from "./schema";
+import { registerRateLimiterComponent } from "./testHelpers";
 function setup() {
-  return convexTest(schema, {
+  const t = convexTest(schema, {
     "./database.ts": () => import("./database"),
     "./toolRegistry.ts": () => import("./toolRegistry"),
     "./_generated/api.js": () => import("./_generated/api.js"),
   });
+  registerRateLimiterComponent(t);
+  return t;
 }
 /** Insert a minimal workspace row and return its Id. */

 import { convexTest } from "convex-test";
 import { internal } from "./_generated/api";
 import schema from "./schema";
+import { registerRateLimiterComponent } from "./testHelpers";
 const GITHUB_OPENAPI_SPEC_URL =
   "https://raw.githubusercontent.com/github/rest-api-description/main/descriptions/api.github.com/api.github.com.json?convex_test_profile=github";
 function setup() {
-  return convexTest(schema, {
+  const t = convexTest(schema, {
     "./database.ts": () => import("./database"),
     "./executorNode.ts": () => import("./executorNode"),
     "./workspaceAuthInternal.ts": () => import("./workspaceAuthInternal"),
     "./runtimeNode.ts": () => import("./runtimeNode"),
     "./_generated/api.js": () => import("./_generated/api.js"),
   });
+  registerRateLimiterComponent(t);
+  return t;
 }
 test("convex-test keeps GitHub inventory build warm-cache fast", async () => {

 import { convexTest } from "convex-test";
 import { internal } from "./_generated/api";
 import schema from "./schema";
+import { registerRateLimiterComponent } from "./testHelpers";
 function setup() {
-  return convexTest(schema, {
+  const t = convexTest(schema, {
     "./openApiSpecCache.ts": () => import("./openApiSpecCache"),
     "./toolRegistry.ts": () => import("./toolRegistry"),
     "./_generated/api.js": () => import("./_generated/api.js"),
   });
+  registerRateLimiterComponent(t);
+  return t;
 }
 describe("openApiSpecCache table operations", () => {

@@ -0,0 +1,14 @@
	1	import type { TestConvex } from "convex-test";
	2	import { defineSchema } from "convex/server";
	3	import type { GenericSchema, SchemaDefinition } from "convex/server";
	4
	5	const rateLimiterSchema = defineSchema({});
	6
	7	const rateLimiterModules = {
	8	"./lib.ts": () => import("../src/testing/rate_limiter_component"),
	9	"./_generated/api.js": () => import("./_generated/api.js"),
	10	};
	11
	12	export function registerRateLimiterComponent(t: TestConvex<SchemaDefinition<GenericSchema, boolean>>): void {
	13	t.registerComponent("rateLimiter", rateLimiterSchema, rateLimiterModules);
	14	}

@@ -0,0 +1,72 @@
	1	import { mutationGeneric, queryGeneric } from "convex/server";
	2	import { v } from "convex/values";
	3
	4	const limitArgs = {
	5	name: v.string(),
	6	config: v.any(),
	7	key: v.optional(v.string()),
	8	count: v.optional(v.number()),
	9	reserve: v.optional(v.boolean()),
	10	throws: v.optional(v.boolean()),
	11	};
	12
	13	const limitResult = v.object({
	14	ok: v.boolean(),
	15	retryAfter: v.optional(v.number()),
	16	});
	17
	18	export const rateLimit = mutationGeneric({
	19	args: limitArgs,
	20	returns: limitResult,
	21	handler: async () => ({ ok: true }),
	22	});
	23
	24	export const checkRateLimit = queryGeneric({
	25	args: limitArgs,
	26	returns: limitResult,
	27	handler: async () => ({ ok: true }),
	28	});
	29
	30	export const resetRateLimit = mutationGeneric({
	31	args: {
	32	name: v.string(),
	33	key: v.optional(v.string()),
	34	},
	35	returns: v.null(),
	36	handler: async () => null,
	37	});
	38
	39	export const getValue = queryGeneric({
	40	args: {
	41	name: v.string(),
	42	config: v.any(),
	43	key: v.optional(v.string()),
	44	sampleShards: v.optional(v.number()),
	45	},
	46	returns: v.object({
	47	config: v.any(),
	48	shard: v.number(),
	49	ts: v.number(),
	50	value: v.number(),
	51	}),
	52	handler: async (_ctx, args) => ({
	53	config: args.config,
	54	shard: 0,
	55	ts: Date.now(),
	56	value: 0,
	57	}),
	58	});
	59
	60	export const clearAll = mutationGeneric({
	61	args: {
	62	before: v.optional(v.number()),
	63	},
	64	returns: v.null(),
	65	handler: async () => null,
	66	});
	67
	68	export const getServerTime = mutationGeneric({
	69	args: {},
	70	returns: v.number(),
	71	handler: async () => Date.now(),
	72	});