Unit tests

Change-Id: Iac033728253a3dd736d47b7140bdd3f45d13634d