Unit tests

Change-Id: I3d45b7c55d4bd4ddd38d3575b769e4ed86151e6d