Added search test with mock