Add tests for policy rules