return type;
    }

    @Override
    public String getExtension() {
        if (extension == null) {
            return type;
        }
        return extension;
    }

    public void setExtension(final String extension) {
        this.extension = extension;
    }

    @Override
    public String getClassifier() {
        return classifier;
    }

public class ArtifactHandlerMock implements ArtifactHandler {

    private String extension, directory, classifier, packaging, language;
    private boolean includesDependencies, addedToClasspath;

    public void setExtension(String extension) {
        this.extension = extension;
    }

    @Override
    public String getExtension() {
        return extension;
    }

    public void setDirectory(String directory) {

        for (Artifact artifact : artifacts) {
            SnapshotVersion sv = new SnapshotVersion();
            sv.setClassifier(artifact.getClassifier());
            sv.setExtension(artifact.getExtension());
            sv.setVersion(getVersion());
            sv.setUpdated(lastUpdated);
            versions.put(getKey(sv.getClassifier(), sv.getExtension()), sv);
        }

            }

            SnapshotVersion sv = new SnapshotVersion();
            sv.setClassifier(artifact.getClassifier());
            sv.setExtension(artifact.getExtension());
            sv.setVersion(version);
            sv.setUpdated(lastUpdated);

            versions.put(getKey(sv.getClassifier(), sv.getExtension()), sv);
        }