Oh Ok, I just follow the existing unit test way. I will update it later