Add test for comments controller