I think this is not as easy as it sounds as we have to deal with asynchronous tests here. If you have a working suggestion, please tell me or add it to this PR (if possible somehow??) :)