Better diagnostics for functional tests

When the test fail during the state mutation expectation phase, now the diagnostics will point to that line, not to the Send or Receive step. That should help to identify the reason for the test failure.