Add unit test for controllers