org.eclipse.egit.core.test.op.TrackUntrackOperationTest.java Source code

Introduction

Here is the source code for org.eclipse.egit.core.test.op.TrackUntrackOperationTest.java

Source

/*******************************************************************************
 * Copyright (C) 2010, Mathias Kinzler <mathias.kinzler@sap.com>
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *******************************************************************************/
package org.eclipse.egit.core.test.op;

import static org.junit.Assert.assertTrue;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;

import org.eclipse.core.resources.IContainer;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.IResourceVisitor;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.egit.core.op.AddToIndexOperation;
import org.eclipse.egit.core.op.UntrackOperation;
import org.eclipse.egit.core.project.RepositoryMapping;
import org.eclipse.egit.core.test.DualRepositoryTestCase;
import org.eclipse.egit.core.test.TestRepository;
import org.eclipse.jgit.dircache.DirCache;
import org.eclipse.jgit.lib.Constants;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class TrackUntrackOperationTest extends DualRepositoryTestCase {

    File workdir;

    IProject project;

    String projectName = "TrackTest";

    @Before
    public void setUp() throws Exception {

        workdir = testUtils.createTempDir("Repository1");

        repository1 = new TestRepository(new File(workdir, Constants.DOT_GIT));

        // now we create a project in repo1
        project = testUtils.createProjectInLocalFileSystem(workdir, projectName);
        testUtils.addFileToProject(project, "folder1/file1.txt", "Hello world");

        repository1.connect(project);
    }

    @After
    public void tearDown() throws Exception {
        project.close(null);
        project.delete(false, false, null);
        repository1.dispose();
        repository1 = null;
        testUtils.deleteTempDirs();
    }

    @Test
    public void testTrackFiles() throws Exception {

        final ArrayList<IFile> files = new ArrayList<IFile>();

        project.accept(new IResourceVisitor() {

            public boolean visit(IResource resource) throws CoreException {
                if (resource instanceof IFile)
                    files.add((IFile) resource);
                return true;
            }
        });

        IFile[] fileArr = files.toArray(new IFile[files.size()]);

        assertTrackedState(fileArr, false);

        AddToIndexOperation trop = new AddToIndexOperation(files);
        trop.execute(new NullProgressMonitor());

        assertTrackedState(fileArr, true);

        UntrackOperation utop = new UntrackOperation(Arrays.asList(fileArr));

        utop.execute(new NullProgressMonitor());

        assertTrackedState(fileArr, false);
    }

    private void assertTrackedState(IFile[] fileArr, boolean expectedState) throws IOException {
        DirCache cache = repository1.getRepository().readDirCache();
        for (IFile file : fileArr) {
            RepositoryMapping rm = RepositoryMapping.getMapping(file);
            String fileDir = rm.getRepoRelativePath(file);
            boolean tracked = cache.findEntry(fileDir) > -1;
            assertTrue("Wrong tracking state", tracked == expectedState);
        }
    }

    @Test
    public void testTrackProject() throws Exception {

        final ArrayList<IContainer> containers = new ArrayList<IContainer>();
        containers.add(project);

        final ArrayList<IFile> files = new ArrayList<IFile>();

        project.accept(new IResourceVisitor() {

            public boolean visit(IResource resource) throws CoreException {
                if (resource instanceof IFile)
                    files.add((IFile) resource);
                return true;
            }
        });

        IFile[] fileArr = files.toArray(new IFile[files.size()]);

        assertTrackedState(fileArr, false);

        AddToIndexOperation trop = new AddToIndexOperation(containers);
        trop.execute(new NullProgressMonitor());

        assertTrackedState(fileArr, true);

        UntrackOperation utrop = new UntrackOperation(containers);
        utrop.execute(new NullProgressMonitor());

        assertTrackedState(fileArr, false);
    }

}