Use expect syntax for mock instead of should syntax