lib/vstorage/tests.py

   1 #!/usr/bin/python
   2 # -*- coding: utf-8 -*-
   3 #
   4 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
   5 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   6 #
   7
   8 import os
   9 import tempfile
  10 from nose.tools import *
  11 from nose.core import runmodule
  12
  13 import vstorage
  14
  15 def clear_directory(top):
  16     for root, dirs, files in os.walk(top, topdown=False):
  17         for name in files:
  18             os.remove(os.path.join(root, name))
  19         for name in dirs:
  20             os.rmdir(os.path.join(root, name))
  21     try:
  22         os.removedirs(top)
  23     except OSError:
  24         pass
  25
  26
  27 class TestVersionedStorage(object):
  28     def setUp(self):
  29         self.repo_path = tempfile.mkdtemp()
  30         self.repo = vstorage.VersionedStorage(self.repo_path)
  31
  32     def tearDown(self):
  33         clear_directory(self.repo_path)
  34
  35     def test_save_text(self):
  36         text = u"test text"
  37         title = u"test title"
  38         author = u"test author"
  39         comment = u"test comment"
  40
  41         self.repo.save_text(
  42             title=title,
  43             text=text,
  44             author=author,
  45             comment=comment,
  46             parent=None,
  47         )
  48
  49         saved_text, rev = self.repo.page_text(title)
  50         assert_equal(saved_text, text)
  51         assert_equal(rev, 0)
  52
  53     def test_save_text_noparent(self):
  54         text = u"test text"
  55         title = u"test title"
  56         author = u"test author"
  57         comment = u"test comment"
  58
  59         self.repo.save_text(title=title,
  60                     text=text, author=author,
  61                     comment=comment, parent=None)
  62
  63
  64         saved_text, rev = self.repo.page_text(title)
  65         assert_equal(saved_text, text)
  66         assert_equal(rev, 0)
  67
  68     def test_save_merge_no_conflict(self):
  69         text = u"test\ntext"
  70         title = u"test title"
  71         author = u"test author"
  72         comment = u"test comment"
  73         self.repo.save_text(title=title,
  74                     text=text, author=author,
  75                     comment=comment, parent=None)
  76         self.repo.save_text(title=title,
  77                     text=text, author=author,
  78                     comment=comment, parent=None)
  79
  80         saved_text, rev = self.repo.page_text(title)
  81         assert_equal(saved_text, text)
  82         assert_equal(rev, 0)
  83
  84     def test_save_merge_line_conflict(self):
  85         text = u"test\ntest\n"
  86         text1 = u"test\ntext\n"
  87         text2 = u"text\ntest\n"
  88         title = u"test title"
  89         author = u"test author"
  90         comment = u"test comment"
  91
  92         self.repo.save_text(title=title,
  93                     text=text, author=author,
  94                     comment=comment, parent=None)
  95
  96         saved_text, rev = self.repo.page_text(title)
  97         assert_equal(saved_text, text)
  98         assert_equal(rev, 0)
  99
 100         self.repo.save_text(title=title,
 101                     text=text1, author=author,
 102                     comment=comment, parent=0)
 103
 104         saved_text, rev = self.repo.page_text(title)
 105         assert_equal(saved_text, text1)
 106         assert_equal(rev, 1)
 107
 108         self.repo.save_text(title=title,
 109                     text=text2, author=author,
 110                     comment=comment, parent=0)
 111
 112         saved_text, rev = self.repo.page_text(title)
 113         # Other conflict markers placement can also be correct
 114         assert_equal(saved_text, u'''\
 115 text
 116 test
 117 <<<<<<< local
 118 =======
 119 text
 120 >>>>>>> other
 121 ''')
 122
 123     def test_delete(self):
 124         text = u"text test"
 125         title = u"test title"
 126         author = u"test author"
 127         comment = u"test comment"
 128         self.repo.save_text(title=title,
 129                     text=text, author=author,
 130                     comment=comment, parent=None)
 131
 132         ok_(title in self.repo, "Document not in repository.")
 133
 134         self.repo.delete_page(title, author, comment)
 135
 136         ok_(title not in self.repo, "Document in repository after delete")
 137
 138     @raises(vstorage.DocumentNotFound)
 139     def test_document_not_found(self):
 140         self.repo.page_text(u'unknown entity')
 141
 142     def test_open_existing_repository(self):
 143         self.repo.save_text(title=u'Python!', text=u'ham and spam')
 144         current_repo_revision = self.repo.repo_revision()
 145         same_repo = vstorage.VersionedStorage(self.repo_path)
 146         assert_equal(same_repo.repo_revision(), current_repo_revision)
 147
 148     def test_history(self):
 149         COMMITS = [
 150             {"author": "bunny", "text":"1", "comment": "Oh yeah!"},
 151             {"author": "frank", "text":"2", "comment": "Second is the best!"},
 152             {"text":"3", "comment": "Third"},
 153             {"author": "welma", "text":"4", "comment": "Fourth"},
 154         ]
 155
 156         for commit in COMMITS:
 157             self.repo.save_text(title=u"Sample", **commit)
 158
 159         for n, entry in enumerate(reversed(list(self.repo.page_history(u"Sample")))):
 160             assert_equal(entry["version"], n)
 161             assert_equal(entry["author"], COMMITS[n].get("author", "anonymous"))
 162             assert_equal(entry["description"], COMMITS[n]["comment"])
 163             assert_equal(entry["tag"], [])
 164
 165     def test_data_revert(self):
 166         COMMITS = [
 167             {u"title": u"one", "author": "bunny", "text":"1.1", "comment": "1"},
 168             {u"title": u"one", "author": "frank", "text":"1.2", "comment": "2"},
 169             {u"title": u"two", "author": "bunny", "text":"2.1", "comment": "3"},
 170             {u"title": u"one", "author": "frank", "text":"1.3", "comment": "4"},
 171         ]
 172
 173         for commit in COMMITS:
 174             self.repo.save_text(**commit)
 175
 176         # now revert last change on one
 177         self.repo.revert(u"one", 0)
 178         assert_equal(self.repo.page_text(u"one"), (u"1.1", 3))
 179         assert_equal(self.repo.page_text(u"two"), (u"2.1", 0))
 180
 181         self.repo.revert(u"one", 2)
 182         assert_equal(self.repo.page_text(u"one"), (u"1.3", 4))
 183
 184 class TestVSTags(object):
 185
 186     TITLE_1 = "Sample"
 187
 188     COMMITS = [
 189         {"author": "bunny", "text":"1", "comment": "Oh yeah!"},
 190         {"author": "frank", "text":"2", "comment": "Second is the best!"},
 191         {"text":"3", "comment": "Third"},
 192         {"author": "welma", "text":"4", "comment": "Fourth"},
 193     ]
 194
 195     def setUp(self):
 196         self.repo_path = tempfile.mkdtemp()
 197         self.repo = vstorage.VersionedStorage(self.repo_path)
 198
 199         # generate some history
 200         for commit in self.COMMITS:
 201             self.repo.save_text(title=u"Sample", **commit)
 202
 203         # verify
 204         for n, entry in enumerate(reversed(list(self.repo.page_history(self.TITLE_1)))):
 205             assert_equal(entry["tag"], [])
 206
 207     def tearDown(self):
 208         clear_directory(self.repo_path)
 209
 210     def test_add_tag(self):
 211         TAG_USER = "mike_the_tagger"
 212         TAG_NAME = "production"
 213         TAG_VERSION = 2
 214
 215         # Add tag
 216         self.repo.add_page_tag(self.TITLE_1, TAG_VERSION, TAG_NAME, TAG_USER)
 217
 218         # check history again
 219         history = list(self.repo.page_history(self.TITLE_1))
 220         for entry in reversed(history):
 221             if entry["version"] == TAG_VERSION:
 222                 assert_equal(entry["tag"], [TAG_NAME])
 223             else:
 224                 assert_equal(entry["tag"], [])
 225
 226     def test_add_many_tags(self):
 227         TAG_USER = "mike_the_tagger"
 228         tags = [
 229             (2, "production", "mike"),
 230             (2, "finished", "jeremy"),
 231             (0, "original", "jeremy"),
 232         ]
 233
 234         for rev, name, user in tags:
 235             self.repo.add_page_tag(self.TITLE_1, rev, name, user)
 236
 237         # check history again
 238         history = list(self.repo.page_history(self.TITLE_1))
 239         for entry in reversed(history):
 240             expected = [tag[1] for tag in tags if tag[0] == entry["version"]]
 241             assert_equal(set(entry["tag"]), set(expected))