Add unit test for web