[test] Updated test and surce code.