Add test for new API

Change-Id: I91f20e78b7dc68269a19f906296ea8e8da12f748