| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 // Mock ServerConnectionManager class for use in client unit tests. | 5 // Mock ServerConnectionManager class for use in client unit tests. |
| 6 | 6 |
| 7 #ifndef CHROME_TEST_SYNC_ENGINE_MOCK_CONNECTION_MANAGER_H_ | 7 #ifndef CHROME_TEST_SYNC_ENGINE_MOCK_CONNECTION_MANAGER_H_ |
| 8 #define CHROME_TEST_SYNC_ENGINE_MOCK_CONNECTION_MANAGER_H_ | 8 #define CHROME_TEST_SYNC_ENGINE_MOCK_CONNECTION_MANAGER_H_ |
| 9 #pragma once | 9 #pragma once |
| 10 | 10 |
| (...skipping 25 matching lines...) Expand all Loading... |
| 36 | 36 |
| 37 protected: | 37 protected: |
| 38 virtual ~MidCommitObserver() {} | 38 virtual ~MidCommitObserver() {} |
| 39 }; | 39 }; |
| 40 | 40 |
| 41 MockConnectionManager(syncable::DirectoryManager* dirmgr, | 41 MockConnectionManager(syncable::DirectoryManager* dirmgr, |
| 42 const std::string& name); | 42 const std::string& name); |
| 43 virtual ~MockConnectionManager(); | 43 virtual ~MockConnectionManager(); |
| 44 | 44 |
| 45 // Overridden ServerConnectionManager functions. | 45 // Overridden ServerConnectionManager functions. |
| 46 virtual bool PostBufferToPath(const PostBufferParams*, | 46 virtual bool PostBufferToPath( |
| 47 const string& path, | 47 const PostBufferParams*, |
| 48 const string& auth_token, | 48 const std::string& path, |
| 49 const std::string& auth_token, |
| 49 browser_sync::ScopedServerStatusWatcher* watcher); | 50 browser_sync::ScopedServerStatusWatcher* watcher); |
| 50 | 51 |
| 51 virtual bool IsServerReachable(); | 52 virtual bool IsServerReachable(); |
| 52 virtual bool IsUserAuthenticated(); | 53 virtual bool IsUserAuthenticated(); |
| 53 | 54 |
| 54 // Control of commit response. | 55 // Control of commit response. |
| 55 void SetMidCommitCallback(Callback0::Type* callback); | 56 void SetMidCommitCallback(Callback0::Type* callback); |
| 56 void SetMidCommitObserver(MidCommitObserver* observer); | 57 void SetMidCommitObserver(MidCommitObserver* observer); |
| 57 | 58 |
| 58 // Set this if you want commit to perform commit time rename. Will request | 59 // Set this if you want commit to perform commit time rename. Will request |
| 59 // that the client renames all commited entries, prepending this string. | 60 // that the client renames all commited entries, prepending this string. |
| 60 void SetCommitTimeRename(string prepend); | 61 void SetCommitTimeRename(std::string prepend); |
| 61 | 62 |
| 62 // Generic versions of AddUpdate functions. Tests using these function should | 63 // Generic versions of AddUpdate functions. Tests using these function should |
| 63 // compile for both the int64 and string id based versions of the server. | 64 // compile for both the int64 and string id based versions of the server. |
| 64 // The SyncEntity returned is only valid until the Sync is completed | 65 // The SyncEntity returned is only valid until the Sync is completed |
| 65 // (e.g. with SyncShare.) It allows to add further entity properties before | 66 // (e.g. with SyncShare.) It allows to add further entity properties before |
| 66 // sync, using SetLastXXX() methods and/or GetMutableLastUpdate(). | 67 // sync, using SetLastXXX() methods and/or GetMutableLastUpdate(). |
| 67 sync_pb::SyncEntity* AddUpdateDirectory(syncable::Id id, | 68 sync_pb::SyncEntity* AddUpdateDirectory(syncable::Id id, |
| 68 syncable::Id parent_id, | 69 syncable::Id parent_id, |
| 69 string name, | 70 std::string name, |
| 70 int64 version, | 71 int64 version, |
| 71 int64 sync_ts); | 72 int64 sync_ts); |
| 72 sync_pb::SyncEntity* AddUpdateBookmark(syncable::Id id, | 73 sync_pb::SyncEntity* AddUpdateBookmark(syncable::Id id, |
| 73 syncable::Id parent_id, | 74 syncable::Id parent_id, |
| 74 string name, | 75 std::string name, |
| 75 int64 version, | 76 int64 version, |
| 76 int64 sync_ts); | 77 int64 sync_ts); |
| 77 // Versions of the AddUpdate functions that accept integer IDs. | 78 // Versions of the AddUpdate functions that accept integer IDs. |
| 78 sync_pb::SyncEntity* AddUpdateDirectory(int id, | 79 sync_pb::SyncEntity* AddUpdateDirectory(int id, |
| 79 int parent_id, | 80 int parent_id, |
| 80 string name, | 81 std::string name, |
| 81 int64 version, | 82 int64 version, |
| 82 int64 sync_ts); | 83 int64 sync_ts); |
| 83 sync_pb::SyncEntity* AddUpdateBookmark(int id, | 84 sync_pb::SyncEntity* AddUpdateBookmark(int id, |
| 84 int parent_id, | 85 int parent_id, |
| 85 string name, | 86 std::string name, |
| 86 int64 version, | 87 int64 version, |
| 87 int64 sync_ts); | 88 int64 sync_ts); |
| 88 // New protocol versions of the AddUpdate functions. | 89 // New protocol versions of the AddUpdate functions. |
| 89 sync_pb::SyncEntity* AddUpdateDirectory(string id, | 90 sync_pb::SyncEntity* AddUpdateDirectory(std::string id, |
| 90 string parent_id, | 91 std::string parent_id, |
| 91 string name, | 92 std::string name, |
| 92 int64 version, | 93 int64 version, |
| 93 int64 sync_ts); | 94 int64 sync_ts); |
| 94 sync_pb::SyncEntity* AddUpdateBookmark(string id, | 95 sync_pb::SyncEntity* AddUpdateBookmark(std::string id, |
| 95 string parent_id, | 96 std::string parent_id, |
| 96 string name, | 97 std::string name, |
| 97 int64 version, | 98 int64 version, |
| 98 int64 sync_ts); | 99 int64 sync_ts); |
| 99 | 100 |
| 100 // Find the last commit sent by the client, and replay it for the next get | 101 // Find the last commit sent by the client, and replay it for the next get |
| 101 // updates command. This can be used to simulate the GetUpdates that happens | 102 // updates command. This can be used to simulate the GetUpdates that happens |
| 102 // immediately after a successful commit. | 103 // immediately after a successful commit. |
| 103 sync_pb::SyncEntity* AddUpdateFromLastCommit(); | 104 sync_pb::SyncEntity* AddUpdateFromLastCommit(); |
| 104 | 105 |
| 105 // Add a deleted item. Deletion records typically contain no | 106 // Add a deleted item. Deletion records typically contain no |
| 106 // additional information beyond the deletion, and no specifics. | 107 // additional information beyond the deletion, and no specifics. |
| 107 // The server may send the originator fields. | 108 // The server may send the originator fields. |
| 108 void AddUpdateTombstone(const syncable::Id& id); | 109 void AddUpdateTombstone(const syncable::Id& id); |
| 109 | 110 |
| 110 void SetLastUpdateDeleted(); | 111 void SetLastUpdateDeleted(); |
| 111 void SetLastUpdateServerTag(const string& tag); | 112 void SetLastUpdateServerTag(const std::string& tag); |
| 112 void SetLastUpdateClientTag(const string& tag); | 113 void SetLastUpdateClientTag(const std::string& tag); |
| 113 void SetLastUpdateOriginatorFields(const string& client_id, | 114 void SetLastUpdateOriginatorFields(const std::string& client_id, |
| 114 const string& entry_id); | 115 const std::string& entry_id); |
| 115 void SetLastUpdatePosition(int64 position_in_parent); | 116 void SetLastUpdatePosition(int64 position_in_parent); |
| 116 void SetNewTimestamp(int ts); | 117 void SetNewTimestamp(int ts); |
| 117 void SetChangesRemaining(int64 count); | 118 void SetChangesRemaining(int64 count); |
| 118 | 119 |
| 119 // Add a new batch of updates after the current one. Allows multiple | 120 // Add a new batch of updates after the current one. Allows multiple |
| 120 // GetUpdates responses to be buffered up, since the syncer may | 121 // GetUpdates responses to be buffered up, since the syncer may |
| 121 // issue multiple requests during a sync cycle. | 122 // issue multiple requests during a sync cycle. |
| 122 void NextUpdateBatch(); | 123 void NextUpdateBatch(); |
| 123 | 124 |
| 124 // For AUTHENTICATE responses. | 125 // For AUTHENTICATE responses. |
| (...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 181 next_new_id_ = value; | 182 next_new_id_ = value; |
| 182 } | 183 } |
| 183 void set_conflict_n_commits(int value) { | 184 void set_conflict_n_commits(int value) { |
| 184 conflict_n_commits_ = value; | 185 conflict_n_commits_ = value; |
| 185 } | 186 } |
| 186 | 187 |
| 187 void set_use_legacy_bookmarks_protocol(bool value) { | 188 void set_use_legacy_bookmarks_protocol(bool value) { |
| 188 use_legacy_bookmarks_protocol_ = value; | 189 use_legacy_bookmarks_protocol_ = value; |
| 189 } | 190 } |
| 190 | 191 |
| 191 void set_store_birthday(string new_birthday) { | 192 void set_store_birthday(std::string new_birthday) { |
| 192 // Multiple threads can set store_birthday_ in our tests, need to lock it to | 193 // Multiple threads can set store_birthday_ in our tests, need to lock it to |
| 193 // ensure atomic read/writes and avoid race conditions. | 194 // ensure atomic read/writes and avoid race conditions. |
| 194 base::AutoLock lock(store_birthday_lock_); | 195 base::AutoLock lock(store_birthday_lock_); |
| 195 store_birthday_ = new_birthday; | 196 store_birthday_ = new_birthday; |
| 196 } | 197 } |
| 197 | 198 |
| 198 // Retrieve the number of GetUpdates requests that the mock server has | 199 // Retrieve the number of GetUpdates requests that the mock server has |
| 199 // seen since the last time this function was called. Can be used to | 200 // seen since the last time this function was called. Can be used to |
| 200 // verify that a GetUpdates actually did or did not happen after running | 201 // verify that a GetUpdates actually did or did not happen after running |
| 201 // the syncer. | 202 // the syncer. |
| (...skipping 19 matching lines...) Expand all Loading... |
| 221 const std::string& store_birthday() { | 222 const std::string& store_birthday() { |
| 222 base::AutoLock lock(store_birthday_lock_); | 223 base::AutoLock lock(store_birthday_lock_); |
| 223 return store_birthday_; | 224 return store_birthday_; |
| 224 } | 225 } |
| 225 | 226 |
| 226 // Locate the most recent update message for purpose of alteration. | 227 // Locate the most recent update message for purpose of alteration. |
| 227 sync_pb::SyncEntity* GetMutableLastUpdate(); | 228 sync_pb::SyncEntity* GetMutableLastUpdate(); |
| 228 | 229 |
| 229 private: | 230 private: |
| 230 sync_pb::SyncEntity* AddUpdateFull(syncable::Id id, syncable::Id parentid, | 231 sync_pb::SyncEntity* AddUpdateFull(syncable::Id id, syncable::Id parentid, |
| 231 string name, int64 version, | 232 std::string name, int64 version, |
| 232 int64 sync_ts, | 233 int64 sync_ts, |
| 233 bool is_dir); | 234 bool is_dir); |
| 234 sync_pb::SyncEntity* AddUpdateFull(string id, string parentid, string name, | 235 sync_pb::SyncEntity* AddUpdateFull(std::string id, |
| 236 std::string parentid, std::string name, |
| 235 int64 version, int64 sync_ts, | 237 int64 version, int64 sync_ts, |
| 236 bool is_dir); | 238 bool is_dir); |
| 237 // Functions to handle the various types of server request. | 239 // Functions to handle the various types of server request. |
| 238 void ProcessGetUpdates(sync_pb::ClientToServerMessage* csm, | 240 void ProcessGetUpdates(sync_pb::ClientToServerMessage* csm, |
| 239 sync_pb::ClientToServerResponse* response); | 241 sync_pb::ClientToServerResponse* response); |
| 240 void ProcessAuthenticate(sync_pb::ClientToServerMessage* csm, | 242 void ProcessAuthenticate(sync_pb::ClientToServerMessage* csm, |
| 241 sync_pb::ClientToServerResponse* response, | 243 sync_pb::ClientToServerResponse* response, |
| 242 const std::string& auth_token); | 244 const std::string& auth_token); |
| 243 void ProcessCommit(sync_pb::ClientToServerMessage* csm, | 245 void ProcessCommit(sync_pb::ClientToServerMessage* csm, |
| 244 sync_pb::ClientToServerResponse* response_buffer); | 246 sync_pb::ClientToServerResponse* response_buffer); |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 277 int conflict_n_commits_; | 279 int conflict_n_commits_; |
| 278 | 280 |
| 279 // Commit messages we've sent, and responses we've returned. | 281 // Commit messages we've sent, and responses we've returned. |
| 280 ScopedVector<sync_pb::CommitMessage> commit_messages_; | 282 ScopedVector<sync_pb::CommitMessage> commit_messages_; |
| 281 ScopedVector<sync_pb::CommitResponse> commit_responses_; | 283 ScopedVector<sync_pb::CommitResponse> commit_responses_; |
| 282 | 284 |
| 283 // The next id the mock will return to a commit. | 285 // The next id the mock will return to a commit. |
| 284 int next_new_id_; | 286 int next_new_id_; |
| 285 | 287 |
| 286 // The store birthday we send to the client. | 288 // The store birthday we send to the client. |
| 287 string store_birthday_; | 289 std::string store_birthday_; |
| 288 base::Lock store_birthday_lock_; | 290 base::Lock store_birthday_lock_; |
| 289 bool store_birthday_sent_; | 291 bool store_birthday_sent_; |
| 290 bool client_stuck_; | 292 bool client_stuck_; |
| 291 string commit_time_rename_prepended_string_; | 293 std::string commit_time_rename_prepended_string_; |
| 292 | 294 |
| 293 // Fail on the next call to PostBufferToPath(). | 295 // Fail on the next call to PostBufferToPath(). |
| 294 bool fail_next_postbuffer_; | 296 bool fail_next_postbuffer_; |
| 295 | 297 |
| 296 // Our directory. | 298 // Our directory. |
| 297 syncable::DirectoryManager* directory_manager_; | 299 syncable::DirectoryManager* directory_manager_; |
| 298 std::string directory_name_; | 300 std::string directory_name_; |
| 299 | 301 |
| 300 // The updates we'll return to the next request. | 302 // The updates we'll return to the next request. |
| 301 std::list<sync_pb::GetUpdatesResponse> update_queue_; | 303 std::list<sync_pb::GetUpdatesResponse> update_queue_; |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 340 int num_get_updates_requests_; | 342 int num_get_updates_requests_; |
| 341 | 343 |
| 342 std::string next_token_; | 344 std::string next_token_; |
| 343 | 345 |
| 344 sync_pb::ClientToServerMessage last_request_; | 346 sync_pb::ClientToServerMessage last_request_; |
| 345 | 347 |
| 346 DISALLOW_COPY_AND_ASSIGN(MockConnectionManager); | 348 DISALLOW_COPY_AND_ASSIGN(MockConnectionManager); |
| 347 }; | 349 }; |
| 348 | 350 |
| 349 #endif // CHROME_TEST_SYNC_ENGINE_MOCK_CONNECTION_MANAGER_H_ | 351 #endif // CHROME_TEST_SYNC_ENGINE_MOCK_CONNECTION_MANAGER_H_ |
| OLD | NEW |