Add test for `Query_Restriction`