update 9.6 add UT


Signed-off-by: jackychen <jackychen129@gmail.com>