add unit test

Change-Id: I2515dfe3e4b428a1c5bf9754054c3e720f48ece0