mirror of
https://github.com/facebook/sapling.git
synced 2024-10-12 01:39:21 +03:00
9dc21f8d0b
Summary: D13853115 adds `edenscm/` to `sys.path` and code still uses `import mercurial`. That has nasty problems if both `import mercurial` and `import edenscm.mercurial` are used, because Python would think `mercurial.foo` and `edenscm.mercurial.foo` are different modules so code like `try: ... except mercurial.error.Foo: ...`, or `isinstance(x, mercurial.foo.Bar)` would fail to handle the `edenscm.mercurial` version. There are also some module-level states (ex. `extensions._extensions`) that would cause trouble if they have multiple versions in a single process. Change imports to use the `edenscm` so ideally the `mercurial` is no longer imported at all. Add checks in extensions.py to catch unexpected extensions importing modules from the old (wrong) locations when running tests. Reviewed By: phillco Differential Revision: D13868981 fbshipit-source-id: f4e2513766957fd81d85407994f7521a08e4de48
98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
from __future__ import absolute_import
|
|
|
|
import os
|
|
import random
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
|
|
import silenttestrunner
|
|
from edenscm.mercurial import node
|
|
from edenscm.mercurial.rust import bookmarkstore
|
|
|
|
|
|
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.assertEquals(
|
|
"\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.assertEquals(
|
|
"\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.assertEquals(["test2", "test"], bmstore.lookup_node(testnode))
|
|
|
|
bmstore.remove("test2")
|
|
self.assertEquals(["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.assertEquals(
|
|
"\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__)
|