Update test driver

Signed-off-by: Bao Haojun <hjbao@marvell.com>