org.eclipse.contribution.weaving.jdt.tests/src/org/eclipse/contribution/weaving/jdt/tests/MockSourceTransformer.java

package org.eclipse.contribution.weaving.jdt.tests;

import org.eclipse.contribution.jdt.sourceprovider.ISourceTransformer;
import org.eclipse.jdt.core.IBuffer;
import org.eclipse.jdt.core.ICompilationUnit;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jdt.internal.core.search.indexing.SourceIndexer;
import org.eclipse.jdt.internal.core.search.indexing.SourceIndexerRequestor;

public class MockSourceTransformer implements ISourceTransformer {

    public static final String MOCK_CLASS_NAME = "Mock";
    
    public static int ensureRealBufferCalled = 0;
    
    public static int ensureSourceIndexerRequestorCreated = 0;
    
    public char[] convert(char[] toConvert) {
        return ("class " + MOCK_CLASS_NAME + " {\n\tint x;\n\tint y;\n}").toCharArray();
    }

    public IBuffer ensureRealBuffer(ICompilationUnit unit)
            throws JavaModelException {
        ensureRealBufferCalled++;
        return unit.getBuffer();
    }

    public SourceIndexerRequestor createIndexerRequestor(SourceIndexer indexer) {
        ensureSourceIndexerRequestorCreated++;
        return new SourceIndexerRequestor(indexer);
    }
}