DefaultJavaSourceTest.java

package com.thoughtworks.qdox.model.impl;

import com.thoughtworks.qdox.library.ClassLibrary;
import com.thoughtworks.qdox.model.JavaClass;
import com.thoughtworks.qdox.model.JavaPackage;
import com.thoughtworks.qdox.model.JavaSourceTest;

import java.util.List;

public class DefaultJavaSourceTest extends JavaSourceTest<DefaultJavaSource>
{
    @Override
	public DefaultJavaSource newJavaSource( ClassLibrary classLibrary )
    {
        return new DefaultJavaSource(classLibrary);
    }

    @Override
	public void setPackage( DefaultJavaSource source, JavaPackage pckg )
    {
        source.setPackage( pckg );
    }

    @Override
    public void setClasses( DefaultJavaSource source, List<JavaClass> classes )
    {
        for(JavaClass cls: classes) {
            source.addClass( cls );
        }
    }
    
    @Override
    public void setImports( DefaultJavaSource source, List<String> imports )
    {
        for(String imprt : imports) {
            source.addImport( imprt );
        }
    }
}