java - java : 集合: 列表文件

Question

我想使用 listfiles 设置对象集合，而不是手动执行此操作：

@Paramaters
public static Collection<Object[]> data()
{
     Object[][] data = new Object[][]{{"test_files/myfilea.txt"}, {"test_files/myfileb.txt"},{"test_files/myfilec.txt"},{"test_files/myfiled.txt"}};

return Arrays.asList(data);
}

任何想法，解决方案，请。

Answer 1

做这个

File theDirectory = new File("test_files");
File[] theFiles;
if(theDirectory.isDirectory()) 
    theFiles = theDirectory.listFiles();
else
    return null; //Or throw exception... up to you
List<Object[]> yourFilesList = new ArrayList<Object[]>();   //Edited here
for(File f: theFiles)
{
    if(f.isFile())
        //yourFilesList.add(new Object[]{f.getName()});       //Edited here again
        // or your code may rely on directory also so you can do this
         yourFilesList.add(new Object[]{"test_files/" + f.getName()}); 
}
return yourFilesList;

Answer 2

像这样的东西？如果不是，请澄清。

public static Collection<File> data() {
    return Arrays.asList(new File("test_files").listFiles());
}

如果您必须在集合中包含数组，请尝试以下操作：

public static Collection<Object[]> data() {
    return java.util.Collections.singleton((Object[])new File("test_files").listFiles());
}

编辑：再试一次：

public static Collection<Object[]> data() {
    String[] fileNames = new File("test_files").list();
    Object[] namesInObjectArray = new Object[fileNames.length];
    for (int i = 0; i < fileNames.length; i++) {
        namesInObjectArray[i] = "test_files/" + fileNames[i];
    }
    Object[][] outerObjectArray = new Object[][] { namesInObjectArray };
    return Arrays.asList(outerObjectArray);
}