Add some tests for the view