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 |