Add mock support.