Support label field in ExtensinoIME API.

chrome/browser/chromeos/input_method/candidate_window_view_unittest.cc

 // Copyright (c) 2012 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 "chrome/browser/chromeos/input_method/candidate_window_view.h"
 
 #include <string>
 
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/chromeos/input_method/candidate_view.h"
@@ skipping 269 matching lines @@
     "Sample Candidate 3"
   };
   const char* kSampleAnnotation[] = {
     "Sample Annotation 1",
     "Sample Annotation 2",
     "Sample Annotation 3"
   };
   const char* kEmptyLabel = "";
   const char* kDefaultVerticalLabel[] = { "1", "2", "3" };
   const char* kDefaultHorizontalLabel[] = { "1.", "2.", "3." };
+  const char* kCustomizedLabel[] = { "a", "s", "d" };
+  const char* kExpectedHorizontalCustomizedLabel[] = { "a.", "s.", "d." };
 
   views::Widget* widget = new views::Widget;
   views::Widget::InitParams params(views::Widget::InitParams::TYPE_WINDOW);
   widget->Init(params);
 
   CandidateWindowView candidate_window_view(widget);
   candidate_window_view.Init();
 
   {
     SCOPED_TRACE("candidate_views allocation test");
@@ skipping 90 matching lines @@
 
     candidate_window_view.UpdateCandidates(table);
 
     ASSERT_EQ(kPageSize, candidate_window_view.candidate_views_.size());
     // Confirm actual labels not containing ".".
     for (size_t i = 0; i < kPageSize; ++i) {
       ExpectLabels(kEmptyLabel, kSampleCandidate[i], kSampleAnnotation[i],
                    candidate_window_view.candidate_views_[i]);
     }
   }
+  {
+    SCOPED_TRACE("Vertical customized label case");
+    const size_t kPageSize = 3;
+    InputMethodLookupTable table;
+    ClearInputMethodLookupTable(kPageSize, &table);
+
+    table.orientation = InputMethodLookupTable::kVertical;
+    for (size_t i = 0; i < kPageSize; ++i) {
+      table.candidates.push_back(kSampleCandidate[i]);
+      table.annotations.push_back(kSampleAnnotation[i]);
+      table.labels.push_back(kCustomizedLabel[i]);
+    }
+
+    candidate_window_view.UpdateCandidates(table);
+
+    ASSERT_EQ(kPageSize, candidate_window_view.candidate_views_.size());
+    // Confirm actual labels not containing ".".
+    for (size_t i = 0; i < kPageSize; ++i) {
+      ExpectLabels(kCustomizedLabel[i],
+                   kSampleCandidate[i],
+                   kSampleAnnotation[i],
+                   candidate_window_view.candidate_views_[i]);
+    }
+  }
+  {
+    SCOPED_TRACE("Horizontal customized label case");
+    const size_t kPageSize = 3;
+    InputMethodLookupTable table;
+    ClearInputMethodLookupTable(kPageSize, &table);
+
+    table.orientation = InputMethodLookupTable::kHorizontal;
+    for (size_t i = 0; i < kPageSize; ++i) {
+      table.candidates.push_back(kSampleCandidate[i]);
+      table.annotations.push_back(kSampleAnnotation[i]);
+      table.labels.push_back(kCustomizedLabel[i]);
+    }
+
+    candidate_window_view.UpdateCandidates(table);
+
+    ASSERT_EQ(kPageSize, candidate_window_view.candidate_views_.size());
+    // Confirm actual labels not containing ".".
+    for (size_t i = 0; i < kPageSize; ++i) {
+      ExpectLabels(kExpectedHorizontalCustomizedLabel[i],
+                   kSampleCandidate[i],
+                   kSampleAnnotation[i],
+                   candidate_window_view.candidate_views_[i]);
+    }
+  }
 
   // We should call CloseNow method, otherwise this test will leak memory.
   widget->CloseNow();
 }
 
 class InfolistWindowViewTest : public views::ViewsTestBase {
 };
 
 TEST_F(InfolistWindowViewTest, ShouldUpdateViewTest) {
   {
@@ skipping 113 matching lines @@
 
   const char kSampleCandidate1[] = "Sample String 1";
   const char kSampleCandidate2[] = "\xE3\x81\x82";  // multi byte string.
   const char kSampleCandidate3[] = ".....";
 
   const char kSampleShortcut1[] = "1";
   const char kSampleShortcut2[] = "b";
   const char kSampleShortcut3[] = "C";
 
   const char kSampleAnnotation1[] = "Sample Annotation 1";
-  const char kSampleAnnotation2[] = "\xE3\x81\x82"; // multi byte string.
+  const char kSampleAnnotation2[] = "\xE3\x81\x82";  // multi byte string.
   const char kSampleAnnotation3[] = "......";
 
   // For testing, we have to prepare empty widget.
   // We should NOT manually free widget by default, otherwise double free will
   // be occurred. So, we should instantiate widget class with "new" operation.
   views::Widget* widget = new views::Widget;
   views::Widget::InitParams params(views::Widget::InitParams::TYPE_WINDOW);
   widget->Init(params);
 
   CandidateWindowView candidate_window_view(widget);
@@ skipping 72 matching lines @@
   for (size_t i = 1; i < candidate_window_view.candidate_views_.size(); ++i) {
     const CandidateView* view = candidate_window_view.candidate_views_[i];
     EXPECT_EQ(before_height, view->GetContentsBounds().height());
   }
 
   // We should call CloseNow method, otherwise this test will leak memory.
   widget->CloseNow();
 }
 }  // namespace input_method
 }  // namespace chromeos