Testing a websocket with multiple responses

[tophopstop]

I have a websocket that returns multiple messages, akin to this workflow:

client             server
-------------------------
==>  start workflow X
starting workflow X  <===
part A done          <===
part B done          <===
workflow X done      <===

I'd like to use Test::Mojo's websocket_ok to test it, but as far as I can see, it only gets the first message.

Am I missing something?