Modify the test code generation to use the new //J Api