sapling/tests/test-simplekeyvaluefile.py
Jun Wu 584656dff3 codemod: join the auto-formatter party
Summary:
Turned on the auto formatter. Ran `arc lint --apply-patches --take BLACK **/*.py`.
Then run `arc lint` again so some other autofixers like spellchecker etc. looked
at the code base. Manually accept the changes whenever they make sense, or use
a workaround (ex. changing "dict()" to "dict constructor") where autofix is false
positive. Disabled linters on files that are hard (i18n/polib.py) to fix, or less
interesting to fix (hgsubversion tests), or cannot be fixed without breaking
OSS build (FBPYTHON4).

Conflicted linters (test-check-module-imports.t, part of test-check-code.t,
test-check-pyflakes.t) are removed or disabled.

Duplicated linters (test-check-pyflakes.t, test-check-pylint.t) are removed.

An issue of the auto-formatter is lines are no longer guarnateed to be <= 80
chars. But that seems less important comparing with the benefit auto-formatter
provides.

As we're here, also remove test-check-py3-compat.t, as it is currently broken
if `PYTHON3=/bin/python3` is set.

Reviewed By: wez, phillco, simpkins, pkaush, singhsrb

Differential Revision: D8173629

fbshipit-source-id: 90e248ae0c5e6eaadbe25520a6ee42d32005621b
2018-05-25 22:17:29 -07:00

90 lines
2.7 KiB
Python

from __future__ import absolute_import
import unittest
import silenttestrunner
from mercurial import error, scmutil
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 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.read("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.assertRaisesRegexp(
error.ProgrammingError, "keys must start with a letter.*"
):
scmutil.simplekeyvaluefile(self.vfs, "kvfile").write(d)
d = {"key1@": "value1", "Key2": "value2"}
with self.assertRaisesRegexp(error.ProgrammingError, "invalid key.*"):
scmutil.simplekeyvaluefile(self.vfs, "kvfile").write(d)
def testinvalidvalues(self):
d = {"key1": "value1", "Key2": "value2\n"}
with self.assertRaisesRegexp(error.ProgrammingError, "invalid val.*"):
scmutil.simplekeyvaluefile(self.vfs, "kvfile").write(d)
def testcorruptedfile(self):
self.vfs.contents["badfile"] = "ababagalamaga\n"
with self.assertRaisesRegexp(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"), "1.0\nkey1=value1\n")
dr = scmutil.simplekeyvaluefile(self.vfs, "fl").read(firstlinenonkeyval=True)
self.assertEqual(dr, {"__firstline": "1.0", "key1": "value1"})
if __name__ == "__main__":
silenttestrunner.main(__name__)