Unit tests

Change-Id: I5d246075656acc5101d23d3ed84d9dd1f7310dcc