stack/apps/e2e/tests/backend/performance/mock-metric-events.sql

BEGIN;

CREATE EXTENSION IF NOT EXISTS "pgcrypto";

-- Spread existing perf users across the past 30 days
WITH tenancy AS (
  SELECT id
  FROM "Tenancy"
  WHERE "projectId" = 'internal'
    AND "branchId" = 'main'
    AND "organizationId" IS NULL
  LIMIT 1
),
ranked AS (
  SELECT
    "tenancyId",
    "projectUserId",
    ROW_NUMBER() OVER (ORDER BY "createdAt", "projectUserId") - 1 AS rn
  FROM "ProjectUser"
  WHERE "tenancyId" = (SELECT id FROM tenancy)
)
UPDATE "ProjectUser" pu
SET "createdAt" = (
      (current_date - ((ranked.rn % 30) * INTERVAL '1 day'))::timestamp
      + ((ranked.rn % 12) * INTERVAL '1 hour')
      + ((ranked.rn * 7 % 60) * INTERVAL '1 minute')
    ),
    "updatedAt" = NOW()
FROM ranked
WHERE pu."tenancyId" = ranked."tenancyId"
  AND pu."projectUserId" = ranked."projectUserId";

-- Add auth-method diversity (passwords already exist)
WITH tenancy AS (
  SELECT id
  FROM "Tenancy"
  WHERE "projectId" = 'internal'
    AND "branchId" = 'main'
    AND "organizationId" IS NULL
  LIMIT 1
),
ordered_users AS (
  SELECT
    "tenancyId",
    "projectUserId",
    ROW_NUMBER() OVER (ORDER BY "createdAt" DESC, "projectUserId") AS rn
  FROM "ProjectUser"
  WHERE "tenancyId" = (SELECT id FROM tenancy)
),
passkey_candidates AS (
  SELECT ou."tenancyId", ou."projectUserId", ou.rn, gen_random_uuid() AS auth_method_id
  FROM ordered_users ou
  LEFT JOIN "PasskeyAuthMethod" existing
    ON existing."tenancyId" = ou."tenancyId"
   AND existing."projectUserId" = ou."projectUserId"
  WHERE ou.rn <= 40
    AND existing."projectUserId" IS NULL
),
otp_candidates AS (
  SELECT ou."tenancyId", ou."projectUserId", ou.rn, gen_random_uuid() AS auth_method_id
  FROM ordered_users ou
  LEFT JOIN "OtpAuthMethod" existing
    ON existing."tenancyId" = ou."tenancyId"
   AND existing."projectUserId" = ou."projectUserId"
  WHERE ou.rn > 40 AND ou.rn <= 120
    AND existing."projectUserId" IS NULL
),
oauth_candidates AS (
  SELECT
    ou."tenancyId",
    ou."projectUserId",
    ou.rn,
    gen_random_uuid() AS auth_method_id,
    CASE ((ou.rn - 121) % 3)
      WHEN 0 THEN 'github'
      WHEN 1 THEN 'google'
      ELSE 'microsoft'
    END AS provider_id,
    'acct-' || lpad(ou.rn::text, 4, '0') AS provider_account_id,
    'oauth-user-' || lpad(ou.rn::text, 4, '0') || '@internal.stack' AS email
  FROM ordered_users ou
  LEFT JOIN "OAuthAuthMethod" existing
    ON existing."tenancyId" = ou."tenancyId"
   AND existing."projectUserId" = ou."projectUserId"
  WHERE ou.rn > 120 AND ou.rn <= 240
    AND existing."projectUserId" IS NULL
),
insert_passkey_auth_methods AS (
  INSERT INTO "AuthMethod" ("tenancyId","id","projectUserId","createdAt","updatedAt")
  SELECT
    "tenancyId",
    auth_method_id,
    "projectUserId",
    NOW() - ((rn % 15) * INTERVAL '1 day'),
    NOW()
  FROM passkey_candidates
  RETURNING "tenancyId","id","projectUserId","createdAt"
),
insert_passkeys AS (
  INSERT INTO "PasskeyAuthMethod"
    ("tenancyId","authMethodId","projectUserId","createdAt","updatedAt",
     "credentialId","publicKey","userHandle","transports","credentialDeviceType","counter")
  SELECT
    p."tenancyId",
    p."id",
    p."projectUserId",
    p."createdAt",
    p."createdAt",
    'cred-' || LPAD((ROW_NUMBER() OVER (ORDER BY p."projectUserId"))::text, 4, '0'),
    encode(gen_random_bytes(24), 'base64'),
    encode(gen_random_bytes(16), 'hex'),
    ARRAY['internal','hybrid'],
    'multiDevice',
    1 + (ROW_NUMBER() OVER (ORDER BY p."projectUserId") % 100)
  FROM insert_passkey_auth_methods p
  RETURNING 1
),
insert_otp_auth_methods AS (
  INSERT INTO "AuthMethod" ("tenancyId","id","projectUserId","createdAt","updatedAt")
  SELECT
    "tenancyId",
    auth_method_id,
    "projectUserId",
    NOW() - ((rn % 10) * INTERVAL '1 day'),
    NOW()
  FROM otp_candidates
  RETURNING "tenancyId","id","projectUserId","createdAt"
),
insert_otp_methods AS (
  INSERT INTO "OtpAuthMethod" ("tenancyId","authMethodId","projectUserId","createdAt","updatedAt")
  SELECT
    "tenancyId",
    "id",
    "projectUserId",
    "createdAt",
    "createdAt"
  FROM insert_otp_auth_methods
  RETURNING 1
),
insert_oauth_auth_methods AS (
  INSERT INTO "AuthMethod" ("tenancyId","id","projectUserId","createdAt","updatedAt")
  SELECT
    "tenancyId",
    auth_method_id,
    "projectUserId",
    NOW() - ((rn % 8) * INTERVAL '1 day'),
    NOW()
  FROM oauth_candidates
  RETURNING "tenancyId","id","projectUserId","createdAt"
),
insert_oauth_accounts AS (
  INSERT INTO "ProjectUserOAuthAccount"
    ("tenancyId","id","projectUserId","configOAuthProviderId","providerAccountId",
     "email","allowConnectedAccounts","allowSignIn","createdAt","updatedAt")
  SELECT
    oc."tenancyId",
    gen_random_uuid(),
    oc."projectUserId",
    oc.provider_id,
    oc.provider_account_id,
    oc.email,
    true,
    true,
    NOW() - ((oc.rn % 8) * INTERVAL '1 day'),
    NOW()
  FROM oauth_candidates oc
  ON CONFLICT ("tenancyId","configOAuthProviderId","projectUserId","providerAccountId") DO NOTHING
),
insert_oauth_methods AS (
  INSERT INTO "OAuthAuthMethod"
    ("tenancyId","authMethodId","configOAuthProviderId","providerAccountId","projectUserId","createdAt","updatedAt")
  SELECT
    oc."tenancyId",
    oc.auth_method_id,
    oc.provider_id,
    oc.provider_account_id,
    oc."projectUserId",
    NOW(),
    NOW()
  FROM oauth_candidates oc
  ON CONFLICT DO NOTHING
  RETURNING 1
)
SELECT
  (SELECT COUNT(*) FROM insert_passkeys) AS passkeys_created,
  (SELECT COUNT(*) FROM insert_otp_methods) AS otp_created,
  (SELECT COUNT(*) FROM insert_oauth_methods) AS oauth_created;

-- Insert user-activity events across 100 countries, 20 per user
WITH run_meta AS (
  SELECT gen_random_uuid()::text AS seed_run_id
),
tenancy AS (
  SELECT id
  FROM "Tenancy"
  WHERE "projectId" = 'internal'
    AND "branchId" = 'main'
    AND "organizationId" IS NULL
  LIMIT 1
),
country_list AS (
  SELECT country_code, ordinality
  FROM unnest(ARRAY[
    'US','IN','BR','DE','GB','ID','FR','CA','VN','AU','PK','HK','NL','JP','ES','NG','BD','PH','KE','ZA',
    'TH','TR','SG','IT','NO','CH','PL','MX','SE','PT','AR','EG','MY','NP','AE','TW','LK','KR','MA','RO',
    'CO','DK','UA','SA','GH','IL','CN','TN','BE','CL','AT','ET','CZ','DZ','RS','NZ','IE','FI','PE','RU',
    'UG','GR','CM','HU','UZ','IQ','RW','EE','KZ','KH','SK','GE','AO','HR','SN','SI','LV','JO','EC','LB',
    'CG','VE','PA','UY','TZ','BG','LT','LU','ZW','DO','BJ','BO','BY','MG','MW','XK','CI','IR','GT','MQ'
  ]) WITH ORDINALITY AS t(country_code, ordinality)
),
country_data AS (
  SELECT
    country_code,
    ordinality,
    'Region-' || country_code AS region_code,
    'City-' || country_code AS city_name,
    (-60 + ((ordinality * 7) % 120))::double precision AS latitude,
    (((ordinality * 13) % 360) - 180)::double precision AS longitude,
    'TZ/' || country_code AS tz_identifier
  FROM country_list
),
activity_users AS (
  SELECT
    "tenancyId",
    "projectUserId",
    ROW_NUMBER() OVER (ORDER BY "createdAt" DESC, "projectUserId") AS rn
  FROM "ProjectUser"
  WHERE "tenancyId" = (SELECT id FROM tenancy)
  LIMIT 300
),
event_matrix AS (
  SELECT
    gen_random_uuid() AS event_id,
    gen_random_uuid() AS ip_info_id,
    au."tenancyId",
    au."projectUserId",
    au.rn,
    occ AS occurrence_index,
    (
      (current_date - ((au.rn + occ) % 30) * INTERVAL '1 day')::timestamp
      + (((au.rn + occ) % 24) + 1) * INTERVAL '1 hour'
      + ((au.rn * 11 + occ * 17) % 60) * INTERVAL '1 minute'
    ) AS started_at,
    cd.country_code,
    cd.region_code,
    cd.city_name,
    cd.latitude,
    cd.longitude,
    cd.tz_identifier,
    ((au.rn + occ) % 500) AS geo_variant
  FROM activity_users au
  CROSS JOIN generate_series(0, 999) AS occ
  JOIN country_data cd ON cd.ordinality = ((au.rn + occ) % 100) + 1
),
insert_ip_info AS (
  INSERT INTO "EventIpInfo"
    ("id","ip","countryCode","regionCode","cityName","latitude","longitude","tzIdentifier","createdAt","updatedAt")
  SELECT
    em.ip_info_id,
    FORMAT('%s.%s.%s.%s',
      10 + (em.geo_variant % 200),
      1 + ((em.rn + em.geo_variant) % 200),
      1 + ((em.occurrence_index * 3 + em.geo_variant) % 200),
      1 + ((em.rn * 11 + em.occurrence_index + em.geo_variant) % 200)
    ),
    em.country_code,
    em.region_code,
    em.city_name,
    em.latitude,
    em.longitude,
    em.tz_identifier,
    em.started_at,
    em.started_at
  FROM event_matrix em
),
insert_events AS (
  INSERT INTO "Event"
    ("id","createdAt","updatedAt","isWide","eventStartedAt","eventEndedAt",
     "systemEventTypeIds","data","endUserIpInfoGuessId","isEndUserIpInfoGuessTrusted")
  SELECT
    em.event_id,
    em.started_at,
    em.started_at + INTERVAL '5 minutes',
    false,
    em.started_at,
    em.started_at + INTERVAL '5 minutes',
    ARRAY['$user-activity'],
    jsonb_build_object(
      'userId', em."projectUserId"::text,
      'projectId', 'internal',
      'branchId', 'main',
      'isAnonymous', 'false',
      'seedTag', 'perf-metrics-mock',
      'seedRunId', rm.seed_run_id,
      'countryCode', em.country_code
    ),
    em.ip_info_id,
    ((em.rn + em.occurrence_index) % 2 = 0)
  FROM event_matrix em
  CROSS JOIN run_meta rm
  RETURNING 1
)
SELECT COUNT(*) AS events_created FROM insert_events;

COMMIT;