Our current UI testing is done using vscode-extension-tester.
The official guidance from the VSCode team is to use @vscode/test-electron and the new @vscode/test-cli packages.
We hit some issues with older versions of the VSCode teams UI framework, but it would be worth investigating if things are working better now.