[jk] refactor test to different class