[Tests] Refactor test to use assertTraceWith and assertBackToState
[Tests] Refactor test to use assertTraceWith and assertBackToState

tlaplus/tlaplus,lemmy/tlaplus,tlaplus/tlaplus,lemmy/tlaplus,lemmy/tlaplus,lemmy/tlaplus,tlaplus/tlaplus,tlaplus/tlaplus