v0.0-695-g661d809: [lexical_context] Also verify reason for internal state machine states.

This makes debugging much easier.

No functional change.

PiperOrigin-RevId: 338308662