1 files changed, 85 insertions, 0 deletions
diff --git a/chromeos/dbus/ibus/mock_ibus_engine_service.h b/chromeos/dbus/ibus/mock_ibus_engine_service.h
index 511650e..128269d 100644
--- a/chromeos/dbus/ibus/mock_ibus_engine_service.h
+++ b/chromeos/dbus/ibus/mock_ibus_engine_service.h
@@ -8,10 +8,41 @@
 #include <string>
 #include "chromeos/dbus/ibus/ibus_engine_service.h"
 
+#include "chromeos/dbus/ibus/ibus_lookup_table.h"
+#include "chromeos/dbus/ibus/ibus_property.h"
+#include "chromeos/dbus/ibus/ibus_text.h"
+
 namespace chromeos {
 
+class IBusText;
+
 class MockIBusEngineService : public IBusEngineService {
  public:
+
+  struct UpdatePreeditArg {
+    UpdatePreeditArg() : is_visible(false) {}
+    IBusText ibus_text;
+    uint32 cursor_pos;
+    bool is_visible;
+  };
+
+  struct UpdateAuxiliaryTextArg {
+    UpdateAuxiliaryTextArg() : is_visible(false) {}
+    IBusText ibus_text;
+    bool is_visible;
+  };
+
+  struct UpdateLookupTableArg {
+    UpdateLookupTableArg() : is_visible(false) {}
+    IBusLookupTable lookup_table;
+    bool is_visible;
+  };
+
+  struct DeleteSurroundingTextArg {
+    int32 offset;
+    uint32 length;
+  };
+
   MockIBusEngineService();
   virtual ~MockIBusEngineService();
 
@@ -37,6 +68,51 @@ class MockIBusEngineService : public IBusEngineService {
 
   IBusEngineHandlerInterface* GetEngine() const;
 
+  void Clear();
+
+  int commit_text_call_count() const { return commit_text_call_count_; }
+  const std::string& last_commit_text() const { return last_commit_text_; }
+
+  int update_preedit_call_count() const { return update_preedit_call_count_; }
+  const UpdatePreeditArg& last_update_preedit_arg() const {
+    return *last_update_preedit_arg_.get();
+  }
+
+  int update_auxiliary_text_call_count() const {
+    return update_auxiliary_text_call_count_;
+  }
+  const UpdateAuxiliaryTextArg& last_update_aux_text_arg() const {
+    return *last_update_aux_text_arg_.get();
+  }
+
+  int update_lookup_table_call_count() const {
+    return update_lookup_table_call_count_;
+  }
+  const UpdateLookupTableArg& last_update_lookup_table_arg() const {
+    return *last_update_lookup_table_arg_.get();
+  }
+
+  int register_properties_call_count() const {
+    return register_properties_call_count_;
+  }
+  const IBusPropertyList& last_registered_properties() const {
+    return *last_registered_properties_.get();
+  }
+
+  int update_property_call_count() const {
+    return update_property_call_count_;
+  }
+  const IBusProperty& last_updated_property() const {
+    return *last_updated_property_.get();
+  }
+
+  int delete_surrounding_text_call_count() const {
+    return delete_surrounding_text_call_count_;
+  }
+  const DeleteSurroundingTextArg& last_delete_surrounding_text_arg() const {
+    return *last_delete_surrounding_text_arg_.get();
+  }
+
  private:
   int register_properties_call_count_;
   int update_preedit_call_count_;
@@ -45,6 +121,15 @@ class MockIBusEngineService : public IBusEngineService {
   int update_property_call_count_;
   int forward_key_event_call_count_;
   int commit_text_call_count_;
+  int delete_surrounding_text_call_count_;
+
+  std::string last_commit_text_;
+  scoped_ptr<UpdatePreeditArg> last_update_preedit_arg_;
+  scoped_ptr<UpdateAuxiliaryTextArg> last_update_aux_text_arg_;
+  scoped_ptr<UpdateLookupTableArg> last_update_lookup_table_arg_;
+  scoped_ptr<IBusPropertyList> last_registered_properties_;
+  scoped_ptr<IBusProperty> last_updated_property_;
+  scoped_ptr<DeleteSurroundingTextArg> last_delete_surrounding_text_arg_;
 
   IBusEngineHandlerInterface* current_engine_;