| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "components/gcm_driver/gcm_account_mapper.h" | 5 #include "components/gcm_driver/gcm_account_mapper.h" |
| 6 | 6 |
| 7 #include "base/test/simple_test_clock.h" | 7 #include "base/test/simple_test_clock.h" |
| 8 #include "base/time/time.h" | 8 #include "base/time/time.h" |
| 9 #include "components/gcm_driver/fake_gcm_driver.h" | 9 #include "components/gcm_driver/fake_gcm_driver.h" |
| 10 #include "google_apis/gcm/engine/account_mapping.h" | 10 #include "google_apis/gcm/engine/account_mapping.h" |
| (...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 71 class CustomFakeGCMDriver : public FakeGCMDriver { | 71 class CustomFakeGCMDriver : public FakeGCMDriver { |
| 72 public: | 72 public: |
| 73 enum LastMessageAction { | 73 enum LastMessageAction { |
| 74 NONE, | 74 NONE, |
| 75 SEND_STARTED, | 75 SEND_STARTED, |
| 76 SEND_FINISHED, | 76 SEND_FINISHED, |
| 77 SEND_ACKNOWLEDGED | 77 SEND_ACKNOWLEDGED |
| 78 }; | 78 }; |
| 79 | 79 |
| 80 CustomFakeGCMDriver(); | 80 CustomFakeGCMDriver(); |
| 81 virtual ~CustomFakeGCMDriver(); | 81 ~CustomFakeGCMDriver() override; |
| 82 | 82 |
| 83 virtual void UpdateAccountMapping( | 83 void UpdateAccountMapping(const AccountMapping& account_mapping) override; |
| 84 const AccountMapping& account_mapping) override; | 84 void RemoveAccountMapping(const std::string& account_id) override; |
| 85 virtual void RemoveAccountMapping(const std::string& account_id) override; | 85 void AddAppHandler(const std::string& app_id, |
| 86 virtual void AddAppHandler(const std::string& app_id, | 86 GCMAppHandler* handler) override; |
| 87 GCMAppHandler* handler) override; | 87 void RemoveAppHandler(const std::string& app_id) override; |
| 88 virtual void RemoveAppHandler(const std::string& app_id) override; | 88 void RegisterImpl(const std::string& app_id, |
| 89 virtual void RegisterImpl( | 89 const std::vector<std::string>& sender_ids) override; |
| 90 const std::string& app_id, | |
| 91 const std::vector<std::string>& sender_ids) override; | |
| 92 | 90 |
| 93 void CompleteRegister(const std::string& registration_id, | 91 void CompleteRegister(const std::string& registration_id, |
| 94 GCMClient::Result result); | 92 GCMClient::Result result); |
| 95 void CompleteSend(const std::string& message_id, GCMClient::Result result); | 93 void CompleteSend(const std::string& message_id, GCMClient::Result result); |
| 96 void AcknowledgeSend(const std::string& message_id); | 94 void AcknowledgeSend(const std::string& message_id); |
| 97 void MessageSendError(const std::string& message_id); | 95 void MessageSendError(const std::string& message_id); |
| 98 | 96 |
| 99 void CompleteSendAllMessages(); | 97 void CompleteSendAllMessages(); |
| 100 void AcknowledgeSendAllMessages(); | 98 void AcknowledgeSendAllMessages(); |
| 101 | 99 |
| 102 void SetLastMessageAction(const std::string& message_id, | 100 void SetLastMessageAction(const std::string& message_id, |
| 103 LastMessageAction action); | 101 LastMessageAction action); |
| 104 void Clear(); | 102 void Clear(); |
| 105 | 103 |
| 106 const AccountMapping& last_account_mapping() const { | 104 const AccountMapping& last_account_mapping() const { |
| 107 return account_mapping_; | 105 return account_mapping_; |
| 108 } | 106 } |
| 109 const std::string& last_message_id() const { return last_message_id_; } | 107 const std::string& last_message_id() const { return last_message_id_; } |
| 110 const std::string& last_removed_account_id() const { | 108 const std::string& last_removed_account_id() const { |
| 111 return last_removed_account_id_; | 109 return last_removed_account_id_; |
| 112 } | 110 } |
| 113 LastMessageAction last_action() const { return last_action_; } | 111 LastMessageAction last_action() const { return last_action_; } |
| 114 bool registration_id_requested() const { return registration_id_requested_; } | 112 bool registration_id_requested() const { return registration_id_requested_; } |
| 115 | 113 |
| 116 protected: | 114 protected: |
| 117 virtual void SendImpl(const std::string& app_id, | 115 void SendImpl(const std::string& app_id, |
| 118 const std::string& receiver_id, | 116 const std::string& receiver_id, |
| 119 const GCMClient::OutgoingMessage& message) override; | 117 const GCMClient::OutgoingMessage& message) override; |
| 120 | 118 |
| 121 private: | 119 private: |
| 122 AccountMapping account_mapping_; | 120 AccountMapping account_mapping_; |
| 123 std::string last_message_id_; | 121 std::string last_message_id_; |
| 124 std::string last_removed_account_id_; | 122 std::string last_removed_account_id_; |
| 125 LastMessageAction last_action_; | 123 LastMessageAction last_action_; |
| 126 std::map<std::string, LastMessageAction> all_messages_; | 124 std::map<std::string, LastMessageAction> all_messages_; |
| 127 bool registration_id_requested_; | 125 bool registration_id_requested_; |
| 128 }; | 126 }; |
| 129 | 127 |
| (...skipping 776 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 906 expected_mappings[1].status_change_timestamp = clock()->Now(); | 904 expected_mappings[1].status_change_timestamp = clock()->Now(); |
| 907 expected_mappings[1].status = AccountMapping::MAPPED; | 905 expected_mappings[1].status = AccountMapping::MAPPED; |
| 908 expected_mappings[2].status_change_timestamp = clock()->Now(); | 906 expected_mappings[2].status_change_timestamp = clock()->Now(); |
| 909 expected_mappings[2].status = AccountMapping::MAPPED; | 907 expected_mappings[2].status = AccountMapping::MAPPED; |
| 910 | 908 |
| 911 VerifyMappings( | 909 VerifyMappings( |
| 912 expected_mappings, GetAccounts(), "Step 6, Three mapped accounts"); | 910 expected_mappings, GetAccounts(), "Step 6, Three mapped accounts"); |
| 913 } | 911 } |
| 914 | 912 |
| 915 } // namespace gcm | 913 } // namespace gcm |
| OLD | NEW |