Add XTEST codegen support