query_rate_limit_test.js

import http from "k6/http";
import { check, sleep } from "k6";

/*
  Test goal:
  - Simulate multiple users
  - Each user has its own rate-limit bucket
  - Validate 200 vs 429 behavior
*/

export const options = {
  scenarios: {
    multi_user_test: {
      executor: "constant-vus",
      vus: 20,          // 20 concurrent users
      duration: "15s",  // run for 15 seconds
    },
  },
};

export default function () {
  // Each VU acts like a unique user
  const userId = `user-${__VU}`;

  const url = "http://127.0.0.1:8000/products/?category=electronics";

  const params = {
    headers: {
      "X-User-Id": userId,
    },
  };

  const res = http.get(url, params);

  // Validate responses
  check(res, {
    "status is 200 or 429": (r) =>
      r.status === 200 || r.status === 429,

    "rate limit header exists": (r) =>
      r.headers["X-RateLimit-Remaining"] !== undefined ||
      r.status === 429,
  });

  // Small delay to allow token refill
  sleep(0.2);
}