Abstracted AutofillMsg_SetNodeText IPC out of core autofill code.

components/autofill/content/browser/autofill_driver_impl_unittest.cc

 // Copyright 2013 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 <algorithm>
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/strings/utf_string_conversions.h"
@@ skipping 64 matching lines @@
   }
 
   virtual void TearDown() OVERRIDE {
     // Reset the driver now to cause all pref observers to be removed and avoid
     // crashes that otherwise occur in the destructor.
     driver_.reset();
     ChromeRenderViewHostTestHarness::TearDown();
   }
 
  protected:
-  // Searches for an |AutofillMsg_AcceptDataListSuggestion| message in the queue
-  // of sent IPC messages. If none is present, returns false. Otherwise,
-  // extracts the first |AutofillMsg_AcceptDataListSuggestion| message, fills
-  // the output parameter with the value of the message's parameter, and clears
-  // the queue of sent messages.
-  bool GetAcceptDataListSuggestion(
-      base::string16* value) {
-    const uint32 kMsgID = AutofillMsg_AcceptDataListSuggestion::ID;
-    const IPC::Message* message =
-        process()->sink().GetFirstMessageMatching(kMsgID);
-    if (!message)
-      return false;
-    Tuple1<base::string16> autofill_param;
-    AutofillMsg_AcceptDataListSuggestion::Read(message, &autofill_param);
-    if (value)
-      *value = autofill_param.a;
-    process()->sink().ClearMessages();
-    return true;
-  }
-
   // Searches for an |AutofillMsg_FormDataFilled| message in the queue of sent
   // IPC messages. If none is present, returns false. Otherwise, extracts the
   // first |AutofillMsg_FormDataFilled| message, fills the output parameters
   // with the values of the message's parameters, and clears the queue of sent
   // messages.
   bool GetAutofillFormDataFilledMessage(int* page_id, FormData* results) {
     const uint32 kMsgID = AutofillMsg_FormDataFilled::ID;
     const IPC::Message* message =
         process()->sink().GetFirstMessageMatching(kMsgID);
     if (!message)
@@ skipping 23 matching lines @@
     Tuple1<std::vector<FormDataPredictions> > autofill_param;
     AutofillMsg_FieldTypePredictionsAvailable::Read(message, &autofill_param);
     if (predictions)
       *predictions = autofill_param.a;
 
     process()->sink().ClearMessages();
     return true;
   }
 
   // Searches for a message matching |messageID| in the queue of sent IPC
+  // messages. If none is present, returns false. Otherwise, extracts the first
+  // matching message, fills the output parameter with the string16 from the
+  // message's parameter, and clears the queue of sent messages.
+  bool GetString16FromMessageWithID(uint32 messageID, base::string16* value) {
+    const IPC::Message* message =
+        process()->sink().GetFirstMessageMatching(messageID);
+    if (!message)
+      return false;
+    Tuple1<base::string16> autofill_param;
+    switch (messageID) {
+      case AutofillMsg_SetNodeText::ID:
+        AutofillMsg_SetNodeText::Read(message, &autofill_param);
+        break;
+      case AutofillMsg_AcceptDataListSuggestion::ID:
+        AutofillMsg_AcceptDataListSuggestion::Read(message, &autofill_param);
+        break;
+      default:
+        NOTREACHED();
+    }
+    if (value)
+      *value = autofill_param.a;
+    process()->sink().ClearMessages();
+    return true;
+  }
+
+  // Searches for a message matching |messageID| in the queue of sent IPC
   // messages. If none is present, returns false. Otherwise, clears the queue
   // of sent messages and returns true.
   bool HasMessageMatchingID(uint32 messageID) {
     const IPC::Message* message =
         process()->sink().GetFirstMessageMatching(messageID);
     if (!message)
       return false;
     process()->sink().ClearMessages();
     return true;
   }
@@ skipping 67 matching lines @@
   EXPECT_TRUE(HasMessageMatchingID(AutofillMsg_SetAutofillActionPreview::ID));
 }
 
 TEST_F(AutofillDriverImplTest, FillActionSentToRenderer) {
   driver_->SetRendererActionOnFormDataReception(
       AutofillDriver::FORM_DATA_ACTION_FILL);
   EXPECT_TRUE(HasMessageMatchingID(AutofillMsg_SetAutofillActionFill::ID));
 }
 
 TEST_F(AutofillDriverImplTest, AcceptDataListSuggestion) {
-  base::string16 inputValue(UTF8ToUTF16("barfoo"));
-  base::string16 outputValue;
-  driver_->RendererShouldAcceptDataListSuggestion(inputValue);
-  EXPECT_TRUE(GetAcceptDataListSuggestion(&outputValue));
-  EXPECT_EQ(inputValue, outputValue);
+  base::string16 input_value(ASCIIToUTF16("barfoo"));
+  base::string16 output_value;
+  driver_->RendererShouldAcceptDataListSuggestion(input_value);
+  EXPECT_TRUE(GetString16FromMessageWithID(
+      AutofillMsg_AcceptDataListSuggestion::ID,
+      &output_value));
+  EXPECT_EQ(input_value, output_value);
 }
 
 TEST_F(AutofillDriverImplTest, ClearFilledFormSentToRenderer) {
   driver_->RendererShouldClearFilledForm();
   EXPECT_TRUE(HasMessageMatchingID(AutofillMsg_ClearForm::ID));
 }
 
 TEST_F(AutofillDriverImplTest, ClearPreviewedFormSentToRenderer) {
   driver_->RendererShouldClearPreviewedForm();
   EXPECT_TRUE(HasMessageMatchingID(AutofillMsg_ClearPreviewedForm::ID));
 }
 
+TEST_F(AutofillDriverImplTest, SetNodeText) {
+  base::string16 input_value(ASCIIToUTF16("barqux"));
+  base::string16 output_value;
+  driver_->RendererShouldSetNodeText(input_value);
+  EXPECT_TRUE(GetString16FromMessageWithID(AutofillMsg_SetNodeText::ID,
+                                           &output_value));
+  EXPECT_EQ(input_value, output_value);
+}
+
 }  // namespace autofill