Alternative to Assembly.GetEntryAssembly() for unit tests

Question

Is there a way to not only get the EXE name when running normally, but also get vstest.executionengine.x86.exe when running under unit tests, without having to