UI: add tests covering all updated/new functionality