-- ########## ID STATIC TESTS ##########

\set ON_ERROR_ROLLBACK 1
\set ON_ERROR_STOP true

BEGIN;
SELECT set_config('search_path','partman, tap',false);

SELECT plan(73);
CREATE SCHEMA partman_test;
CREATE ROLE partman_basic;
CREATE ROLE partman_revoke;
CREATE ROLE partman_owner;

CREATE TABLE partman_test.id_static_table (col1 int primary key, col2 text, col3 timestamptz DEFAULT now());
INSERT INTO partman_test.id_static_table (col1) VALUES (generate_series(1,9));
GRANT SELECT,INSERT,UPDATE ON partman_test.id_static_table TO partman_basic;
GRANT ALL ON partman_test.id_static_table TO partman_revoke;

SELECT create_parent('partman_test.id_static_table', 'col1', 'id-static', '10');
SELECT has_table('partman_test', 'id_static_table_p0', 'Check id_static_table_p0 exists');
SELECT has_table('partman_test', 'id_static_table_p10', 'Check id_static_table_p10 exists');
SELECT has_table('partman_test', 'id_static_table_p20', 'Check id_static_table_p20 exists');
SELECT has_table('partman_test', 'id_static_table_p30', 'Check id_static_table_p30 exists');
SELECT has_table('partman_test', 'id_static_table_p40', 'Check id_static_table_p40 exists');
SELECT hasnt_table('partman_test', 'id_static_table_p50', 'Check id_static_table_p50 doesn''t exists yet');
SELECT col_is_pk('partman_test', 'id_static_table_p0', ARRAY['col1'], 'Check for primary key in id_static_table_p0');
SELECT col_is_pk('partman_test', 'id_static_table_p10', ARRAY['col1'], 'Check for primary key in id_static_table_p10');
SELECT col_is_pk('partman_test', 'id_static_table_p20', ARRAY['col1'], 'Check for primary key in id_static_table_p20');
SELECT col_is_pk('partman_test', 'id_static_table_p30', ARRAY['col1'], 'Check for primary key in id_static_table_p30');
SELECT col_is_pk('partman_test', 'id_static_table_p40', ARRAY['col1'], 'Check for primary key in id_static_table_p40');
SELECT table_privs_are('partman_test', 'id_static_table_p0', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p0');
SELECT table_privs_are('partman_test', 'id_static_table_p10', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p10');
SELECT table_privs_are('partman_test', 'id_static_table_p20', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p20');
SELECT table_privs_are('partman_test', 'id_static_table_p30', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p30');
SELECT table_privs_are('partman_test', 'id_static_table_p40', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p40');
SELECT table_privs_are('partman_test', 'id_static_table_p0', 'partman_revoke', ARRAY['SELECT', 'INSERT', 'UPDATE', 'DELETE', 'TRUNCATE', 'REFERENCES', 'TRIGGER'], 'Check partman_revoke privileges of id_static_table_p0');
SELECT table_privs_are('partman_test', 'id_static_table_p10', 'partman_revoke', ARRAY['SELECT', 'INSERT', 'UPDATE', 'DELETE', 'TRUNCATE', 'REFERENCES', 'TRIGGER'], 'Check partman_revoke privileges of id_static_table_p10');
SELECT table_privs_are('partman_test', 'id_static_table_p20', 'partman_revoke', ARRAY['SELECT', 'INSERT', 'UPDATE', 'DELETE', 'TRUNCATE', 'REFERENCES', 'TRIGGER'], 'Check partman_revoke privileges of id_static_table_p20');
SELECT table_privs_are('partman_test', 'id_static_table_p30', 'partman_revoke', ARRAY['SELECT', 'INSERT', 'UPDATE', 'DELETE', 'TRUNCATE', 'REFERENCES', 'TRIGGER'], 'Check partman_revoke privileges of id_static_table_p30');
SELECT table_privs_are('partman_test', 'id_static_table_p40', 'partman_revoke', ARRAY['SELECT', 'INSERT', 'UPDATE', 'DELETE', 'TRUNCATE', 'REFERENCES', 'TRIGGER'], 'Check partman_revoke privileges of id_static_table_p40');

SELECT create_prev_id_partition('partman_test.id_static_table');
SELECT is_empty('SELECT * FROM ONLY partman_test.id_static_table', 'Check that parent table has had data moved to partition');
SELECT results_eq('SELECT count(*)::int FROM partman_test.id_static_table', ARRAY[9], 'Check count from parent table');
SELECT results_eq('SELECT count(*)::int FROM partman_test.id_static_table_p0', ARRAY[9], 'Check count from id_static_table_p0');

REVOKE INSERT, UPDATE, DELETE, TRUNCATE, REFERENCES, TRIGGER ON partman_test.id_static_table FROM partman_revoke;
INSERT INTO partman_test.id_static_table (col1) VALUES (generate_series(10,25));

SELECT is_empty('SELECT * FROM ONLY partman_test.id_static_table', 'Check that parent table has had no data inserted to it');
SELECT results_eq('SELECT count(*)::int FROM partman_test.id_static_table_p10', ARRAY[10], 'Check count from id_static_table_p10');
SELECT results_eq('SELECT count(*)::int FROM partman_test.id_static_table_p20', ARRAY[6], 'Check count from id_static_table_p20');

SELECT has_table('partman_test', 'id_static_table_p50', 'Check id_static_table_p50 exists');
SELECT hasnt_table('partman_test', 'id_static_table_p60', 'Check id_static_table_p60 doesn''t exists yet');
SELECT col_is_pk('partman_test', 'id_static_table_p50', ARRAY['col1'], 'Check for primary key in id_static_table_p50');
SELECT table_privs_are('partman_test', 'id_static_table_p0', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p0');
SELECT table_privs_are('partman_test', 'id_static_table_p10', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p10');
SELECT table_privs_are('partman_test', 'id_static_table_p20', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p20');
SELECT table_privs_are('partman_test', 'id_static_table_p30', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p30');
SELECT table_privs_are('partman_test', 'id_static_table_p40', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p40');
SELECT table_privs_are('partman_test', 'id_static_table_p50', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p50');
SELECT table_privs_are('partman_test', 'id_static_table_p50', 'partman_revoke', ARRAY['SELECT'], 'Check partman_revoke privileges of id_static_table_p50');

GRANT DELETE ON partman_test.id_static_table TO partman_basic;
REVOKE ALL ON partman_test.id_static_table FROM partman_revoke;
ALTER TABLE partman_test.id_static_table OWNER TO partman_owner;
INSERT INTO partman_test.id_static_table (col1) VALUES (generate_series(26,38));

SELECT is_empty('SELECT * FROM ONLY partman_test.id_static_table', 'Check that parent table has had no data inserted to it');
SELECT results_eq('SELECT count(*)::int FROM partman_test.id_static_table_p20', ARRAY[10], 'Check count from id_static_table_p20');
SELECT results_eq('SELECT count(*)::int FROM partman_test.id_static_table_p30', ARRAY[9], 'Check count from id_static_table_p30');

SELECT has_table('partman_test', 'id_static_table_p60', 'Check id_static_table_p60 exists');
SELECT has_table('partman_test', 'id_static_table_p70', 'Check id_static_table_p70 exists');
SELECT hasnt_table('partman_test', 'id_static_table_p80', 'Check id_static_table_p80 doesn''t exists yet');
SELECT col_is_pk('partman_test', 'id_static_table_p60', ARRAY['col1'], 'Check for primary key in id_static_table_p60');
SELECT col_is_pk('partman_test', 'id_static_table_p70', ARRAY['col1'], 'Check for primary key in id_static_table_p70');
SELECT table_privs_are('partman_test', 'id_static_table_p0', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p0');
SELECT table_privs_are('partman_test', 'id_static_table_p10', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p10');
SELECT table_privs_are('partman_test', 'id_static_table_p20', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p20');
SELECT table_privs_are('partman_test', 'id_static_table_p30', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p30');
SELECT table_privs_are('partman_test', 'id_static_table_p40', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p40');
SELECT table_privs_are('partman_test', 'id_static_table_p50', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE'], 'Check partman_basic privileges of id_static_table_p50');
SELECT table_privs_are('partman_test', 'id_static_table_p50', 'partman_revoke', ARRAY['SELECT'], 'Check partman_revoke privileges of id_static_table_p50');
SELECT table_privs_are('partman_test', 'id_static_table_p60', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE','DELETE'], 'Check partman_basic privileges of id_static_table_p60');
SELECT table_privs_are('partman_test', 'id_static_table_p70', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE','DELETE'], 'Check partman_basic privileges of id_static_table_p70');
-- Currently unable to test that all privileges have been revoked. Sent in request to pgtap developer.
SELECT table_owner_is ('partman_test', 'id_static_table_p60', 'partman_owner', 'Check that ownership change worked for id_static_table_p60');
SELECT table_owner_is ('partman_test', 'id_static_table_p70', 'partman_owner', 'Check that ownership change worked for id_static_table_p70');

INSERT INTO partman_test.id_static_table (col1) VALUES (generate_series(200,210));
SELECT results_eq('SELECT count(*)::int FROM ONLY partman_test.id_static_table', ARRAY[11], 'Check that data outside trigger scope goes to parent');

SELECT reapply_privileges('partman_test.id_static_table');
SELECT table_privs_are('partman_test', 'id_static_table_p0', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE','DELETE'], 'Check partman_basic privileges of id_static_table_p0');
SELECT table_privs_are('partman_test', 'id_static_table_p10', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE','DELETE'], 'Check partman_basic privileges of id_static_table_p10');
SELECT table_privs_are('partman_test', 'id_static_table_p20', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE','DELETE'], 'Check partman_basic privileges of id_static_table_p20');
SELECT table_privs_are('partman_test', 'id_static_table_p30', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE','DELETE'], 'Check partman_basic privileges of id_static_table_p30');
SELECT table_privs_are('partman_test', 'id_static_table_p40', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE','DELETE'], 'Check partman_basic privileges of id_static_table_p40');
SELECT table_privs_are('partman_test', 'id_static_table_p50', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE','DELETE'], 'Check partman_basic privileges of id_static_table_p50');
SELECT table_privs_are('partman_test', 'id_static_table_p60', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE','DELETE'], 'Check partman_basic privileges of id_static_table_p60');
SELECT table_privs_are('partman_test', 'id_static_table_p70', 'partman_basic', ARRAY['SELECT','INSERT','UPDATE','DELETE'], 'Check partman_basic privileges of id_static_table_p70');
SELECT table_owner_is ('partman_test', 'id_static_table_p0', 'partman_owner', 'Check that ownership change worked for id_static_table_p0');
SELECT table_owner_is ('partman_test', 'id_static_table_p10', 'partman_owner', 'Check that ownership change worked for id_static_table_p10');
SELECT table_owner_is ('partman_test', 'id_static_table_p20', 'partman_owner', 'Check that ownership change worked for id_static_table_p20');
SELECT table_owner_is ('partman_test', 'id_static_table_p30', 'partman_owner', 'Check that ownership change worked for id_static_table_p30');
SELECT table_owner_is ('partman_test', 'id_static_table_p40', 'partman_owner', 'Check that ownership change worked for id_static_table_p40');
SELECT table_owner_is ('partman_test', 'id_static_table_p50', 'partman_owner', 'Check that ownership change worked for id_static_table_p50');
SELECT table_owner_is ('partman_test', 'id_static_table_p60', 'partman_owner', 'Check that ownership change worked for id_static_table_p60');
SELECT table_owner_is ('partman_test', 'id_static_table_p70', 'partman_owner', 'Check that ownership change worked for id_static_table_p70');

SELECT * FROM finish();
ROLLBACK;