Add tests for views