Chromium Code Reviews| Index: net/tools/testserver/chromiumsync_test.py |
| diff --git a/net/tools/testserver/chromiumsync_test.py b/net/tools/testserver/chromiumsync_test.py |
| index d704ac3ee2de2700764e072dd48d96908a1ed207..b91fbfad0f4468660f94cb22c0eb64b15f7e8cc6 100755 |
| --- a/net/tools/testserver/chromiumsync_test.py |
| +++ b/net/tools/testserver/chromiumsync_test.py |
| @@ -14,6 +14,8 @@ import chromiumsync |
| import sync_pb2 |
| import theme_specifics_pb2 |
| +from google.protobuf import text_format |
|
ncarter (slow)
2012/03/01 22:13:05
Debugging cruft; remove these lines.
akalin
2012/03/02 23:56:49
not removed yet?
|
| + |
| class SyncDataModelTest(unittest.TestCase): |
| def setUp(self): |
| self.model = chromiumsync.SyncDataModel() |
| @@ -25,8 +27,9 @@ class SyncDataModelTest(unittest.TestCase): |
| message = sync_pb2.GetUpdatesMessage() |
| message.from_timestamp = timestamp |
| for data_type in requested_types: |
| - message.requested_types.Extensions[ |
| - chromiumsync.SYNC_TYPE_TO_EXTENSION[data_type]].SetInParent() |
| + getattr(message.requested_types, |
| + chromiumsync.SYNC_TYPE_TO_DESCRIPTOR[ |
| + data_type].name).SetInParent() |
| return self.model.GetChanges( |
| chromiumsync.UpdateSieve(message, self.model.migration_history)) |
| @@ -298,12 +301,12 @@ class SyncDataModelTest(unittest.TestCase): |
| def testUpdateSieve(self): |
| # from_timestamp, legacy mode |
| - autofill = autofill_specifics_pb2.autofill |
| - theme = theme_specifics_pb2.theme |
| + autofill = chromiumsync.SYNC_TYPE_FIELDS['autofill'] |
| + theme = chromiumsync.SYNC_TYPE_FIELDS['theme'] |
| msg = sync_pb2.GetUpdatesMessage() |
| msg.from_timestamp = 15412 |
| - msg.requested_types.Extensions[autofill].SetInParent() |
| - msg.requested_types.Extensions[theme].SetInParent() |
| + msg.requested_types.autofill.SetInParent() |
| + msg.requested_types.theme.SetInParent() |
| sieve = chromiumsync.UpdateSieve(msg) |
| self.assertEqual(sieve._state, |
| @@ -448,8 +451,8 @@ class SyncDataModelTest(unittest.TestCase): |
| self.assertTrue(testserver.transient_error) |
| def testUpdateSieveStoreMigration(self): |
| - autofill = autofill_specifics_pb2.autofill |
| - theme = theme_specifics_pb2.theme |
| + autofill = chromiumsync.SYNC_TYPE_FIELDS['autofill'] |
| + theme = chromiumsync.SYNC_TYPE_FIELDS['theme'] |
| migrator = chromiumsync.MigrationHistory() |
| msg = sync_pb2.GetUpdatesMessage() |
| marker = msg.from_progress_marker.add() |