modify unit test

Docker-DCO-1.1-Signed-off-by: cheneydeng <cheneydeng@qq.com> (github: cheneydeng)