chrome/renderer/autofill/form_classifier_browsertest.cc

// Copyright 2016 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 "base/macros.h"
#include "base/strings/utf_string_conversions.h"
#include "chrome/test/base/chrome_render_view_test.h"
#include "components/autofill/content/renderer/form_classifier.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/WebKit/public/web/WebDocument.h"
#include "third_party/WebKit/public/web/WebFormElement.h"
#include "third_party/WebKit/public/web/WebLocalFrame.h"

namespace autofill {

class FormClassifierTest : public ChromeRenderViewTest {
 public:
  FormClassifierTest() {}

  void TearDown() override {
    LoadHTML("");
    ChromeRenderViewTest::TearDown();
  }

  bool GetGenerationField(std::string* generation_field) {
    blink::WebDocument document = GetMainFrame()->document();
    blink::WebFormElement form =
        document.getElementById("test_form").to<blink::WebFormElement>();
    base::string16 generation_field16;
    bool generation_availalbe =
        ClassifyFormAndFindGenerationField(form, &generation_field16);
    *generation_field = base::UTF16ToUTF8(generation_field16);
    return generation_availalbe;
  }

 private:
  DISALLOW_COPY_AND_ASSIGN(FormClassifierTest);
};

const char kSigninFormHTML[] =
    "<FORM id = 'test_form'> "
    "  <SELECT id='account_type'>"
    "    <OPTION value = 'personal'>"
    "    <OPTION value = 'corporate'>"
    "  </SELECT>"
    "  <INPUT type = 'text' id = 'username'/>"
    "  <INPUT type = 'password' id = 'password'/>"
    "  <INPUT type = 'checkbox' id = 'remember_me'/>"
    "  <INPUT type = 'checkbox' id = 'secure_login'/>"
    "  <INPUT type = 'submit' id = 'signin' />"
    "  <INPUT type = 'hidden' id = 'ignore_this' />"
    "  <INPUT type = 'hidden' id = 'ignore_this_too' />"
    "  <INPUT type = 'submit' id = 'submit' />"
    "</FORM>";

const char kSignupFormWithSeveralTextFieldsFormHTML[] =
    "<FORM id = 'test_form'> "
    "  <INPUT type = 'text' id = 'full_name'/>"
    "  <INPUT type = 'text' id = 'username'/>"
    "  <INPUT type = 'password' id = 'password'/>"
    "  <INPUT type = 'submit' id = 'submit' />"
    "</FORM>";

const char kSignupFormWithSeveralPasswordFieldsHTML[] =
    "<FORM id = 'test_form'> "
    "  <INPUT type = 'text' id = 'username'/>"
    "  <INPUT type = 'password' id = 'password'/>"
    "  <INPUT type = 'password' id = 'confirm_password'/>"
    "  <INPUT type = 'submit' id = 'submit' />"
    "</FORM>";

const char kSignupFormWithManyCheckboxesHTML[] =
    "<FORM id = 'test_form'> "
    "  </SELECT>"
    "  <INPUT type = 'text' id = 'username' />"
    "  <INPUT type = 'password' id = 'password' />"
    "  <INPUT type = 'checkbox' id = 'subscribe_science' />"
    "  <INPUT type = 'checkbox' id = 'subscribe_music' />"
    "  <INPUT type = 'checkbox' id = 'subscribe_sport' />"
    "  <INPUT type = 'submit' id = 'submit' />"
    "</FORM>";

const char kSignupFormWithOtherFieldsHTML[] =
    "<FORM id = 'test_form'> "
    "  <INPUT type = 'text' id = 'username' />"
    "  <INPUT type = 'password' id = 'password' />"
    "  <INPUT type = 'color' id = 'account_color' />"
    "  <INPUT type = 'date' id = 'date_of_birth' />"
    "  <INPUT type = 'submit' id = 'submit' />"
    "</FORM>";

const char kSignupFormWithTextFeatureInInputElementHTML[] =
    "<FORM id = 'test_form'> "
    "  <INPUT type = 'text' id = 'username' class = 'sign-up_field' />"
    "  <INPUT type = 'password' id = 'password' class = 'sign-up_field' />"
    "  <INPUT type = 'submit' id = 'submit' />"
    "</FORM>";

const char kSignupFormWithTextFeatureInFormTagHTML[] =
    "<FORM id = 'test_form' some_attribute='sign_up_form' > "
    "  <INPUT type = 'text' id = 'username' />"
    "  <INPUT type = 'password' id = 'password' />"
    "  <INPUT type = 'submit' id = 'submit' />"
    "</FORM>";

const char kSigninFormWithInvisibleFieldsHTML[] =
    "<FORM id = 'test_form'> "
    "  <INPUT type = 'text' id = 'username' />"
    "  <INPUT type = 'password' id = 'password' />"
    "  <INPUT type = 'input' hidden id = 'hidden_field1' />"
    "  <INPUT type = 'password' hidden id = 'hidden_field2'/>"
    "  <INPUT type = 'submit' id = 'submit' />"
    "</FORM>";

const char kSignupFormWithSigninButtonHTML[] =
    "<FORM id = 'test_form' >"
    "  <INPUT type = 'text' id = 'username' />"
    "  <INPUT type = 'password' id = 'password' />"
    "  <INPUT type = 'password' id = 'confirm_password' />"
    "  <INPUT type = 'submit' id = 'submit' />"
    "  <INPUT type = 'button' id = 'goto_signin_form' />"
    "  <INPUT type = 'image' id = 'goto_auth_form' />"
    "</FORM>";

const char kSomeFormWithoutPasswordFields[] =
    "<FORM id = 'test_form' >"
    "  <INPUT type = 'text' id = 'username' />"
    "  <INPUT type = 'text' id = 'fullname' />"
    "  <INPUT type = 'text' id = 'address' />"
    "  <INPUT type = 'text' id = 'phone' />"
    "  <INPUT type = 'submit' id = 'submit' />"
    "</FORM>";

const char kSignupFormWithSigninTextFeatureAndManyFieldsHTML[] =
    "<FORM id = 'test_form' class = 'log-on_container'> "
    "  <INPUT type = 'text' id = 'fullname' />"
    "  <INPUT type = 'text' id = 'username' />"
    "  <INPUT type = 'password' id = 'password' />"
    "  <INPUT type = 'submit' id = 'submit' />"
    "</FORM>";

const char kChangeFormWithTreePasswordFieldsHTML[] =
    "<FORM id = 'test_form' >"
    "  <INPUT type = 'password' id = 'old_password' />"
    "  <INPUT type = 'password' id = 'password' />"
    "  <INPUT type = 'password' id = 'confirm_password' />"
    "  <INPUT type = 'submit' id = 'submit' />"
    "</FORM>";

TEST_F(FormClassifierTest, SigninForm) {
  // Signin form with as many as possible visible elements,
  // i.e. if one more text/password/checkbox/other field is added, the form
  // will be recognized as a signup form.
  LoadHTML(kSigninFormHTML);
  std::string generation_field;
  EXPECT_FALSE(GetGenerationField(&generation_field));
}

TEST_F(FormClassifierTest, SignupFormWithSeveralTextFields) {
  LoadHTML(kSignupFormWithSeveralTextFieldsFormHTML);
  std::string generation_field;
  EXPECT_TRUE(GetGenerationField(&generation_field));
  EXPECT_EQ("password", generation_field);
}

TEST_F(FormClassifierTest, SignupFormWithSeveralPasswordFieldsHTML) {
  LoadHTML(kSignupFormWithSeveralPasswordFieldsHTML);
  std::string generation_field;
  EXPECT_TRUE(GetGenerationField(&generation_field));
  EXPECT_EQ("password", generation_field);
}

TEST_F(FormClassifierTest, SignupFormWithManyCheckboxesHTML) {
  LoadHTML(kSignupFormWithManyCheckboxesHTML);
  std::string generation_field;
  EXPECT_TRUE(GetGenerationField(&generation_field));
  EXPECT_EQ("password", generation_field);
}

TEST_F(FormClassifierTest, SignupFormWithOtherFieldsHTML) {
  LoadHTML(kSignupFormWithOtherFieldsHTML);
  std::string generation_field;
  EXPECT_TRUE(GetGenerationField(&generation_field));
  EXPECT_EQ("password", generation_field);
}

TEST_F(FormClassifierTest, SignupFormWithTextFeatureInInputElementHTML) {
  LoadHTML(kSignupFormWithTextFeatureInInputElementHTML);
  std::string generation_field;
  EXPECT_TRUE(GetGenerationField(&generation_field));
  EXPECT_EQ("password", generation_field);
}

TEST_F(FormClassifierTest, SigninFormWithTextFeatureInFormTagHTML) {
  LoadHTML(kSignupFormWithTextFeatureInFormTagHTML);
  std::string generation_field;
  EXPECT_TRUE(GetGenerationField(&generation_field));
  EXPECT_EQ("password", generation_field);
}

TEST_F(FormClassifierTest, SigninFormWithInvisibleFieldsHTML) {
  LoadHTML(kSigninFormWithInvisibleFieldsHTML);
  std::string generation_field;
  EXPECT_FALSE(GetGenerationField(&generation_field));
}

TEST_F(FormClassifierTest, SignupFormWithSigninButtonHTML) {
  LoadHTML(kSignupFormWithSigninButtonHTML);
  std::string generation_field;
  EXPECT_TRUE(GetGenerationField(&generation_field));
  EXPECT_EQ("password", generation_field);
}

TEST_F(FormClassifierTest, SomeFormWithoutPasswordFields) {
  LoadHTML(kSomeFormWithoutPasswordFields);
  std::string generation_field;
  EXPECT_FALSE(GetGenerationField(&generation_field));
}

TEST_F(FormClassifierTest, SignupFormWithSigninTextFeatureAndManyFieldsHTML) {
  // Even if there is signin text feature, the number of fields is more reliable
  // signal of signup form. So, this form should be classified as signup.
  LoadHTML(kSignupFormWithSigninTextFeatureAndManyFieldsHTML);
  std::string generation_field;
  EXPECT_TRUE(GetGenerationField(&generation_field));
  EXPECT_EQ("password", generation_field);
}

TEST_F(FormClassifierTest, kChangeFormWithTreePasswordFieldsHTML) {
  LoadHTML(kChangeFormWithTreePasswordFieldsHTML);
  std::string generation_field;
  EXPECT_TRUE(GetGenerationField(&generation_field));
  EXPECT_EQ("password", generation_field);
}

}  // namespace autofill