[Java]SuffixFileFilter を使ったファイルフィルタのスニペット。

SuffixFileFilterTest.java

package net.tomoyamkung.org.apache.commons.io.filefilter;

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

import java.io.File;

import org.apache.commons.io.filefilter.SuffixFileFilter;
import org.junit.Before;
import org.junit.Test;

public class SuffixFileFilterTest {

	private File dir;
	
	@Before
	public void setUp() throws Exception {
		dir = new File("testdata/org_apache_commons_io_filefilter/suffix_file_filter_test");	
	}

	@Test
	public void 拡張子がtxtのファイル名を取得する() throws Exception {
		// Setup
		String suffix = ".txt";
		
		// Exercise
		String[] list = dir.list(new SuffixFileFilter(suffix));
		
		// Verify
		assertThat(list.length, is(2));
		assertThat(list[0], is("hoge1.txt"));
		assertThat(list[1], is("hoge2.txt"));
	}
	
	@Test
	public void 拡張子がjpgとcsvのファイル名を取得する() throws Exception {
		// Setup
		String[] suffix = {".jpg", ".csv"};
		
		// Exercise
		String[] list = dir.list(new SuffixFileFilter(suffix));
		
		// Verify
		assertThat(list.length, is(2));
		assertThat(list[0], is("piyo.jpg"));
		assertThat(list[1], is("var.csv"));
	}

}

テストデータとして、"testdata/org_apache_commons_io_filefilter/suffix_file_filter_test" ディレクトリを作成し、そこに次のファイルを用意する。

• hoge1.txt
• hoge2.txt
• piyo.jpg
• var.csv