Sup Goose, hope you're well?
On the assessment they are looking for how you react then what you do, if you're flying keep flying and let the PM tell you what’s going on, the classic "Aviate, Navigate, Communicate" comes into play.
Ultimately they want to see you are looking ahead and if you're on fire, someone is dieing etc, you are going back to the ground for help, getting the calls and squawk are great, but making the decision to get it on the ground is the one they will be looking for more.
Enjoy
YYZ