Refactoring test code.

Signed-off-by: Kouhei Maeda <mkouhei@palmtb.net>