Unit tests

Change-Id: I94ec3a95790581386d8c90097ed183a5536f101c