public class SolutionCodeBuilderTest {
private CodeBuilder codeBuilder = new SolutionCodeBuilder();
@Test
public void testBasicClassTemplateStructure() throws Exception {
final Problem problem = make(a(ProblemMaker.Problem, with(className, "Lottery"),
with(returnType, "String[]"), with(methodName, "sortByOdds"),
with(paramTypes, new String[]{"String[]"}), with(paramNames, new String[] {"rules"})));
final String[] expectedTemplate = {"public class Lottery","{","public String[]","sortByOdds","(","String[]","rules",")","{","}","}"};
verifyGeneratedClassTemplate(problem, expectedTemplate);
}