Move test into actual Junit test