def test_deletes_upstream_revisions(self): textfile = VersionedTextFile(data=SAMPLE_TEXT) textfile.save() textfile.update(SAMPLE_TEXT + " 2") self.assertThat(VersionedTextFile.objects.count(), Equals(2)) textfile.delete() self.assertThat(VersionedTextFile.objects.count(), Equals(0))
def test_update_with_no_changes_returns_current_vision(self): textfile = VersionedTextFile(data=SAMPLE_TEXT) textfile.save() textfile2 = textfile.update(SAMPLE_TEXT) from_db = VersionedTextFile.objects.get(id=textfile2.id) self.assertThat(from_db.data, Equals(SAMPLE_TEXT)) self.assertThat(from_db.previous_version, Is(None))
def test_revert_call_gc_hook(self): textfile = VersionedTextFile(data=SAMPLE_TEXT) textfile.save() textfile_ids = [textfile.id] for _ in range(10): textfile = textfile.update(factory.make_string()) textfile_ids.append(textfile.id) # gc_hook only runs when there is something to revert to so # make sure we're actually reverting revert_to = random.choice(textfile_ids[:-1]) reverted_ids = [] remaining_ids = [] reverted_or_remaining = remaining_ids for i in textfile_ids: reverted_or_remaining.append(i) if i == revert_to: reverted_or_remaining = reverted_ids gc_hook = Mock() textfile = textfile.revert(revert_to, gc_hook=gc_hook) for i in reverted_ids: self.assertRaises( VersionedTextFile.DoesNotExist, VersionedTextFile.objects.get, id=i, ) for i in remaining_ids: self.assertIsNotNone(VersionedTextFile.objects.get(id=i)) self.assertThat(gc_hook, MockCalledOnceWith(textfile))
def test_update_links_previous_revision(self): textfile = VersionedTextFile(data=SAMPLE_TEXT) textfile.save() textfile2 = textfile.update(SAMPLE_TEXT + " 2") from_db = VersionedTextFile.objects.get(id=textfile2.id) self.assertThat(from_db.data, Equals(SAMPLE_TEXT + " 2")) self.assertThat(from_db.previous_version, Equals(textfile))
def test_revert_by_id_with_garbage_collection(self): textfile = VersionedTextFile(data=SAMPLE_TEXT) textfile.save() textfile_ids = [textfile.id] for _ in range(10): textfile = textfile.update(factory.make_string()) textfile_ids.append(textfile.id) revert_to = random.choice(textfile_ids) reverted_ids = [] remaining_ids = [] reverted_or_remaining = remaining_ids for i in textfile_ids: reverted_or_remaining.append(i) if i == revert_to: reverted_or_remaining = reverted_ids self.assertEquals( VersionedTextFile.objects.get(id=revert_to), textfile.revert(revert_to), ) for i in reverted_ids: self.assertRaises( VersionedTextFile.DoesNotExist, VersionedTextFile.objects.get, id=i, ) for i in remaining_ids: self.assertIsNotNone(VersionedTextFile.objects.get(id=i))
def test_revert_by_negative_raises_value_error_when_too_far_back(self): textfile = VersionedTextFile(data=SAMPLE_TEXT) textfile.save() textfile_ids = [textfile.id] for _ in range(10): textfile = textfile.update(factory.make_string()) textfile_ids.append(textfile.id) self.assertRaises(ValueError, textfile.revert, -11)
def test_previous_versions(self): textfile = VersionedTextFile(data=factory.make_string()) textfile.save() textfiles = [textfile] for _ in range(10): textfile = textfile.update(factory.make_string()) textfiles.append(textfile) for f in textfile.previous_versions(): self.assertTrue(f in textfiles)
def test_revert_zero_does_nothing(self): textfile = VersionedTextFile(data=SAMPLE_TEXT) textfile.save() textfile_ids = [textfile.id] for _ in range(10): textfile = textfile.update(factory.make_string()) textfile_ids.append(textfile.id) self.assertEquals(textfile, textfile.revert(0)) self.assertItemsEqual(textfile_ids, [f.id for f in textfile.previous_versions()])
def test_revert_by_id_raises_value_error_when_id_not_in_history(self): textfile = VersionedTextFile(data=SAMPLE_TEXT) textfile.save() textfile_ids = [textfile.id] for _ in range(10): textfile = textfile.update(factory.make_string()) textfile_ids.append(textfile.id) other_textfile = VersionedTextFile(data=SAMPLE_TEXT) other_textfile.save() self.assertRaises(ValueError, textfile.revert, other_textfile.id)
def test_deletes_all_upstream_revisions_from_oldest_parent(self): textfile = VersionedTextFile(data=SAMPLE_TEXT) textfile.save() textfile2 = textfile.update(SAMPLE_TEXT + " 2") textfile3 = textfile.update(SAMPLE_TEXT + " 3") # Create a text file with multiple children. textfile2.update(SAMPLE_TEXT + " 20") textfile2.update(SAMPLE_TEXT + " 21") self.assertThat(VersionedTextFile.objects.count(), Equals(5)) textfile3.get_oldest_version().delete() self.assertThat(VersionedTextFile.objects.count(), Equals(0))
def test_revert_by_negative_without_garbage_collection(self): textfile = VersionedTextFile(data=SAMPLE_TEXT) textfile.save() textfile_ids = [textfile.id] for _ in range(10): textfile = textfile.update(factory.make_string()) textfile_ids.append(textfile.id) revert_to = random.randint(-10, -1) self.assertEquals(textfile_ids[revert_to - 1], textfile.revert(revert_to, False).id) for i in textfile_ids: self.assertIsNotNone(VersionedTextFile.objects.get(id=i))
def test_revert_by_negative_with_garbage_collection(self): textfile = VersionedTextFile(data=SAMPLE_TEXT) textfile.save() textfile_ids = [textfile.id] for _ in range(10): textfile = textfile.update(factory.make_string()) textfile_ids.append(textfile.id) revert_to = random.randint(-10, -1) reverted_ids = textfile_ids[revert_to:] remaining_ids = textfile_ids[:revert_to] self.assertEquals(textfile_ids[revert_to - 1], textfile.revert(revert_to).id) for i in reverted_ids: self.assertRaises(VersionedTextFile.DoesNotExist, VersionedTextFile.objects.get, id=i) for i in remaining_ids: self.assertIsNotNone(VersionedTextFile.objects.get(id=i))