[TEST][CHECKSTYLE] Add new test. Fix some checkstyle warning.

@mattyb149, @brosander Could you review new tests and some checkstyle fix?