Unit tests

Change-Id: I4e10d86d76e875e56ffc84910e769c56a00654f1