-- Copyright (C) 2026 Intel Corporation
-- SPDX-License-Identifier: PostgreSQL
CREATE EXTENSION svs_cpu_budget_test;
-- No configuration: everyone gets 1 search thread, builds run serial.
SELECT * FROM svs_cpu_budget_test(
    8, 0, 0, 0,
    ARRAY[100,200]::oid[], ARRAY[true,true], ARRAY[-1,-1], ARRAY[0,0],
    ARRAY[]::oid[], ARRAY[]::int4[], ARRAY[]::int4[])
ORDER BY kind, db_oid;
  kind  | db_oid | request_pid | desired | granted | reserved | reserved_floors_exceed_pool 
--------+--------+-------------+---------+---------+----------+-----------------------------
 search |    100 |             |       1 |       1 |        0 | f
 search |    200 |             |       1 |       1 |        0 | f
(2 rows)

-- Demand fits the pool: everyone gets their full ask, no arbitration needed.
SELECT * FROM svs_cpu_budget_test(
    8, 0, 0, 0,
    ARRAY[100,200]::oid[], ARRAY[true,true], ARRAY[3,1], ARRAY[0,0],
    ARRAY[]::oid[], ARRAY[]::int4[], ARRAY[]::int4[])
ORDER BY kind, db_oid;
  kind  | db_oid | request_pid | desired | granted | reserved | reserved_floors_exceed_pool 
--------+--------+-------------+---------+---------+----------+-----------------------------
 search |    100 |             |       3 |       3 |        0 | f
 search |    200 |             |       1 |       1 |        0 | f
(2 rows)

-- Contention: hot_db's floor is honored in full, the remainder is split,
-- and hot_db's own build absorbs the squeeze.
SELECT * FROM svs_cpu_budget_test(
    8, 0, 0, 0,
    ARRAY[100,200]::oid[], ARRAY[true,true], ARRAY[6,1], ARRAY[4,0],
    ARRAY[100]::oid[], ARRAY[555], ARRAY[4])
ORDER BY kind, db_oid;
  kind  | db_oid | request_pid | desired | granted | reserved | reserved_floors_exceed_pool 
--------+--------+-------------+---------+---------+----------+-----------------------------
 build  |    100 |         555 |         |       1 |          | f
 search |    100 |             |       6 |       6 |        4 | f
 search |    200 |             |       1 |       1 |        0 | f
(3 rows)

-- Proportional split of the elastic remainder between two unequal floors,
-- with no builds competing.
SELECT * FROM svs_cpu_budget_test(
    9, 0, 0, 0,
    ARRAY[100,200]::oid[], ARRAY[true,true], ARRAY[8,4], ARRAY[4,2],
    ARRAY[]::oid[], ARRAY[]::int4[], ARRAY[]::int4[])
ORDER BY kind, db_oid;
  kind  | db_oid | request_pid | desired | granted | reserved | reserved_floors_exceed_pool 
--------+--------+-------------+---------+---------+----------+-----------------------------
 search |    100 |             |       8 |       6 |        4 | f
 search |    200 |             |       4 |       3 |        2 | f
(2 rows)

-- Configured floors sum past the pool: the clamp lands on the lowest dbOid
-- first, and the flag comes back set.
SELECT * FROM svs_cpu_budget_test(
    8, 0, 0, 0,
    ARRAY[10,20,30]::oid[], ARRAY[true,true,true], ARRAY[4,4,4], ARRAY[4,4,4],
    ARRAY[]::oid[], ARRAY[]::int4[], ARRAY[]::int4[])
ORDER BY kind, db_oid;
  kind  | db_oid | request_pid | desired | granted | reserved | reserved_floors_exceed_pool 
--------+--------+-------------+---------+---------+----------+-----------------------------
 search |     10 |             |       4 |       0 |        0 | t
 search |     20 |             |       4 |       4 |        4 | t
 search |     30 |             |       4 |       4 |        4 | t
(3 rows)

-- No databases, no builds: no rows.
SELECT * FROM svs_cpu_budget_test(
    8, 0, 0, 0,
    ARRAY[]::oid[], ARRAY[]::boolean[], ARRAY[]::int4[], ARRAY[]::int4[],
    ARRAY[]::oid[], ARRAY[]::int4[], ARRAY[]::int4[])
ORDER BY kind, db_oid;
 kind | db_oid | request_pid | desired | granted | reserved | reserved_floors_exceed_pool 
------+--------+-------------+---------+---------+----------+-----------------------------
(0 rows)

-- A dead/backing-off database contributes nothing, floor or otherwise.
SELECT * FROM svs_cpu_budget_test(
    8, 0, 0, 0,
    ARRAY[100,200]::oid[], ARRAY[false,true], ARRAY[6,1], ARRAY[4,0],
    ARRAY[]::oid[], ARRAY[]::int4[], ARRAY[]::int4[])
ORDER BY kind, db_oid;
  kind  | db_oid | request_pid | desired | granted | reserved | reserved_floors_exceed_pool 
--------+--------+-------------+---------+---------+----------+-----------------------------
 search |    100 |             |       0 |       0 |        0 | f
 search |    200 |             |       1 |       1 |        0 | f
(2 rows)

-- A database with no vamana index holds no search claim, floor or otherwise.
SELECT * FROM svs_cpu_budget_test(
    8, 0, 0, 0,
    ARRAY[100,200]::oid[], ARRAY[true,true], ARRAY[6,1], ARRAY[4,0],
    ARRAY[]::oid[], ARRAY[]::int4[], ARRAY[]::int4[],
    ARRAY[false,true])
ORDER BY kind, db_oid;
  kind  | db_oid | request_pid | desired | granted | reserved | reserved_floors_exceed_pool 
--------+--------+-------------+---------+---------+----------+-----------------------------
 search |    100 |             |       0 |       0 |        0 | f
 search |    200 |             |       1 |       1 |        0 | f
(2 rows)

-- Its first build gets the full pool, as nothing is parked against it.
SELECT * FROM svs_cpu_budget_test(
    8, 0, 0, 0,
    ARRAY[100]::oid[], ARRAY[true], ARRAY[6], ARRAY[4],
    ARRAY[100]::oid[], ARRAY[555], ARRAY[8],
    ARRAY[false])
ORDER BY kind, db_oid;
  kind  | db_oid | request_pid | desired | granted | reserved | reserved_floors_exceed_pool 
--------+--------+-------------+---------+---------+----------+-----------------------------
 build  |    100 |         555 |         |       8 |          | f
 search |    100 |             |       0 |       0 |        0 | f
(2 rows)