Since #3705 is including CodeLens, it should also work for main functions (not just tests), allowing you to set templateFor (as a path prefix) on any launch config for it to be used by:

• main function code lens
• group/test code lens
• test runner