Add unit tests for sandbox UI & utils