package org.schoellerfamily.gedbrowser.selenium;

import java.util.HashMap;
import java.util.Map;

import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;

/**
 * @author Dick Schoeller
 */
public class SourcePage extends PageBase {
    // FIXME reimplement this to go to the database for information

    /** Database ID associated with this page. */
    private final String id;

    /** Previous page, will always be a person page. */
    private final PersonPage previous;

    /**
     * Relationship between ID and validation data.
     */
    private static final Map<String, String> TITLE_MAP = new HashMap<>();
    static {
        TITLE_MAP.put("S4",
                "Source: S4 - Robinson, Lisa Hope and Schoeller,"
                        + " Richard John, certificate of marriage");
        TITLE_MAP.put("S21",
                "Source: S21 - St. John's Lutheran Church, Pine Grove, PA,"
                        + " church record transcription");
    };

    /**
     * PageObject pattern for a page representing a person.
     *
     * @param driver this is the basic web driver
     * @param id this is the ID of the person page being tested
     * @param previous where we came from. Can be null.
     * @param waiter handles driver specific waits
     */
    public SourcePage(final WebDriver driver, final String id,
            final PersonPage previous, final PageWaiter waiter) {
        super(driver, url(id), waiter);
        this.id = id;
        this.previous = previous;
        final WebElement source = driver.findElements(By.linkText(id))
                .get(0);
        source.click();
    }

    /**
     * Build the URL string for this page.
     *
     * @param id the ID of the person on the page
     * @return the built url string
     */
    private static String url(final String id) {
        return BASE_URL + "/source?db=schoeller&id=" + id;
    }

    /**
     * Go back to the previous person.
     *
     * @return the associated page object.
     */
    public final PersonPage back() {
        if (previous != null) {
            navigateBack();
        }
        return previous;
    }

    /**
     * Check that it is the right title.
     *
     * @return true if the title matches the expected value
     */
    public final boolean titleCheck() {
        System.out.println("Page title is: " + getTitle());
        return getTitle().equals(TITLE_MAP.get(id));
    }
}

@@ -902,18 +902,20 @@
  S21 - S
+ain
 t
-.
  John's 

Fix test