Added unit tests for policy module