org.sipfoundry.sipxconfig.backup.FtpBackupPlanTest.java Source code

Introduction

Here is the source code for org.sipfoundry.sipxconfig.backup.FtpBackupPlanTest.java

Source

/*
 *
 *
 * Copyright (C) 2008 Pingtel Corp., certain elements licensed under a Contributor Agreement.
 * Contributors retain copyright to elements licensed under a Contributor Agreement.
 * Licensed to the User under the LGPL license.
 *
 *
 */
package org.sipfoundry.sipxconfig.backup;

import static org.apache.commons.lang.ArrayUtils.EMPTY_STRING_ARRAY;
import static org.easymock.EasyMock.createMock;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.replay;
import static org.easymock.EasyMock.verify;

import java.util.List;
import java.util.Map;

import junit.framework.TestCase;

import org.sipfoundry.sipxconfig.backup.BackupBean.Type;
import org.sipfoundry.sipxconfig.ftp.FtpExternalServerConfig;
import org.sipfoundry.sipxconfig.ftp.FtpExternalServer;

public class FtpBackupPlanTest extends TestCase {

    public void testGetBackups() {
        final FtpExternalServer ftpContext = createMock(FtpExternalServer.class);

        String[] directoryNames = new String[] { "200804221100", "200804221101", "200804221102" };
        String[] names1 = new String[] { BackupPlan.CONFIGURATION_ARCHIVE, BackupPlan.VOICEMAIL_ARCHIVE };
        String[] names2 = new String[] { BackupPlan.CONFIGURATION_ARCHIVE, BackupPlan.CDR_ARCHIVE };
        String[] names3 = new String[] { BackupPlan.VOICEMAIL_ARCHIVE };

        ftpContext.openConnection();

        expect(ftpContext.listDirectories(".")).andReturn(directoryNames).once();
        expect(ftpContext.listDirectories(directoryNames[0])).andReturn(EMPTY_STRING_ARRAY);
        expect(ftpContext.listFiles(directoryNames[0])).andReturn(names1).anyTimes();
        expect(ftpContext.listDirectories(directoryNames[1])).andReturn(EMPTY_STRING_ARRAY);
        expect(ftpContext.listFiles(directoryNames[1])).andReturn(names2).anyTimes();
        expect(ftpContext.listDirectories(directoryNames[2])).andReturn(EMPTY_STRING_ARRAY);
        expect(ftpContext.listFiles(directoryNames[2])).andReturn(names3).anyTimes();

        ftpContext.closeConnection();

        replay(ftpContext);

        FtpExternalServerConfig ftpConfiguration = new FtpExternalServerConfig() {
            @Override
            public FtpExternalServer getFtpContext() {
                return ftpContext;
            }
        };

        FtpBackupPlan backupPlan = new FtpBackupPlan();
        backupPlan.setFtpConfiguration(ftpConfiguration);

        List<Map<Type, BackupBean>> backups = backupPlan.getBackups();

        assertEquals(3, backups.size());
        Map<Type, BackupBean> first = backups.get(0);
        assertEquals(2, first.size());
        assertTrue(first.containsKey(Type.CONFIGURATION));
        assertTrue(first.containsKey(Type.VOICEMAIL));
        assertFalse(first.containsKey(Type.CDR));
        Map<Type, BackupBean> second = backups.get(1);
        assertEquals(2, second.size());
        assertTrue(second.containsKey(Type.CONFIGURATION));
        assertFalse(second.containsKey(Type.VOICEMAIL));
        assertTrue(second.containsKey(Type.CDR));
        Map<Type, BackupBean> third = backups.get(2);
        assertEquals(1, third.size());
        assertFalse(third.containsKey(Type.CONFIGURATION));
        assertTrue(third.containsKey(Type.VOICEMAIL));
        assertFalse(third.containsKey(Type.CDR));
        verify(ftpContext);
    }

}