src/sync/tools/testserver/chromiumsync_test.py

   1 #!/usr/bin/env python
   2 # Copyright 2013 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 """Tests exercising chromiumsync and SyncDataModel."""
   7
   8 import pickle
   9 import unittest
  10
  11 import autofill_specifics_pb2
  12 import bookmark_specifics_pb2
  13 import chromiumsync
  14 import managed_user_specifics_pb2
  15 import sync_pb2
  16 import theme_specifics_pb2
  17
  18 class SyncDataModelTest(unittest.TestCase):
  19   def setUp(self):
  20     self.model = chromiumsync.SyncDataModel()
  21     # The Synced Bookmarks folder is not created by default
  22     self._expect_synced_bookmarks_folder = False
  23
  24   def AddToModel(self, proto):
  25     self.model._entries[proto.id_string] = proto
  26
  27   def GetChangesFromTimestamp(self, requested_types, timestamp):
  28     message = sync_pb2.GetUpdatesMessage()
  29     message.from_timestamp = timestamp
  30     for data_type in requested_types:
  31       getattr(message.requested_types,
  32               chromiumsync.SYNC_TYPE_TO_DESCRIPTOR[
  33                   data_type].name).SetInParent()
  34     return self.model.GetChanges(
  35         chromiumsync.UpdateSieve(message, self.model.migration_history))
  36
  37   def FindMarkerByNumber(self, markers, datatype):
  38     """Search a list of progress markers and find the one for a datatype."""
  39     for marker in markers:
  40       if marker.data_type_id == datatype.number:
  41         return marker
  42     self.fail('Required marker not found: %s' % datatype.name)
  43
  44   def testPermanentItemSpecs(self):
  45     specs = chromiumsync.SyncDataModel._PERMANENT_ITEM_SPECS
  46
  47     declared_specs = set(['0'])
  48     for spec in specs:
  49       self.assertTrue(spec.parent_tag in declared_specs, 'parent tags must '
  50                       'be declared before use')
  51       declared_specs.add(spec.tag)
  52
  53     unique_datatypes = set([x.sync_type for x in specs])
  54     self.assertEqual(unique_datatypes,
  55                      set(chromiumsync.ALL_TYPES[1:]),
  56                      'Every sync datatype should have a permanent folder '
  57                      'associated with it')
  58
  59   def testSaveEntry(self):
  60     proto = sync_pb2.SyncEntity()
  61     proto.id_string = 'abcd'
  62     proto.version = 0
  63     self.assertFalse(self.model._ItemExists(proto.id_string))
  64     self.model._SaveEntry(proto)
  65     self.assertEqual(1, proto.version)
  66     self.assertTrue(self.model._ItemExists(proto.id_string))
  67     self.model._SaveEntry(proto)
  68     self.assertEqual(2, proto.version)
  69     proto.version = 0
  70     self.assertTrue(self.model._ItemExists(proto.id_string))
  71     self.assertEqual(2, self.model._entries[proto.id_string].version)
  72
  73   def testCreatePermanentItems(self):
  74     self.model._CreateDefaultPermanentItems(chromiumsync.ALL_TYPES)
  75     self.assertEqual(len(chromiumsync.ALL_TYPES) + 1,
  76                      len(self.model._entries))
  77
  78   def ExpectedPermanentItemCount(self, sync_type):
  79     if sync_type == chromiumsync.BOOKMARK:
  80       if self._expect_synced_bookmarks_folder:
  81         return 4
  82       else:
  83         return 3
  84     else:
  85       return 1
  86
  87   def testGetChangesFromTimestampZeroForEachType(self):
  88     all_types = chromiumsync.ALL_TYPES[1:]
  89     for sync_type in all_types:
  90       self.model = chromiumsync.SyncDataModel()
  91       request_types = [sync_type]
  92
  93       version, changes, remaining = (
  94           self.GetChangesFromTimestamp(request_types, 0))
  95
  96       expected_count = self.ExpectedPermanentItemCount(sync_type)
  97       self.assertEqual(expected_count, version)
  98       self.assertEqual(expected_count, len(changes))
  99       for change in changes:
 100         self.assertTrue(change.HasField('server_defined_unique_tag'))
 101         self.assertEqual(change.version, change.sync_timestamp)
 102         self.assertTrue(change.version <= version)
 103
 104       # Test idempotence: another GetUpdates from ts=0 shouldn't recreate.
 105       version, changes, remaining = (
 106           self.GetChangesFromTimestamp(request_types, 0))
 107       self.assertEqual(expected_count, version)
 108       self.assertEqual(expected_count, len(changes))
 109       self.assertEqual(0, remaining)
 110
 111       # Doing a wider GetUpdates from timestamp zero shouldn't recreate either.
 112       new_version, changes, remaining = (
 113           self.GetChangesFromTimestamp(all_types, 0))
 114       if self._expect_synced_bookmarks_folder:
 115         self.assertEqual(len(chromiumsync.SyncDataModel._PERMANENT_ITEM_SPECS),
 116                          new_version)
 117       else:
 118         self.assertEqual(
 119             len(chromiumsync.SyncDataModel._PERMANENT_ITEM_SPECS) -1,
 120             new_version)
 121       self.assertEqual(new_version, len(changes))
 122       self.assertEqual(0, remaining)
 123       version, changes, remaining = (
 124           self.GetChangesFromTimestamp(request_types, 0))
 125       self.assertEqual(new_version, version)
 126       self.assertEqual(expected_count, len(changes))
 127       self.assertEqual(0, remaining)
 128
 129   def testBatchSize(self):
 130     for sync_type in chromiumsync.ALL_TYPES[1:]:
 131       specifics = chromiumsync.GetDefaultEntitySpecifics(sync_type)
 132       self.model = chromiumsync.SyncDataModel()
 133       request_types = [sync_type]
 134
 135       for i in range(self.model._BATCH_SIZE*3):
 136         entry = sync_pb2.SyncEntity()
 137         entry.id_string = 'batch test %d' % i
 138         entry.specifics.CopyFrom(specifics)
 139         self.model._SaveEntry(entry)
 140       last_bit = self.ExpectedPermanentItemCount(sync_type)
 141       version, changes, changes_remaining = (
 142           self.GetChangesFromTimestamp(request_types, 0))
 143       self.assertEqual(self.model._BATCH_SIZE, version)
 144       self.assertEqual(self.model._BATCH_SIZE*2 + last_bit, changes_remaining)
 145       version, changes, changes_remaining = (
 146           self.GetChangesFromTimestamp(request_types, version))
 147       self.assertEqual(self.model._BATCH_SIZE*2, version)
 148       self.assertEqual(self.model._BATCH_SIZE + last_bit, changes_remaining)
 149       version, changes, changes_remaining = (
 150           self.GetChangesFromTimestamp(request_types, version))
 151       self.assertEqual(self.model._BATCH_SIZE*3, version)
 152       self.assertEqual(last_bit, changes_remaining)
 153       version, changes, changes_remaining = (
 154           self.GetChangesFromTimestamp(request_types, version))
 155       self.assertEqual(self.model._BATCH_SIZE*3 + last_bit, version)
 156       self.assertEqual(0, changes_remaining)
 157
 158       # Now delete a third of the items.
 159       for i in xrange(self.model._BATCH_SIZE*3 - 1, 0, -3):
 160         entry = sync_pb2.SyncEntity()
 161         entry.id_string = 'batch test %d' % i
 162         entry.deleted = True
 163         self.model._SaveEntry(entry)
 164
 165       # The batch counts shouldn't change.
 166       version, changes, changes_remaining = (
 167           self.GetChangesFromTimestamp(request_types, 0))
 168       self.assertEqual(self.model._BATCH_SIZE, len(changes))
 169       self.assertEqual(self.model._BATCH_SIZE*2 + last_bit, changes_remaining)
 170       version, changes, changes_remaining = (
 171           self.GetChangesFromTimestamp(request_types, version))
 172       self.assertEqual(self.model._BATCH_SIZE, len(changes))
 173       self.assertEqual(self.model._BATCH_SIZE + last_bit, changes_remaining)
 174       version, changes, changes_remaining = (
 175           self.GetChangesFromTimestamp(request_types, version))
 176       self.assertEqual(self.model._BATCH_SIZE, len(changes))
 177       self.assertEqual(last_bit, changes_remaining)
 178       version, changes, changes_remaining = (
 179           self.GetChangesFromTimestamp(request_types, version))
 180       self.assertEqual(last_bit, len(changes))
 181       self.assertEqual(self.model._BATCH_SIZE*4 + last_bit, version)
 182       self.assertEqual(0, changes_remaining)
 183
 184   def testCommitEachDataType(self):
 185     for sync_type in chromiumsync.ALL_TYPES[1:]:
 186       specifics = chromiumsync.GetDefaultEntitySpecifics(sync_type)
 187       self.model = chromiumsync.SyncDataModel()
 188       my_cache_guid = '112358132134'
 189       parent = 'foobar'
 190       commit_session = {}
 191
 192       # Start with a GetUpdates from timestamp 0, to populate permanent items.
 193       original_version, original_changes, changes_remaining = (
 194           self.GetChangesFromTimestamp([sync_type], 0))
 195
 196       def DoCommit(original=None, id_string='', name=None, parent=None,
 197                    position=0):
 198         proto = sync_pb2.SyncEntity()
 199         if original is not None:
 200           proto.version = original.version
 201           proto.id_string = original.id_string
 202           proto.parent_id_string = original.parent_id_string
 203           proto.name = original.name
 204         else:
 205           proto.id_string = id_string
 206           proto.version = 0
 207         proto.specifics.CopyFrom(specifics)
 208         if name is not None:
 209           proto.name = name
 210         if parent:
 211           proto.parent_id_string = parent.id_string
 212         proto.insert_after_item_id = 'please discard'
 213         proto.position_in_parent = position
 214         proto.folder = True
 215         proto.deleted = False
 216         result = self.model.CommitEntry(proto, my_cache_guid, commit_session)
 217         self.assertTrue(result)
 218         return (proto, result)
 219
 220       # Commit a new item.
 221       proto1, result1 = DoCommit(name='namae', id_string='Foo',
 222                                  parent=original_changes[-1], position=100)
 223       # Commit an item whose parent is another item (referenced via the
 224       # pre-commit ID).
 225       proto2, result2 = DoCommit(name='Secondo', id_string='Bar',
 226                                  parent=proto1, position=-100)
 227         # Commit a sibling of the second item.
 228       proto3, result3 = DoCommit(name='Third!', id_string='Baz',
 229                                  parent=proto1, position=-50)
 230
 231       self.assertEqual(3, len(commit_session))
 232       for p, r in [(proto1, result1), (proto2, result2), (proto3, result3)]:
 233         self.assertNotEqual(r.id_string, p.id_string)
 234         self.assertEqual(r.originator_client_item_id, p.id_string)
 235         self.assertEqual(r.originator_cache_guid, my_cache_guid)
 236         self.assertTrue(r is not self.model._entries[r.id_string],
 237                         "Commit result didn't make a defensive copy.")
 238         self.assertTrue(p is not self.model._entries[r.id_string],
 239                         "Commit result didn't make a defensive copy.")
 240         self.assertEqual(commit_session.get(p.id_string), r.id_string)
 241         self.assertTrue(r.version > original_version)
 242       self.assertEqual(result1.parent_id_string, proto1.parent_id_string)
 243       self.assertEqual(result2.parent_id_string, result1.id_string)
 244       version, changes, remaining = (
 245           self.GetChangesFromTimestamp([sync_type], original_version))
 246       self.assertEqual(3, len(changes))
 247       self.assertEqual(0, remaining)
 248       self.assertEqual(original_version + 3, version)
 249       self.assertEqual([result1, result2, result3], changes)
 250       for c in changes:
 251         self.assertTrue(c is not self.model._entries[c.id_string],
 252                         "GetChanges didn't make a defensive copy.")
 253       self.assertTrue(result2.position_in_parent < result3.position_in_parent)
 254       self.assertEqual(-100, result2.position_in_parent)
 255
 256       # Now update the items so that the second item is the parent of the
 257       # first; with the first sandwiched between two new items (4 and 5).
 258       # Do this in a new commit session, meaning we'll reference items from
 259       # the first batch by their post-commit, server IDs.
 260       commit_session = {}
 261       old_cache_guid = my_cache_guid
 262       my_cache_guid = 'A different GUID'
 263       proto2b, result2b = DoCommit(original=result2,
 264                                    parent=original_changes[-1])
 265       proto4, result4 = DoCommit(id_string='ID4', name='Four',
 266                                  parent=result2, position=-200)
 267       proto1b, result1b = DoCommit(original=result1,
 268                                    parent=result2, position=-150)
 269       proto5, result5 = DoCommit(id_string='ID5', name='Five', parent=result2,
 270                                  position=150)
 271
 272       self.assertEqual(2, len(commit_session), 'Only new items in second '
 273                        'batch should be in the session')
 274       for p, r, original in [(proto2b, result2b, proto2),
 275                              (proto4, result4, proto4),
 276                              (proto1b, result1b, proto1),
 277                              (proto5, result5, proto5)]:
 278         self.assertEqual(r.originator_client_item_id, original.id_string)
 279         if original is not p:
 280           self.assertEqual(r.id_string, p.id_string,
 281                            'Ids should be stable after first commit')
 282           self.assertEqual(r.originator_cache_guid, old_cache_guid)
 283         else:
 284           self.assertNotEqual(r.id_string, p.id_string)
 285           self.assertEqual(r.originator_cache_guid, my_cache_guid)
 286           self.assertEqual(commit_session.get(p.id_string), r.id_string)
 287         self.assertTrue(r is not self.model._entries[r.id_string],
 288                         "Commit result didn't make a defensive copy.")
 289         self.assertTrue(p is not self.model._entries[r.id_string],
 290                         "Commit didn't make a defensive copy.")
 291         self.assertTrue(r.version > p.version)
 292       version, changes, remaining = (
 293           self.GetChangesFromTimestamp([sync_type], original_version))
 294       self.assertEqual(5, len(changes))
 295       self.assertEqual(0, remaining)
 296       self.assertEqual(original_version + 7, version)
 297       self.assertEqual([result3, result2b, result4, result1b, result5], changes)
 298       for c in changes:
 299         self.assertTrue(c is not self.model._entries[c.id_string],
 300                         "GetChanges didn't make a defensive copy.")
 301       self.assertTrue(result4.parent_id_string ==
 302                       result1b.parent_id_string ==
 303                       result5.parent_id_string ==
 304                       result2b.id_string)
 305       self.assertTrue(result4.position_in_parent <
 306                       result1b.position_in_parent <
 307                       result5.position_in_parent)
 308
 309   def testUpdateSieve(self):
 310     # from_timestamp, legacy mode
 311     autofill = chromiumsync.SYNC_TYPE_FIELDS['autofill']
 312     theme = chromiumsync.SYNC_TYPE_FIELDS['theme']
 313     msg = sync_pb2.GetUpdatesMessage()
 314     msg.from_timestamp = 15412
 315     msg.requested_types.autofill.SetInParent()
 316     msg.requested_types.theme.SetInParent()
 317
 318     sieve = chromiumsync.UpdateSieve(msg)
 319     self.assertEqual(sieve._state,
 320         {chromiumsync.TOP_LEVEL: 15412,
 321          chromiumsync.AUTOFILL: 15412,
 322          chromiumsync.THEME: 15412})
 323
 324     response = sync_pb2.GetUpdatesResponse()
 325     sieve.SaveProgress(15412, response)
 326     self.assertEqual(0, len(response.new_progress_marker))
 327     self.assertFalse(response.HasField('new_timestamp'))
 328
 329     response = sync_pb2.GetUpdatesResponse()
 330     sieve.SaveProgress(15413, response)
 331     self.assertEqual(0, len(response.new_progress_marker))
 332     self.assertTrue(response.HasField('new_timestamp'))
 333     self.assertEqual(15413, response.new_timestamp)
 334
 335     # Existing tokens
 336     msg = sync_pb2.GetUpdatesMessage()
 337     marker = msg.from_progress_marker.add()
 338     marker.data_type_id = autofill.number
 339     marker.token = pickle.dumps((15412, 1))
 340     marker = msg.from_progress_marker.add()
 341     marker.data_type_id = theme.number
 342     marker.token = pickle.dumps((15413, 1))
 343     sieve = chromiumsync.UpdateSieve(msg)
 344     self.assertEqual(sieve._state,
 345         {chromiumsync.TOP_LEVEL: 15412,
 346          chromiumsync.AUTOFILL: 15412,
 347          chromiumsync.THEME: 15413})
 348
 349     response = sync_pb2.GetUpdatesResponse()
 350     sieve.SaveProgress(15413, response)
 351     self.assertEqual(1, len(response.new_progress_marker))
 352     self.assertFalse(response.HasField('new_timestamp'))
 353     marker = response.new_progress_marker[0]
 354     self.assertEqual(marker.data_type_id, autofill.number)
 355     self.assertEqual(pickle.loads(marker.token), (15413, 1))
 356     self.assertFalse(marker.HasField('timestamp_token_for_migration'))
 357
 358     # Empty tokens indicating from timestamp = 0
 359     msg = sync_pb2.GetUpdatesMessage()
 360     marker = msg.from_progress_marker.add()
 361     marker.data_type_id = autofill.number
 362     marker.token = pickle.dumps((412, 1))
 363     marker = msg.from_progress_marker.add()
 364     marker.data_type_id = theme.number
 365     marker.token = ''
 366     sieve = chromiumsync.UpdateSieve(msg)
 367     self.assertEqual(sieve._state,
 368         {chromiumsync.TOP_LEVEL: 0,
 369          chromiumsync.AUTOFILL: 412,
 370          chromiumsync.THEME: 0})
 371     response = sync_pb2.GetUpdatesResponse()
 372     sieve.SaveProgress(1, response)
 373     self.assertEqual(1, len(response.new_progress_marker))
 374     self.assertFalse(response.HasField('new_timestamp'))
 375     marker = response.new_progress_marker[0]
 376     self.assertEqual(marker.data_type_id, theme.number)
 377     self.assertEqual(pickle.loads(marker.token), (1, 1))
 378     self.assertFalse(marker.HasField('timestamp_token_for_migration'))
 379
 380     response = sync_pb2.GetUpdatesResponse()
 381     sieve.SaveProgress(412, response)
 382     self.assertEqual(1, len(response.new_progress_marker))
 383     self.assertFalse(response.HasField('new_timestamp'))
 384     marker = response.new_progress_marker[0]
 385     self.assertEqual(marker.data_type_id, theme.number)
 386     self.assertEqual(pickle.loads(marker.token), (412, 1))
 387     self.assertFalse(marker.HasField('timestamp_token_for_migration'))
 388
 389     response = sync_pb2.GetUpdatesResponse()
 390     sieve.SaveProgress(413, response)
 391     self.assertEqual(2, len(response.new_progress_marker))
 392     self.assertFalse(response.HasField('new_timestamp'))
 393     marker = self.FindMarkerByNumber(response.new_progress_marker, theme)
 394     self.assertEqual(pickle.loads(marker.token), (413, 1))
 395     self.assertFalse(marker.HasField('timestamp_token_for_migration'))
 396     marker = self.FindMarkerByNumber(response.new_progress_marker, autofill)
 397     self.assertEqual(pickle.loads(marker.token), (413, 1))
 398     self.assertFalse(marker.HasField('timestamp_token_for_migration'))
 399
 400     # Migration token timestamps (client gives timestamp, server returns token)
 401     # These are for migrating from the old 'timestamp' protocol to the
 402     # progressmarker protocol, and have nothing to do with the MIGRATION_DONE
 403     # error code.
 404     msg = sync_pb2.GetUpdatesMessage()
 405     marker = msg.from_progress_marker.add()
 406     marker.data_type_id = autofill.number
 407     marker.timestamp_token_for_migration = 15213
 408     marker = msg.from_progress_marker.add()
 409     marker.data_type_id = theme.number
 410     marker.timestamp_token_for_migration = 15211
 411     sieve = chromiumsync.UpdateSieve(msg)
 412     self.assertEqual(sieve._state,
 413         {chromiumsync.TOP_LEVEL: 15211,
 414          chromiumsync.AUTOFILL: 15213,
 415          chromiumsync.THEME: 15211})
 416     response = sync_pb2.GetUpdatesResponse()
 417     sieve.SaveProgress(16000, response)  # There were updates
 418     self.assertEqual(2, len(response.new_progress_marker))
 419     self.assertFalse(response.HasField('new_timestamp'))
 420     marker = self.FindMarkerByNumber(response.new_progress_marker, theme)
 421     self.assertEqual(pickle.loads(marker.token), (16000, 1))
 422     self.assertFalse(marker.HasField('timestamp_token_for_migration'))
 423     marker = self.FindMarkerByNumber(response.new_progress_marker, autofill)
 424     self.assertEqual(pickle.loads(marker.token), (16000, 1))
 425     self.assertFalse(marker.HasField('timestamp_token_for_migration'))
 426
 427     msg = sync_pb2.GetUpdatesMessage()
 428     marker = msg.from_progress_marker.add()
 429     marker.data_type_id = autofill.number
 430     marker.timestamp_token_for_migration = 3000
 431     marker = msg.from_progress_marker.add()
 432     marker.data_type_id = theme.number
 433     marker.timestamp_token_for_migration = 3000
 434     sieve = chromiumsync.UpdateSieve(msg)
 435     self.assertEqual(sieve._state,
 436         {chromiumsync.TOP_LEVEL: 3000,
 437          chromiumsync.AUTOFILL: 3000,
 438          chromiumsync.THEME: 3000})
 439     response = sync_pb2.GetUpdatesResponse()
 440     sieve.SaveProgress(3000, response)  # Already up to date
 441     self.assertEqual(2, len(response.new_progress_marker))
 442     self.assertFalse(response.HasField('new_timestamp'))
 443     marker = self.FindMarkerByNumber(response.new_progress_marker, theme)
 444     self.assertEqual(pickle.loads(marker.token), (3000, 1))
 445     self.assertFalse(marker.HasField('timestamp_token_for_migration'))
 446     marker = self.FindMarkerByNumber(response.new_progress_marker, autofill)
 447     self.assertEqual(pickle.loads(marker.token), (3000, 1))
 448     self.assertFalse(marker.HasField('timestamp_token_for_migration'))
 449
 450   def testCheckRaiseTransientError(self):
 451     testserver = chromiumsync.TestServer()
 452     http_code, raw_respon = testserver.HandleSetTransientError()
 453     self.assertEqual(http_code, 200)
 454     try:
 455       testserver.CheckTransientError()
 456       self.fail('Should have raised transient error exception')
 457     except chromiumsync.TransientError:
 458       self.assertTrue(testserver.transient_error)
 459
 460   def testUpdateSieveStoreMigration(self):
 461     autofill = chromiumsync.SYNC_TYPE_FIELDS['autofill']
 462     theme = chromiumsync.SYNC_TYPE_FIELDS['theme']
 463     migrator = chromiumsync.MigrationHistory()
 464     msg = sync_pb2.GetUpdatesMessage()
 465     marker = msg.from_progress_marker.add()
 466     marker.data_type_id = autofill.number
 467     marker.token = pickle.dumps((15412, 1))
 468     marker = msg.from_progress_marker.add()
 469     marker.data_type_id = theme.number
 470     marker.token = pickle.dumps((15413, 1))
 471     sieve = chromiumsync.UpdateSieve(msg, migrator)
 472     sieve.CheckMigrationState()
 473
 474     migrator.Bump([chromiumsync.BOOKMARK, chromiumsync.PASSWORD])  # v=2
 475     sieve = chromiumsync.UpdateSieve(msg, migrator)
 476     sieve.CheckMigrationState()
 477     self.assertEqual(sieve._state,
 478         {chromiumsync.TOP_LEVEL: 15412,
 479          chromiumsync.AUTOFILL: 15412,
 480          chromiumsync.THEME: 15413})
 481
 482     migrator.Bump([chromiumsync.AUTOFILL, chromiumsync.PASSWORD])  # v=3
 483     sieve = chromiumsync.UpdateSieve(msg, migrator)
 484     try:
 485       sieve.CheckMigrationState()
 486       self.fail('Should have raised.')
 487     except chromiumsync.MigrationDoneError, error:
 488       # We want this to happen.
 489       self.assertEqual([chromiumsync.AUTOFILL], error.datatypes)
 490
 491     msg = sync_pb2.GetUpdatesMessage()
 492     marker = msg.from_progress_marker.add()
 493     marker.data_type_id = autofill.number
 494     marker.token = ''
 495     marker = msg.from_progress_marker.add()
 496     marker.data_type_id = theme.number
 497     marker.token = pickle.dumps((15413, 1))
 498     sieve = chromiumsync.UpdateSieve(msg, migrator)
 499     sieve.CheckMigrationState()
 500     response = sync_pb2.GetUpdatesResponse()
 501     sieve.SaveProgress(15412, response)  # There were updates
 502     self.assertEqual(1, len(response.new_progress_marker))
 503     self.assertFalse(response.HasField('new_timestamp'))
 504     self.assertFalse(marker.HasField('timestamp_token_for_migration'))
 505     marker = self.FindMarkerByNumber(response.new_progress_marker, autofill)
 506     self.assertEqual(pickle.loads(marker.token), (15412, 3))
 507     self.assertFalse(marker.HasField('timestamp_token_for_migration'))
 508     msg = sync_pb2.GetUpdatesMessage()
 509     marker = msg.from_progress_marker.add()
 510     marker.data_type_id = autofill.number
 511     marker.token = pickle.dumps((15412, 3))
 512     marker = msg.from_progress_marker.add()
 513     marker.data_type_id = theme.number
 514     marker.token = pickle.dumps((15413, 1))
 515     sieve = chromiumsync.UpdateSieve(msg, migrator)
 516     sieve.CheckMigrationState()
 517
 518     migrator.Bump([chromiumsync.THEME, chromiumsync.AUTOFILL])  # v=4
 519     migrator.Bump([chromiumsync.AUTOFILL])                      # v=5
 520     sieve = chromiumsync.UpdateSieve(msg, migrator)
 521     try:
 522       sieve.CheckMigrationState()
 523       self.fail("Should have raised.")
 524     except chromiumsync.MigrationDoneError, error:
 525       # We want this to happen.
 526       self.assertEqual(set([chromiumsync.THEME, chromiumsync.AUTOFILL]),
 527                        set(error.datatypes))
 528     msg = sync_pb2.GetUpdatesMessage()
 529     marker = msg.from_progress_marker.add()
 530     marker.data_type_id = autofill.number
 531     marker.token = ''
 532     marker = msg.from_progress_marker.add()
 533     marker.data_type_id = theme.number
 534     marker.token = pickle.dumps((15413, 1))
 535     sieve = chromiumsync.UpdateSieve(msg, migrator)
 536     try:
 537       sieve.CheckMigrationState()
 538       self.fail("Should have raised.")
 539     except chromiumsync.MigrationDoneError, error:
 540       # We want this to happen.
 541       self.assertEqual([chromiumsync.THEME], error.datatypes)
 542
 543     msg = sync_pb2.GetUpdatesMessage()
 544     marker = msg.from_progress_marker.add()
 545     marker.data_type_id = autofill.number
 546     marker.token = ''
 547     marker = msg.from_progress_marker.add()
 548     marker.data_type_id = theme.number
 549     marker.token = ''
 550     sieve = chromiumsync.UpdateSieve(msg, migrator)
 551     sieve.CheckMigrationState()
 552     response = sync_pb2.GetUpdatesResponse()
 553     sieve.SaveProgress(15412, response)  # There were updates
 554     self.assertEqual(2, len(response.new_progress_marker))
 555     self.assertFalse(response.HasField('new_timestamp'))
 556     self.assertFalse(marker.HasField('timestamp_token_for_migration'))
 557     marker = self.FindMarkerByNumber(response.new_progress_marker, autofill)
 558     self.assertEqual(pickle.loads(marker.token), (15412, 5))
 559     self.assertFalse(marker.HasField('timestamp_token_for_migration'))
 560     marker = self.FindMarkerByNumber(response.new_progress_marker, theme)
 561     self.assertEqual(pickle.loads(marker.token), (15412, 4))
 562     self.assertFalse(marker.HasField('timestamp_token_for_migration'))
 563     msg = sync_pb2.GetUpdatesMessage()
 564     marker = msg.from_progress_marker.add()
 565     marker.data_type_id = autofill.number
 566     marker.token = pickle.dumps((15412, 5))
 567     marker = msg.from_progress_marker.add()
 568     marker.data_type_id = theme.number
 569     marker.token = pickle.dumps((15413, 4))
 570     sieve = chromiumsync.UpdateSieve(msg, migrator)
 571     sieve.CheckMigrationState()
 572
 573   def testCreateSyncedBookmarks(self):
 574     version1, changes, remaining = (
 575         self.GetChangesFromTimestamp([chromiumsync.BOOKMARK], 0))
 576     id_string = self.model._MakeCurrentId(chromiumsync.BOOKMARK,
 577                                           '<server tag>synced_bookmarks')
 578     self.assertFalse(self.model._ItemExists(id_string))
 579     self._expect_synced_bookmarks_folder = True
 580     self.model.TriggerCreateSyncedBookmarks()
 581     self.assertTrue(self.model._ItemExists(id_string))
 582
 583     # Check that the version changed when the folder was created and the only
 584     # change was the folder creation.
 585     version2, changes, remaining = (
 586         self.GetChangesFromTimestamp([chromiumsync.BOOKMARK], version1))
 587     self.assertEqual(len(changes), 1)
 588     self.assertEqual(changes[0].id_string, id_string)
 589     self.assertNotEqual(version1, version2)
 590     self.assertEqual(
 591         self.ExpectedPermanentItemCount(chromiumsync.BOOKMARK),
 592         version2)
 593
 594     # Ensure getting from timestamp 0 includes the folder.
 595     version, changes, remaining = (
 596         self.GetChangesFromTimestamp([chromiumsync.BOOKMARK], 0))
 597     self.assertEqual(
 598         self.ExpectedPermanentItemCount(chromiumsync.BOOKMARK),
 599         len(changes))
 600     self.assertEqual(version2, version)
 601
 602   def testAcknowledgeManagedUser(self):
 603     # Create permanent items.
 604     self.GetChangesFromTimestamp([chromiumsync.MANAGED_USER], 0)
 605     proto = sync_pb2.SyncEntity()
 606     proto.id_string = 'abcd'
 607     proto.version = 0
 608
 609     # Make sure the managed_user field exists.
 610     proto.specifics.managed_user.acknowledged = False
 611     self.assertTrue(proto.specifics.HasField('managed_user'))
 612     self.AddToModel(proto)
 613     version1, changes1, remaining1 = (
 614         self.GetChangesFromTimestamp([chromiumsync.MANAGED_USER], 0))
 615     for change in changes1:
 616       self.assertTrue(not change.specifics.managed_user.acknowledged)
 617
 618     # Turn on managed user acknowledgement
 619     self.model.acknowledge_managed_users = True
 620
 621     version2, changes2, remaining2 = (
 622         self.GetChangesFromTimestamp([chromiumsync.MANAGED_USER], 0))
 623     for change in changes2:
 624       self.assertTrue(change.specifics.managed_user.acknowledged)
 625
 626   def testGetKey(self):
 627     [key1] = self.model.GetKeystoreKeys()
 628     [key2] = self.model.GetKeystoreKeys()
 629     self.assertTrue(len(key1))
 630     self.assertEqual(key1, key2)
 631
 632     # Trigger the rotation. A subsequent GetUpdates should return the nigori
 633     # node (whose timestamp was bumped by the rotation).
 634     version1, changes, remaining = (
 635         self.GetChangesFromTimestamp([chromiumsync.NIGORI], 0))
 636     self.model.TriggerRotateKeystoreKeys()
 637     version2, changes, remaining = (
 638         self.GetChangesFromTimestamp([chromiumsync.NIGORI], version1))
 639     self.assertNotEqual(version1, version2)
 640     self.assertEquals(len(changes), 1)
 641     self.assertEquals(changes[0].name, "Nigori")
 642
 643     # The current keys should contain the old keys, with the new key appended.
 644     [key1, key3] = self.model.GetKeystoreKeys()
 645     self.assertEquals(key1, key2)
 646     self.assertNotEqual(key1, key3)
 647     self.assertTrue(len(key3) > 0)
 648
 649   def testTriggerEnableKeystoreEncryption(self):
 650     version1, changes, remaining = (
 651         self.GetChangesFromTimestamp([chromiumsync.EXPERIMENTS], 0))
 652     keystore_encryption_id_string = (
 653         self.model._ClientTagToId(
 654             chromiumsync.EXPERIMENTS,
 655             chromiumsync.KEYSTORE_ENCRYPTION_EXPERIMENT_TAG))
 656
 657     self.assertFalse(self.model._ItemExists(keystore_encryption_id_string))
 658     self.model.TriggerEnableKeystoreEncryption()
 659     self.assertTrue(self.model._ItemExists(keystore_encryption_id_string))
 660
 661     # The creation of the experiment should be downloaded on the next
 662     # GetUpdates.
 663     version2, changes, remaining = (
 664         self.GetChangesFromTimestamp([chromiumsync.EXPERIMENTS], version1))
 665     self.assertEqual(len(changes), 1)
 666     self.assertEqual(changes[0].id_string, keystore_encryption_id_string)
 667     self.assertNotEqual(version1, version2)
 668
 669     # Verify the experiment was created properly and is enabled.
 670     self.assertEqual(chromiumsync.KEYSTORE_ENCRYPTION_EXPERIMENT_TAG,
 671                      changes[0].client_defined_unique_tag)
 672     self.assertTrue(changes[0].HasField("specifics"))
 673     self.assertTrue(changes[0].specifics.HasField("experiments"))
 674     self.assertTrue(
 675         changes[0].specifics.experiments.HasField("keystore_encryption"))
 676     self.assertTrue(
 677         changes[0].specifics.experiments.keystore_encryption.enabled)
 678
 679 if __name__ == '__main__':
 680   unittest.main()