[Java]FilenameFilter を使ったフィルタクラス。

ExtensionFileFilter.java

package net.tomoyamkung;

import java.io.File;
import java.io.FilenameFilter;

/**
 * 指定した拡張子と一致するファイルを取得するフィルタクラス。
 * 
 * @author tomoyamkung
 *
 */
public class ExtensionFileFilter implements FilenameFilter {

	private String extension;

	public ExtensionFileFilter(String extension) {
		this.extension = extension;
	}

	@Override
	public boolean accept(File dir, String name) {
		File file = new File(name);
		if(file == null || file.isDirectory()) {
			return false;
		}
		return name.endsWith(extension);
	}

}

ExtensionFileFilterTest.java

package net.tomoyamkung;

import static org.hamcrest.CoreMatchers.*;
import static org.junit.Assert.*;

import java.io.File;

import org.junit.Test;

public class ExtensionFileFilterTest {

	@Test
	public void 拡張子がtxtのファイル名を取得する() throws Exception {
		// Setup
		String extension = ".txt";
		ExtensionFileFilter filter = new ExtensionFileFilter(extension);
		File dir = new File("testdata/file_filter_test");
		
		// Exercise
		String[] list = dir.list(filter);
		
		// Verify
		assertThat(list.length, is(2));
		assertThat(list[0], is("hoge1.txt"));
		assertThat(list[1], is("hoge2.txt"));
	}

}