refactor some test

Signed-off-by: Ahmad Shah Hafizan Hamidin <ahmadshahhafizan@gmail.com>