dirpolltest.java

java pos,你可以直接编译运行,

package dirpoll;

import java.io.*;

import junit.framework.TestCase;
import org.jpos.util.DirPoll;
import org.jpos.util.DirPoll.DirPollException;

/**
 * $Revision: 1.2 $
 * $Date: 2005/08/06 18:40:18 $
 * $Author: Matthew Milliss <[e-mail]>
 */
public class DirPollTest extends TestCase {

    private DirPoll dirPoll;
    private static final String BASE_PATH = 
        System.getProperty ("java.io.tmpdir");
    private static final String ARCHIVE_DIR = BASE_PATH + "/archive";
    
    private File testIncomingFile;
    private boolean fileProcessed;
    
    public DirPollTest(String arg0) {   
        super(arg0);
    }
    
    protected void setUp() throws Exception {
        super.setUp();        
        dirPoll = new DirPoll();
        dirPoll.setPath(BASE_PATH);
        dirPoll.setProcessor(new DirPoll.FileProcessor() {
            public void process(File name) throws DirPollException { 
                System.out.println ("File processed " + name.toString());
                fileProcessed = true;
            }
        });
        dirPoll.createDirs();       
        new Thread(dirPoll).start();
    }
    
    protected void tearDown() throws Exception {
        super.tearDown();
        dirPoll.destroy();
        File archiveDir = new File(ARCHIVE_DIR);
        File[] files = archiveDir.listFiles();
        for (int i = 0; i < files.length; i++) {
            files[i].delete();
        } 
    }
    
    public void testArchiveFile() throws IOException, InterruptedException {
        String dateFormatString = "yyyyMMddHHmmss";
        String filename = "dodgyTestFile.test";
        
        dirPoll.setShouldArchive(true);
        dirPoll.setArchiveDateFormat(dateFormatString);
        dirPoll.setShouldTimestampArchive(true);
        testIncomingFile = new File(BASE_PATH + "/request/" + filename);
        FileOutputStream fileOutputStream = new FileOutputStream(testIncomingFile);
        fileOutputStream.write(new String("test").getBytes());
        fileOutputStream.flush();
        fileOutputStream.close();
        assertTrue(testIncomingFile.exists());
        
        while(!fileProcessed) {            
            Thread.yield();
        }
        System.out.println ("");
        Thread.sleep(200);
        File archiveDirectory = new File(ARCHIVE_DIR);        
        assertEquals(1, archiveDirectory.listFiles().length);
        assertEquals((dateFormatString.length() + filename.length() + 1), archiveDirectory.listFiles()[0].getName().length());
    }

    public void testDoNotArchiveFile() throws IOException, InterruptedException {
        dirPoll.setShouldArchive(false);
        testIncomingFile = new File(BASE_PATH + "/request/dodgyTestFile2.test");
        FileOutputStream fileOutputStream = new FileOutputStream(testIncomingFile);
        fileOutputStream.write(new String("test").getBytes());
        fileOutputStream.flush();
        fileOutputStream.close();
        assertTrue(testIncomingFile.exists());
        
        while(!fileProcessed) {            
            Thread.yield();
        }
        Thread.sleep(200);
        File archiveDirectory = new File(ARCHIVE_DIR);        
        assertEquals(0, archiveDirectory.listFiles().length);
    }
    
    public void testArchiveNoTimestamp() throws Exception {

        String filename = "dodgyTestFile3.test";
        
        dirPoll.setShouldArchive(true);
        dirPoll.setShouldTimestampArchive(false);
        testIncomingFile = new File(BASE_PATH + "/request/" + filename);
        FileOutputStream fileOutputStream = new FileOutputStream(testIncomingFile);
        fileOutputStream.write(new String("test").getBytes());
        fileOutputStream.flush();
        fileOutputStream.close();
        assertTrue(testIncomingFile.exists());
        
        while(!fileProcessed) {            
            Thread.yield();
        }
        Thread.sleep(200);
        File archiveDirectory = new File(ARCHIVE_DIR);        
        assertEquals(1, archiveDirectory.listFiles().length);
        assertEquals(filename.length(), archiveDirectory.listFiles()[0].getName().length());
        assertEquals(filename, archiveDirectory.listFiles()[0].getName());
        
    }
    
}