sapling/eden/hg-server/tests/test-simplekeyvaluefile.py
Durham Goode 98d9269874 server: copy hg to a new hg-server directory
Summary:
Create a fork of the Mercurial code that we can use to build server
rpms. The hg servers will continue to exist for a few more months while we move
the darkstorm and ediscovery use cases off them. In the mean time, we want to
start making breaking changes to the client, so let's create a stable copy of
the hg code to produce rpms for the hg servers.

The fork is based off c7770c78d, the latest hg release.

This copies the files as is, then adds some minor tweaks to get it to build:
- Disables some lint checks that appear to be bypassed by path
- sed replace eden/scm with eden/hg-server
- Removed a dependency on scm/telemetry from the edenfs-client tests since
  scm/telemetry pulls in the original eden/scm/lib/configparser which conflicts
  with the hg-server conflict parser.

allow-large-files

Reviewed By: quark-zju

Differential Revision: D27632557

fbshipit-source-id: b2f442f4ec000ea08e4d62de068750832198e1f4
2021-04-09 10:09:06 -07:00

97 lines
3.0 KiB
Python

from __future__ import absolute_import
import unittest
import silenttestrunner
from edenscm.mercurial import error, scmutil
from edenscm.mercurial.pycompat import decodeutf8
from hghave import require
class mockfile(object):
def __init__(self, name, fs):
self.name = name
self.fs = fs
def __enter__(self):
return self
def __exit__(self, *args, **kwargs):
pass
def write(self, text):
self.fs.contents[self.name] = text
def read(self):
return self.fs.contents[self.name]
class mockvfs(object):
def __init__(self):
self.contents = {}
def read(self, path):
return mockfile(path, self).read()
def readutf8(self, path):
return decodeutf8(mockfile(path, self).read())
def readlines(self, path):
# lines need to contain the trailing '\n' to mock the real readlines
return [l for l in mockfile(path, self).read().splitlines(True)]
def __call__(self, path, mode, atomictemp):
return mockfile(path, self)
class testsimplekeyvaluefile(unittest.TestCase):
def setUp(self):
self.vfs = mockvfs()
def testbasicwritingiandreading(self):
dw = {"key1": "value1", "Key2": "value2"}
scmutil.simplekeyvaluefile(self.vfs, "kvfile").write(dw)
self.assertEqual(
sorted(self.vfs.readutf8("kvfile").split("\n")),
["", "Key2=value2", "key1=value1"],
)
dr = scmutil.simplekeyvaluefile(self.vfs, "kvfile").read()
self.assertEqual(dr, dw)
def testinvalidkeys(self):
d = {"0key1": "value1", "Key2": "value2"}
with self.assertRaisesRegex(
error.ProgrammingError, "keys must start with a letter.*"
):
scmutil.simplekeyvaluefile(self.vfs, "kvfile").write(d)
d = {"key1@": "value1", "Key2": "value2"}
with self.assertRaisesRegex(error.ProgrammingError, "invalid key.*"):
scmutil.simplekeyvaluefile(self.vfs, "kvfile").write(d)
def testinvalidvalues(self):
d = {"key1": "value1", "Key2": "value2\n"}
with self.assertRaisesRegex(error.ProgrammingError, "invalid val.*"):
scmutil.simplekeyvaluefile(self.vfs, "kvfile").write(d)
def testcorruptedfile(self):
self.vfs.contents["badfile"] = b"ababagalamaga\n"
with self.assertRaisesRegex(error.CorruptedState, "dictionary.*element.*"):
scmutil.simplekeyvaluefile(self.vfs, "badfile").read()
def testfirstline(self):
dw = {"key1": "value1"}
scmutil.simplekeyvaluefile(self.vfs, "fl").write(dw, firstline="1.0")
self.assertEqual(self.vfs.read("fl"), b"1.0\nkey1=value1\n")
dr = scmutil.simplekeyvaluefile(self.vfs, "fl").read(firstlinenonkeyval=True)
self.assertEqual(dr, {"__firstline": "1.0", "key1": "value1"})
if not hasattr(unittest.TestCase, "assertRaisesRegex"):
unittest.TestCase.assertRaisesRegex = getattr(
unittest.TestCase, "assertRaisesRegexp"
)
if __name__ == "__main__":
silenttestrunner.main(__name__)