mirror of
https://github.com/facebook/sapling.git
synced 2024-10-10 00:45:18 +03:00
f42333f17c
Summary: Formats a subset of opted-in Python files in fbsource. Black formatting was applied first, which is guaranteed safe as the AST will not have changed during formatting. Pyfmt was then run, which also includes import sorting. The changes from isort were manually reviewed, and some potentially dangerous changes were reverted, and the directive was added to those files. A final run of pyfmt shows no more changes to be applied. Reviewed By: zertosh Differential Revision: D24101830 fbshipit-source-id: 0f2616873117a821dbc6cfb6d8e4f64f4420312b
97 lines
3.0 KiB
Python
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__)
|