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_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_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))
def test_contents_immutable(self):
textfile = VersionedTextFile(data=SAMPLE_TEXT)
textfile.save()
textfile.data = "foo"
with ExpectedException(ValidationError, ".*immutable.*"):
textfile.save()
def test_creates_versionedtextfile(self):
textfile = VersionedTextFile(data=SAMPLE_TEXT)
textfile.save()
from_db = VersionedTextFile.objects.get(id=textfile.id)
self.assertEqual((from_db.id, from_db.data),
(textfile.id, SAMPLE_TEXT))