Write some tests for search feature