mirror of
https://github.com/facebook/sapling.git
synced 2024-10-11 01:07:15 +03:00
d9be71caf7
Summary: Easy fix Reviewed By: quark-zju Differential Revision: D22143857 fbshipit-source-id: c35e7121a2be4152da24d19615dc38d551011282
102 lines
2.9 KiB
Python
102 lines
2.9 KiB
Python
from __future__ import absolute_import
|
|
|
|
import os
|
|
import random
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
|
|
import silenttestrunner
|
|
from bindings import bookmarkstore
|
|
from edenscm.mercurial import node
|
|
from hghave import require
|
|
|
|
|
|
require(["py2"])
|
|
|
|
|
|
class bookmarkstoretests(unittest.TestCase):
|
|
def setUp(self):
|
|
random.seed(0)
|
|
self._tempdirs = []
|
|
|
|
def tearDown(self):
|
|
for d in self._tempdirs:
|
|
shutil.rmtree(d)
|
|
|
|
def makeTempDir(self):
|
|
tempdir = tempfile.mkdtemp()
|
|
self._tempdirs.append(tempdir)
|
|
return tempdir
|
|
|
|
def testAddingBookmarks(self):
|
|
bmdir = self.makeTempDir()
|
|
bmstore = bookmarkstore.bookmarkstore(bmdir)
|
|
self.assertIsNone(bmstore.lookup_bookmark("not_real"))
|
|
|
|
bmstore.update("test", node.nullid)
|
|
self.assertEqual(
|
|
b"\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00",
|
|
bmstore.lookup_bookmark("test"),
|
|
)
|
|
|
|
bmstore.update("test", node.bin("1" * 40))
|
|
self.assertEqual(
|
|
b"\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11",
|
|
bmstore.lookup_bookmark("test"),
|
|
)
|
|
|
|
bmstore.remove("test")
|
|
|
|
def testAddingBookmarksToSameNode(self):
|
|
bmdir = self.makeTempDir()
|
|
bmstore = bookmarkstore.bookmarkstore(bmdir)
|
|
|
|
testnode = node.bin("2" * 40)
|
|
self.assertIsNone(bmstore.lookup_node(testnode))
|
|
|
|
bmstore.update("test", testnode)
|
|
bmstore.update("test2", testnode)
|
|
|
|
self.assertEqual(["test2", "test"], bmstore.lookup_node(testnode))
|
|
|
|
bmstore.remove("test2")
|
|
self.assertEqual(["test"], bmstore.lookup_node(testnode))
|
|
|
|
def testMalformedBookmarks(self):
|
|
bmdir = self.makeTempDir()
|
|
bmstore = bookmarkstore.bookmarkstore(bmdir)
|
|
bmstore.update("test", node.bin("1" * 40))
|
|
bmstore.flush()
|
|
|
|
def truncateFilesInDir(d):
|
|
for f in os.listdir(d):
|
|
with open(os.path.join(d, f), "w"):
|
|
pass
|
|
|
|
truncateFilesInDir(bmdir)
|
|
self.assertRaises(IOError, bookmarkstore.bookmarkstore, bmdir)
|
|
|
|
def testLoadingBookmarks(self):
|
|
bmdir = self.makeTempDir()
|
|
bmstore1 = bookmarkstore.bookmarkstore(bmdir)
|
|
bmstore1.update("test", node.bin("1" * 40))
|
|
bmstore1.flush()
|
|
|
|
bmstore2 = bookmarkstore.bookmarkstore(bmdir)
|
|
self.assertEqual(
|
|
b"\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11",
|
|
bmstore2.lookup_bookmark("test"),
|
|
)
|
|
|
|
def testFlushingBookmarks(self):
|
|
bmdir = self.makeTempDir()
|
|
bmstore = bookmarkstore.bookmarkstore(bmdir)
|
|
bmstore.update("test", node.bin("1" * 40))
|
|
bmstore.flush()
|
|
self.assertTrue(len(os.listdir(bmdir)) > 0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
silenttestrunner.main(__name__)
|