Create tests to catch prompt_session breaking #364
Description
Issue #362 broke builds without tests catching the issue. We need to introduce tests to catch this.
Follow repro steps in the issue linked above and test on commit 9cb84e21ac50f2082c1c136780b674cb35a0c5d8.