Add tests for stock_location_search_stock

Also add a missing import