Add unit test for wizard cancellation