Write UI tests

Created by: sashaostrikov

/cc @jplahn @ryphil