Add tests to WaterRightsManagementController