Add tests for the view