Convert e2ee tests to use Vitest contexts

[Half-Shot]

https://vitest.dev/guide/test-context.html looks like a much nicer way of setting up a test environment than our current approach.