fix test framework's code to fit new style.

Signed-off-by: Daogang Tang <daogangtang@gmail.com>