Unit tests

Change-Id: I1195594e6e67367534dad9cff39d96984560e779