123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122 |
- /* ====================================================================
- Licensed to the Apache Software Foundation (ASF) under one or more
- contributor license agreements. See the NOTICE file distributed with
- this work for additional information regarding copyright ownership.
- The ASF licenses this file to You under the Apache License, Version 2.0
- (the "License"); you may not use this file except in compliance with
- the License. You may obtain a copy of the License at
-
- http://www.apache.org/licenses/LICENSE-2.0
-
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- ==================================================================== */
- package org.apache.poi.hssf.dev;
-
- import static org.junit.Assert.assertNotNull;
-
- import java.io.File;
- import java.io.FileInputStream;
- import java.io.FilenameFilter;
- import java.io.OutputStream;
- import java.util.ArrayList;
- import java.util.HashMap;
- import java.util.List;
- import java.util.Locale;
- import java.util.Map;
-
- import org.apache.poi.POIDataSamples;
- import org.apache.poi.hssf.record.crypto.Biff8EncryptionKey;
- import org.apache.poi.hssf.usermodel.HSSFWorkbook;
- import org.apache.poi.util.NullOutputStream;
- import org.junit.Rule;
- import org.junit.Test;
- import org.junit.rules.ExpectedException;
- import org.junit.runner.RunWith;
- import org.junit.runners.Parameterized;
- import org.junit.runners.Parameterized.Parameter;
- import org.junit.runners.Parameterized.Parameters;
-
- /**
- * Base class for integration-style tests which iterate over all test-files
- * and execute the same action to find out if any change breaks these applications.
- *
- * This test uses {@link Parameterized} to run the test for each file separatedely.
- */
- @RunWith(Parameterized.class)
- public abstract class BaseXLSIteratingTest {
- protected static final OutputStream NULL_OUTPUT_STREAM = new NullOutputStream();
-
- @Rule
- public ExpectedException thrown = ExpectedException.none();
-
- protected static final Map<String,Class<? extends Throwable>> EXCLUDED =
- new HashMap<String,Class<? extends Throwable>>();
-
- @Parameters(name="{index}: {0}")
- public static Iterable<Object[]> files() {
- String dataDirName = System.getProperty(POIDataSamples.TEST_PROPERTY);
- if(dataDirName == null) {
- dataDirName = "test-data";
- }
-
- List<Object[]> files = new ArrayList<Object[]>();
- findFile(files, dataDirName + "/spreadsheet");
- findFile(files, dataDirName + "/hpsf");
-
- return files;
- }
-
- private static void findFile(List<Object[]> list, String dir) {
- String[] files = new File(dir).list(new FilenameFilter() {
- @Override
- public boolean accept(File arg0, String arg1) {
- return arg1.toLowerCase(Locale.ROOT).endsWith(".xls");
- }
- });
-
- assertNotNull("Did not find any xls files in directory " + dir, files);
-
- for(String file : files) {
- list.add(new Object[] { new File(dir, file) });
- }
- }
-
- @Parameter
- public File file;
-
- @Test
- public void testMain() throws Exception {
- // we had intermittent problems when this was set differently somehow, let's try to set it here so it always is set correctly for these tests
- Biff8EncryptionKey.setCurrentUserPassword(null);
-
- String fileName = file.getName();
- if (EXCLUDED.containsKey(fileName)) {
- thrown.expect(EXCLUDED.get(fileName));
- }
-
- try {
- runOneFile(file);
- } catch (Exception e) {
- // try to read it in HSSFWorkbook to quickly fail if we cannot read the file there at all and thus probably should use EXCLUDED instead
- FileInputStream stream = new FileInputStream(file);
- HSSFWorkbook wb = null;
- try {
- wb = new HSSFWorkbook(stream);
- assertNotNull(wb);
- } finally {
- if (wb != null) {
- wb.close();
- }
- stream.close();
- }
-
- throw e;
- }
- }
-
- abstract void runOneFile(File pFile) throws Exception;
- }
|