Add Unit Test

[mirdaki]

Testing is of course important, but sort of difficult for UI. We need a good and useful pattern.