Add more tests for comments controller