[FEAT] Add util functions to make building UX tests easier

[rj45]

The ux layer was designed to make adding tests possible, but currently there's too high of friction to adding them. Add some utility functions to make it easier to set up scenarios. Might be best to add tests and figure out what can be pulled out to make the next test easier to make.