Unit tests

Change-Id: Ia0e9b974ec2611eaa4f4a760483ec38c8765f88e