We should do `context.terminate()` as well. Otherwise the test will leave threads running.

But I see the same applies to other methods in this class as well. Not letting that block this PR.