Unit tests

Change-Id: I144d5d8e22f593b263f0b7263fc04f7ad736999b