In theory we should be able to add suites/groups/test nodes to the test tree from the outline data (which we use to drive code lens). Initially we could do this just as we discover them naturally (eg. when files are opened), but perhaps scanning all _test.dart files in the test folder after analysis has finished would work too.