[code-infra] Changes for test util to work in vitest

[code-infra] Changes for test util to work in vitest #54098