Added test for after spy

Added test to validate after-call on spy and new sample flow.