test: add adjustment to also check tracker string

nestjs · Dec 8, 2023 · b5f8240 · b5f8240
1 parent 41b7482
commit b5f8240
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/function-overrides/app.module.ts b/test/function-overrides/app.module.ts
@@ -3,6 +3,7 @@ import { APP_GUARD } from '@nestjs/core';
 import { ThrottlerGuard, ThrottlerModule, seconds } from '../../src';
 import { FunctionOverridesThrottlerController } from './funciton-overrides-throttler.controller';
 import md5 = require('md5');
+import assert = require('assert');
 
 @Module({
   imports: [
@@ -16,8 +17,10 @@ import md5 = require('md5');
         name: 'custom',
         ttl: seconds(3),
         limit: 2,
-        getTracker: (req) => req.ip,
+        getTracker: () => 'customTrackerString',
         generateKey: (context: ExecutionContext, trackerString: string, throttlerName: string) => {
+          // check if tracker string is passed correctly
+          assert(trackerString === 'customTrackerString');
           // use the same key for all endpoints
           return md5(`${throttlerName}-${trackerString}`);
         },