use `t.Cleanup` instead of `defer func` in tests

[ceyonur]

We're doing some cleanups like vm.Shutdown etc in tests with defers, they should be replaced with t.Cleanup.

Originally posted by @ceyonur in #1876 (comment)