Fix --test code generation to properly avoid unstable warnings

[brson]

The solution to #20823 in #20828 is a blanked allow that will let unstable user code into test cases. Need to fix this through smarter expansion.

Nominating because it's a stability hole.

[brson]

Still broken on beta!

[brson]

This appears to be fixed.