Tests are "hard to read, hard to modify, hard to debug" (@melekes). One idea how tests can be improved is splitting business (algorithm) logic in a function and separate it from a real message sending. This logic should capture core algorithm, i.e., for a given state, what messages we should send and how we should react upon message reception.