// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "config.h"
#include "core/html/HTMLSelectElement.h"

#include "core/frame/FrameView.h"
#include "core/html/HTMLDocument.h"
#include "core/html/forms/FormController.h"
#include "core/loader/EmptyClients.h"
#include "core/testing/DummyPageHolder.h"
#include <gtest/gtest.h>

namespace blink {

class HTMLSelectElementTest : public::testing::Test {
protected:
    virtual void SetUp() override;
    HTMLDocument& document() const { return *m_document; }

private:
    OwnPtr<DummyPageHolder> m_dummyPageHolder;
    RefPtrWillBePersistent<HTMLDocument> m_document;
};

void HTMLSelectElementTest::SetUp()
{
    Page::PageClients pageClients;
    fillWithEmptyClients(pageClients);
    m_dummyPageHolder = DummyPageHolder::create(IntSize(800, 600), &pageClients);

    m_document = toHTMLDocument(&m_dummyPageHolder->document());
    m_document->setMimeType("text/html");
    m_document->setCharset("utf-8");
}

TEST_F(HTMLSelectElementTest, SaveRestoreSelectSingleFormControlState)
{
    document().documentElement()->setInnerHTML(String("<!DOCTYPE HTML><select id='sel'>"
        "<option value='111' id='0'>111</option>"
        "<option value='222'>222</option>"
        "<option value='111' selected id='2'>!666</option>"
        "<option value='999'>999</option></select>"), ASSERT_NO_EXCEPTION);
    document().view()->updateLayoutAndStyleForPainting();
    Element* element = document().getElementById("sel");
    HTMLFormControlElementWithState* select = toHTMLSelectElement(element);
    HTMLOptionElement* opt0 = toHTMLOptionElement(document().getElementById("0"));
    HTMLOptionElement* opt2 = toHTMLOptionElement(document().getElementById("2"));

    // Save the select element state, and then restore again.
    // Test passes if the restored state is not changed.
    EXPECT_EQ(2, toHTMLSelectElement(element)->selectedIndex());
    EXPECT_FALSE(opt0->selected());
    EXPECT_TRUE(opt2->selected());
    FormControlState selectState = select->saveFormControlState();
    EXPECT_EQ(2U, selectState.valueSize());

    // Clear the selected state, to be restored by restoreFormControlState.
    toHTMLSelectElement(select)->setSelectedIndex(-1);
    ASSERT_FALSE(opt2->selected());

    // Restore
    select->restoreFormControlState(selectState);
    EXPECT_EQ(2, toHTMLSelectElement(element)->selectedIndex());
    EXPECT_FALSE(opt0->selected());
    EXPECT_TRUE(opt2->selected());
}

TEST_F(HTMLSelectElementTest, SaveRestoreSelectMultipleFormControlState)
{
    document().documentElement()->setInnerHTML(String("<!DOCTYPE HTML><select id='sel' multiple>"
        "<option value='111' id='0'>111</option>"
        "<option value='222'>222</option>"
        "<option value='111' selected id='2'>!666</option>"
        "<option value='999' selected id='3'>999</option></select>"), ASSERT_NO_EXCEPTION);
    document().view()->updateLayoutAndStyleForPainting();
    HTMLFormControlElementWithState* select = toHTMLSelectElement(document().getElementById("sel"));

    HTMLOptionElement* opt0 = toHTMLOptionElement(document().getElementById("0"));
    HTMLOptionElement* opt2 = toHTMLOptionElement(document().getElementById("2"));
    HTMLOptionElement* opt3 = toHTMLOptionElement(document().getElementById("3"));

    // Save the select element state, and then restore again.
    // Test passes if the selected options are not changed.
    EXPECT_FALSE(opt0->selected());
    EXPECT_TRUE(opt2->selected());
    EXPECT_TRUE(opt3->selected());
    FormControlState selectState = select->saveFormControlState();
    EXPECT_EQ(4U, selectState.valueSize());

    // Clear the selected state, to be restored by restoreFormControlState.
    opt2->setSelected(false);
    opt3->setSelected(false);
    ASSERT_FALSE(opt2->selected());
    ASSERT_FALSE(opt3->selected());

    // Restore
    select->restoreFormControlState(selectState);
    EXPECT_FALSE(opt0->selected());
    EXPECT_TRUE(opt2->selected());
    EXPECT_TRUE(opt3->selected());
}

} // namespace blink
