fitnesse.testsystems.slim.SlimPage Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fitnesse Show documentation

The fully integrated standalone wiki, and acceptance testing framework.

There is a newer version: 20241026

package fitnesse.testsystems.slim;

import fitnesse.slim.SlimError;
import fitnesse.testsystems.TestPage;
import fitnesse.testsystems.slim.tables.SlimTable;
import fitnesse.testsystems.slim.tables.SlimTableFactory;
import org.htmlparser.Parser;
import org.htmlparser.lexer.Lexer;
import org.htmlparser.lexer.Page;
import org.htmlparser.util.NodeList;
import org.htmlparser.util.ParserException;

import java.util.LinkedList;
import java.util.List;

public class SlimPage {

  public static SlimPage Make(TestPage testPage, SlimTestContext testContext, SlimTableFactory slimTableFactory, CustomComparatorRegistry customComparatorRegistry) {
    SlimPage page = new SlimPage(slimTableFactory, customComparatorRegistry);
    page.createSlimTables(testPage, testContext);
    return page;
  }

  public List getTables() { return tables; }
  public HtmlTableScanner getTableScanner() { return tableScanner; }

  private SlimPage(SlimTableFactory slimTableFactory, CustomComparatorRegistry customComparatorRegistry) {
    this.slimTableFactory = slimTableFactory;
    this.customComparatorRegistry = customComparatorRegistry;
  }

  private void createSlimTables(TestPage testPage, SlimTestContext testContext) {
    NodeList nodeList = makeNodeList(testPage);
    tableScanner = new HtmlTableScanner(nodeList);
    tables =  createSlimTables(tableScanner, testContext);
  }

  private NodeList makeNodeList(TestPage pageToTest) {
    String html = pageToTest.getHtml();
    Parser parser = new Parser(new Lexer(new Page(html)));
    try {
      return parser.parse(null);
    } catch (ParserException e) {
      throw new SlimError(e);
    }
  }

  private List createSlimTables(TableScanner tableScanner, SlimTestContext testContext) {
    List allTables = new LinkedList<>();
    for (Table table : tableScanner)
      createSlimTable(allTables, table, testContext);

    return allTables;
  }

  private void createSlimTable(List allTables, Table table, SlimTestContext testContext) {
    String tableId = "" + allTables.size();
    SlimTable slimTable = slimTableFactory.makeSlimTable(table, tableId, testContext);
    // TODO: should customComparatorRegistry be baked into the slimTableFactory?
    if (slimTable != null) {
      slimTable.setCustomComparatorRegistry(customComparatorRegistry);
      allTables.add(slimTable);
    }
  }

  private final SlimTableFactory slimTableFactory;
  private final CustomComparatorRegistry customComparatorRegistry;

  private HtmlTableScanner tableScanner;
  private List tables;
}