Fix permission tests

ildus · ildus · commit c7d8fd4491a6 · 2017-04-04T19:33:26.000+03:00
diff --git a/expected/pathman_permissions.out b/expected/pathman_permissions.out
@@ -19,8 +19,7 @@ GRANT SELECT ON permissions.user1_table TO user2;
 /* Should fail (don't own parent) */
 SET ROLE user2;
 SELECT create_range_partitions('permissions.user1_table', 'id', 1, 10, 2);
-WARNING:  only the owner or superuser can change partitioning configuration of table "user1_table"
-ERROR:  new row violates row-level security policy for table "pathman_config"
+ERROR:  only the owner or superuser can change partitioning configuration of table "user1_table"
 /* Should be ok */
 SET ROLE user1;
 SELECT create_range_partitions('permissions.user1_table', 'id', 1, 10, 2);
diff --git a/src/pl_funcs.c b/src/pl_funcs.c
@@ -723,6 +723,13 @@ add_to_pathman_config(PG_FUNCTION_ARGS)
 	if (!check_relation_exists(relid))
 		elog(ERROR, "Invalid relation %u", relid);
 
+	if (!check_security_policy_internal(relid, GetUserId()))
+	{
+		elog(ERROR, "only the owner or superuser can change "
+					  "partitioning configuration of table \"%s\"",
+			 get_rel_name_or_relid(relid));
+	}
+
 	/* Select partitioning type using 'range_interval' */
 	parttype = PG_ARGISNULL(2) ? PT_HASH : PT_RANGE;
 
