def test_recovery_from_racy_record_creation(self):
timestamp = get_timestamp()
# Simulate race for forcing creation of two rows with same timestamp.
user1 = self.backend.allocate_user("sync-1.0", "*****@*****.**",
timestamp=timestamp)
user2 = self.backend.allocate_user("sync-1.0", "*****@*****.**",
timestamp=timestamp)
self.assertNotEqual(user1["uid"], user2["uid"])
# Neither is marked replaced initially.
old_records = list(self.backend.get_old_user_records("sync-1.0", 0))
self.assertEqual(len(old_records), 0)
# Reading current details will detect the problem and fix it.
self.backend.get_user("sync-1.0", "*****@*****.**")
old_records = list(self.backend.get_old_user_records("sync-1.0", 0))
self.assertEqual(len(old_records), 1)
def test_that_race_recovery_respects_generation_number_monotonicity(self):
timestamp = get_timestamp()
# Simulate race between clients with different generation numbers,
# in which the out-of-date client gets a higher timestamp.
user1 = self.backend.allocate_user("sync-1.0", "*****@*****.**",
generation=1, timestamp=timestamp)
user2 = self.backend.allocate_user("sync-1.0", "*****@*****.**",
generation=2, timestamp=timestamp-1)
self.assertNotEqual(user1["uid"], user2["uid"])
# Reading current details should promote the higher-generation one.
user = self.backend.get_user("sync-1.0", "*****@*****.**")
self.assertEqual(user["generation"], 2)
self.assertEqual(user["uid"], user2["uid"])
# And the other record should get marked as replaced.
old_records = list(self.backend.get_old_user_records("sync-1.0", 0))
self.assertEqual(len(old_records), 1)
def test_count_users(self):
user = self.backend.allocate_user("sync-1.0", "*****@*****.**")
self.assertEqual(self.backend.count_users(), 1)
old_timestamp = get_timestamp()
time.sleep(0.01)
# Adding users increases the count.
user = self.backend.allocate_user("sync-1.0", "*****@*****.**")
self.assertEqual(self.backend.count_users(), 2)
# Updating a user doesn't change the count.
self.backend.update_user("sync-1.0", user, client_state="aaa")
self.assertEqual(self.backend.count_users(), 2)
# Looking back in time doesn't count newer users.
self.assertEqual(self.backend.count_users(old_timestamp), 1)
# Retiring a user decreases the count.
self.backend.retire_user("*****@*****.**")
self.assertEqual(self.backend.count_users(), 1)
def test_count_users(self):
user = self.backend.allocate_user("sync-1.0", "*****@*****.**")
self.assertEqual(self.backend.count_users(), 1)
old_timestamp = get_timestamp()
time.sleep(0.01)
# Adding users increases the count.
user = self.backend.allocate_user("sync-1.0", "*****@*****.**")
self.assertEqual(self.backend.count_users(), 2)
# Updating a user doesn't change the count.
self.backend.update_user("sync-1.0", user, client_state="aaa")
self.assertEqual(self.backend.count_users(), 2)
# Looking back in time doesn't count newer users.
self.assertEqual(self.backend.count_users(old_timestamp), 1)
# Retiring a user decreases the count.
self.backend.retire_user("*****@*****.**")
self.assertEqual(self.backend.count_users(), 1)
def test_that_race_recovery_respects_generation_after_reassignment(self):
timestamp = get_timestamp()
# Simulate race between clients with different generation numbers,
# in which the out-of-date client gets a higher timestamp.
user1 = self.backend.allocate_user("sync-1.0", "*****@*****.**",
generation=1, timestamp=timestamp)
user2 = self.backend.allocate_user("sync-1.0", "*****@*****.**",
generation=2, timestamp=timestamp-1)
self.assertNotEqual(user1["uid"], user2["uid"])
# Force node re-assignment by marking all records as replaced.
self.backend.replace_user_records("sync-1.0", "*****@*****.**",
timestamp=timestamp + 1)
# The next client to show up should get a new assignment, marked
# with the correct generation number.
user = self.backend.get_user("sync-1.0", "*****@*****.**")
self.assertEqual(user["generation"], 2)
self.assertNotEqual(user["uid"], user1["uid"])
self.assertNotEqual(user["uid"], user2["uid"])