org.sventon.export.DefaultExportExecutorTest.java Source code

Introduction

Here is the source code for org.sventon.export.DefaultExportExecutorTest.java

Source

/*
 * ====================================================================
 * Copyright (c) 2005-2012 sventon project. All rights reserved.
 *
 * This software is licensed as described in the file LICENSE, which
 * you should have received as part of this distribution. The terms
 * are also available at http://www.sventon.org.
 * If newer versions of this license are posted there, you may use a
 * newer version instead, at your option.
 * ====================================================================
 */
package org.sventon.export;

import org.easymock.classextension.EasyMock;
import org.junit.Test;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockHttpServletResponse;
import org.sventon.SVNConnection;
import org.sventon.TestUtils;
import org.sventon.appl.ConfigDirectory;
import org.sventon.model.PathRevision;
import org.sventon.service.RepositoryService;
import org.sventon.util.WebUtils;

import java.io.File;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;

import static org.easymock.classextension.EasyMock.*;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;

public class DefaultExportExecutorTest {

    private static final String UUID_STRING = "c5eaa2ba-2655-444b-aa64-c15ecff3e6da";

    @Test
    public void testSetArchiveFileCharset() throws Exception {
        final ConfigDirectory configDirectoryMock = EasyMock.createMock(ConfigDirectory.class);
        expect(configDirectoryMock.getExportDirectory()).andStubReturn(new File("."));
        replay(configDirectoryMock);
        final DefaultExportExecutor executor = new DefaultExportExecutor(configDirectoryMock);
        assertEquals(DefaultExportExecutor.FALLBACK_CHARSET, executor.getArchiveFileCharset().toString());
        executor.setArchiveFileCharset(null);
        assertEquals(DefaultExportExecutor.FALLBACK_CHARSET, executor.getArchiveFileCharset().toString());
        executor.setArchiveFileCharset("UTF-8");
        assertEquals("UTF-8", executor.getArchiveFileCharset().toString());
        verify(configDirectoryMock);
    }

    @Test
    public void testPrepareResponse() throws Exception {
        final ConfigDirectory configDirectoryMock = createMock(ConfigDirectory.class);
        final DefaultExportExecutor exportExecutor = new DefaultExportExecutor(configDirectoryMock);
        final MockHttpServletRequest request = new MockHttpServletRequest();
        final MockHttpServletResponse response = new MockHttpServletResponse();

        assertEquals(0, response.getHeaderNames().size());
        assertNull(response.getContentType());
        exportExecutor.prepareResponse(request, response, new File("testfile"));
        assertEquals(1, response.getHeaderNames().size());
        assertEquals("attachment; filename=\"testfile\"", response.getHeader(WebUtils.CONTENT_DISPOSITION_HEADER));
        assertEquals(WebUtils.APPLICATION_OCTET_STREAM, response.getContentType());
    }

    @Test
    public void testExportTask() throws Exception {
        final List<PathRevision> entries = new ArrayList<PathRevision>();

        final ConfigDirectory configDirectoryMock = createMock(ConfigDirectory.class);
        final ExportDirectory exportDirectoryMock = createMock(ExportDirectory.class);
        final RepositoryService repositoryServiceMock = createMock(RepositoryService.class);
        final SVNConnection connection = createMock(SVNConnection.class);
        final File exportDir = new File(TestUtils.TEMP_DIR);

        expect(configDirectoryMock.getExportDirectory()).andStubReturn(exportDir);
        expect(configDirectoryMock.getExportDirectory()).andStubReturn(exportDir);
        expect(exportDirectoryMock.getDirectory()).andReturn(new File("."));
        expect(exportDirectoryMock.getUUID()).andStubReturn(UUID.fromString(UUID_STRING));
        expect(exportDirectoryMock.mkdirs()).andStubReturn(true);
        repositoryServiceMock.export(connection, entries, 123, exportDir);
        exportDirectoryMock.delete();
        expect(exportDirectoryMock.compress()).andStubReturn(exportDir);

        replay(configDirectoryMock);
        replay(exportDirectoryMock);
        replay(connection);

        final DefaultExportExecutor exportExecutor = new DefaultExportExecutor(configDirectoryMock);

        exportExecutor.setRepositoryService(repositoryServiceMock);

        final DefaultExportExecutor.ExportTask exportTask = exportExecutor.new ExportTask(exportDirectoryMock,
                connection, entries, 123);

        exportTask.call();

        verify(configDirectoryMock);
        verify(exportDirectoryMock);
        verify(connection);
    }
}