test: update test code for 'Test simulation'

This update is because of interfaces changing in Data.

Signed-off-by: Zhengwang Ruan <ruan.zhengwang@gmail.com>