Tests moved to /test, routes logic improved

Author: gkoos

README.md

@@ -100,6 +100,7 @@ Chaos Proxy uses Koa Router for path matching, supporting named parameters (e.g.
 - There is no inheritance between global and route-specific middleware.
 - Global middlewares apply to every request.
 - Route middlewares only apply to requests matching that route.
+- Routes can be defined with or without HTTP methods. If a method is specified (e.g., `GET /path`), the rule only applies to that method. If no method is specified (e.g., `/path`), the rule applies to all methods for that path.
 - If a request matches a route, only the middlewares for that route (plus global) are applied. Route rules do not inherit or merge from parent routes or wildcards.
 - If multiple routes match, the most specific one is chosen (e.g., `/users/:id` over `/users/*`).
 - If no route matches, only global middlewares are applied.
@@ -210,6 +211,26 @@ This configuration adds a `foo: 'bar'` property to every JSON request body befor
 **Note:**
 For maximum flexibility, the `transform` option in `bodyTransform` can be specified as a JavaScript function string in your YAML config. This allows you to define custom transformation logic directly in the config file. Be aware that evaluating JS from config can introduce security and syntax risks. Use with care and only in trusted environments.
 
+
+If you call `startServer` programmatically, you can also pass a real function instead of a string:
+
+```ts
+import { startServer, bodyTransform } from 'chaos-proxy';
+
+startServer({
+  target: 'http://localhost:4000',
+  port: 5000,
+  global: [
+    bodyTransform({
+      transform: (body, ctx) => {
+        body.foo = 'bar';
+        return body;
+      }
+    })
+  ]
+});
+```
+
 ---
 
 ## Extensibility

src/cli.ts

@@ -1,8 +1,6 @@
 #!/usr/bin/env node
 
-import { loadConfig, startServer, registerBuiltins } from './index';
-
-registerBuiltins();
+import { loadConfig, startServer } from './index';
 
 const args = process.argv.slice(2);
 let configPath = 'chaos.yaml';

src/index.ts

@@ -5,4 +5,4 @@ import { registerBuiltins } from './registry/builtin';
 
 registerBuiltins();
 
-export { loadConfig, startServer, registerMiddleware, registerBuiltins };
+export { loadConfig, startServer, registerMiddleware };

src/registry/builtin.ts

@@ -8,6 +8,8 @@ import { dropConnection } from '../middlewares/dropConnection';
 import { fail } from '../middlewares/fail';
 import { rateLimit } from '../middlewares/rateLimit';
 import type { RateLimitOptions } from '../middlewares/rateLimit';
+import { throttle } from '../middlewares/throttle';
+import type { ThrottleOptions } from '../middlewares/throttle';
 
 export function registerBuiltins() {
   // Register built-in middleware primitives
@@ -24,4 +26,5 @@ export function registerBuiltins() {
     cors(opts as { origin?: string; methods?: string; headers?: string })
   );
   registerMiddleware('rateLimit', (opts) => rateLimit(opts as unknown as RateLimitOptions));
+  registerMiddleware('throttle', (opts) => throttle(opts as unknown as ThrottleOptions));
 }

src/server.ts

@@ -42,7 +42,11 @@ export function startServer(config: ChaosConfig, options?: { verbose?: boolean }
         throw new Error(`Unsupported HTTP method: ${method}`);
       }
     } else {
-      router.use(routeKey, ...middlewares);
+      // Mount for all HTTP methods if no method is specified
+      const methods = ['get', 'post', 'put', 'delete', 'patch', 'head', 'options'];
+      for (const method of methods) {
+        (router[method as keyof typeof router] as (path: string, ...middleware: Array<Koa.Middleware>) => Router)(routeKey, ...middlewares);
+      }
     }
   }
   app.use(router.routes());

src/cli.test.ts renamed to test/cli.test.ts

@@ -1,7 +1,7 @@
 import { describe, it, expect } from 'vitest';
-import { loadConfig } from './config/loader';
-import { startServer } from './server';
-import { parseConfig } from './config/parser';
+import { loadConfig } from '../src/config/loader';
+import { startServer } from '../src/server';
+import { parseConfig } from '../src/config/parser';
 // No longer needed: import express from 'express';
 
 describe('chaos-proxy CLI logic', () => {

src/config/loader.test.ts renamed to test/config/loader.test.ts

@@ -1,5 +1,5 @@
 import { describe, it, expect } from 'vitest';
-import { loadConfig } from './loader';
+import { loadConfig } from '../../src/config/loader';
 
 describe('loadConfig', () => {
   it('throws if config file is missing', () => {

src/config/parser.test.ts renamed to test/config/parser.test.ts

@@ -1,9 +1,9 @@
 import { describe, it, expect } from 'vitest';
-import { parseConfig, resolveConfigMiddlewares } from './parser';
+import { parseConfig, resolveConfigMiddlewares } from '../../src/config/parser';
 // ...existing code...
 // ...existing code...
 // ...existing code...
-import type { ChaosConfig } from './loader';
+import type { ChaosConfig } from '../../src/config/loader';
 
 describe('parseConfig', () => {
   it('parses a valid config with default port', () => {

src/middlewares/bodyTransform.test.ts renamed to test/middlewares/bodyTransform.test.ts

@@ -1,6 +1,6 @@
 
 import { describe, it, expect } from 'vitest';
-import { bodyTransform } from './bodyTransform';
+import { bodyTransform } from '../../src/middlewares/bodyTransform';
 import type { Context } from 'koa';
 
 function createMockCtx(body: unknown, contentType = 'application/json'): Context {

src/middlewares/cors.test.ts renamed to test/middlewares/cors.test.ts

@@ -1,5 +1,5 @@
 import { describe, it, expect, vi } from 'vitest';
-import { cors } from './cors';
+import { cors } from '../../src/middlewares/cors';
 import type { Context } from 'koa';
 
 describe('cors middleware', () => {