Add search unit tests