Add unit tests for UI

No match.