LOAD 'pg_diffix';
SET pg_diffix.noise_layer_sd = 7;
SET pg_diffix.low_count_layer_sd = 2;
SET pg_diffix.low_count_min_threshold = 2;
SET ROLE diffix_test;
SET pg_diffix.session_access_level = 'anonymized_trusted';
----------------------------------------------------------------
-- Sanity checks
----------------------------------------------------------------
SELECT diffix.access_level();
    access_level    
--------------------
 anonymized_trusted
(1 row)

----------------------------------------------------------------
-- Basic queries
----------------------------------------------------------------
SELECT COUNT(*) FROM test_customers;
 count 
-------
     4
(1 row)

SELECT COUNT(*) FROM test_purchases;
 count 
-------
    23
(1 row)

SELECT COUNT(city), COUNT(DISTINCT city) FROM test_customers;
 count | count 
-------+-------
     4 |     1
(1 row)

SELECT COUNT(DISTINCT cid) FROM test_purchases;
 count 
-------
     9
(1 row)

SELECT city, COUNT(DISTINCT id) FROM test_customers GROUP BY 1;
 city | count 
------+-------
 *    |    16
 Rome |     8
(2 rows)

----------------------------------------------------------------
-- Basic queries - expanding constants in target expressions
----------------------------------------------------------------
SELECT 1 FROM test_patients;
 ?column? 
----------
        1
        1
        1
        1
(4 rows)

SELECT cast(1 as real) FROM test_patients;
 float4 
--------
      1
      1
      1
      1
(4 rows)

SELECT 1, COUNT(*) FROM test_patients;
 ?column? | count 
----------+-------
        1 |     4
(1 row)

SELECT 1, city FROM test_customers;
 ?column? | city 
----------+------
        1 | *
        1 | *
        1 | *
        1 | *
        1 | *
        1 | *
        1 | *
        1 | *
        1 | *
        1 | *
        1 | *
        1 | *
        1 | *
        1 | *
        1 | *
        1 | *
        1 | Rome
        1 | Rome
        1 | Rome
        1 | Rome
        1 | Rome
        1 | Rome
        1 | Rome
        1 | Rome
(24 rows)

SELECT city, 'aaaa' FROM test_customers;
 city | ?column? 
------+----------
 *    | aaaa
 *    | aaaa
 *    | aaaa
 *    | aaaa
 *    | aaaa
 *    | aaaa
 *    | aaaa
 *    | aaaa
 *    | aaaa
 *    | aaaa
 *    | aaaa
 *    | aaaa
 *    | aaaa
 *    | aaaa
 *    | aaaa
 *    | aaaa
 Rome | aaaa
 Rome | aaaa
 Rome | aaaa
 Rome | aaaa
 Rome | aaaa
 Rome | aaaa
 Rome | aaaa
 Rome | aaaa
(24 rows)

----------------------------------------------------------------
-- Multi-AID queries
----------------------------------------------------------------
SELECT city FROM test_patients GROUP BY 1;
 city 
------
 *
(1 row)

SELECT COUNT(*), COUNT(city), COUNT(DISTINCT city) FROM test_patients;
 count | count | count 
-------+-------+-------
     4 |     4 |     0
(1 row)

----------------------------------------------------------------
-- LCF & Filtering
----------------------------------------------------------------
SELECT id FROM test_customers;
 id 
----
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
(16 rows)

SELECT city FROM test_customers;
 city 
------
 *
 *
 *
 *
 *
 *
 *
 *
 *
 *
 *
 *
 *
 *
 *
 *
 Rome
 Rome
 Rome
 Rome
 Rome
 Rome
 Rome
 Rome
(24 rows)

SELECT city FROM test_customers GROUP BY 1 HAVING length(city) <> 4;
 city 
------
 *
(1 row)

SELECT COUNT(*), COUNT(city), COUNT(DISTINCT city) FROM london_customers;
 count | count | count 
-------+-------+-------
     0 |     0 |     0
(1 row)

-- LCF doesn't depend on the bucket seed, both queries should have same noisy threshold.
SELECT diffix.floor_by(age, 30), COUNT(*) FROM test_patients GROUP BY 1;
 floor_by | count 
----------+-------
          |    23
(1 row)

SELECT diffix.floor_by(age, 30), diffix.floor_by(age, 106), COUNT(*) FROM test_patients GROUP BY 1, 2;
 floor_by | floor_by | count 
----------+----------+-------
          |          |     9
(1 row)

----------------------------------------------------------------
-- Empty tables
----------------------------------------------------------------
SELECT COUNT(*), COUNT(city), COUNT(DISTINCT city) FROM empty_test_customers;
 count | count | count 
-------+-------+-------
     0 |     0 |     0
(1 row)