com.codeborne.xlstest.XLS Maven / Gradle / Ivy

Go to download
package com.codeborne.xlstest;

import com.codeborne.xlstest.matchers.ContainsRow;
import com.codeborne.xlstest.matchers.ContainsText;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.ss.usermodel.WorkbookFactory;
import org.hamcrest.Matcher;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URI;
import java.net.URL;

import static com.codeborne.xlstest.IO.readBytes;
import static com.codeborne.xlstest.IO.readFile;

public class XLS {
  public final String name;
  public final Workbook excel;

  private XLS(String name, byte[] content) {
    this.name = name;
    try (InputStream inputStream = new ByteArrayInputStream(content)) {
      excel = WorkbookFactory.create(inputStream);
    }
    catch (Exception e) {
      throw new IllegalArgumentException("Invalid XLS " + name, e);
    }
  }

  public XLS(File xlsFile) {
    this(xlsFile.getAbsolutePath(), readFile(xlsFile));
  }

  public XLS(URL url) throws IOException {
    this(url.toString(), readBytes(url));
  }

  public XLS(URI uri) throws IOException {
    this(uri.toURL());
  }
  
  public XLS(byte[] content) {
    this("", content);
  }

  public XLS(InputStream inputStream) throws IOException {
    this(readBytes(inputStream));
  }
  
  public static Matcher containsText(String text) {
    return new ContainsText(text);
  }

  public static Matcher containsRow(String... cellTexts) {
    return new ContainsRow(cellTexts);
  }
}