This works for me, but is quite hacky. I hope they have a better solution soon.
It only works if you’re in a PIE or Game session unless you modified the function. I make my test throw up a warning if the return value was null, telling people to start a PIE or game session to run the test. Fortunately this isn’t a problem for my specific test I’m working on even though it’s quite hacky.