Unit tests

Change-Id: I5c0fb760794312ded014a0385a94552fcd5b48d1