##// END OF EJS Templates
fix: use templater to substitute values in command string...
fix: use templater to substitute values in command string bytes.format() isn't supported on Python 3. Luckily, our template syntax is similar so we can reuse it. We need a hack to disable \-escapes as '\' is a directory separator on Windows.

File last commit:

r37733:1859b9a7 default
r37792:d6970628 default
Show More
test-simplekeyvaluefile.py
90 lines | 3.0 KiB | text/x-python | PythonLexer
/ tests / test-simplekeyvaluefile.py
Kostia Balytskyi
scmutil: add a simple key-value file helper...
r31553 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):
Kostia Balytskyi
scmutil: add simplekeyvaluefile reading test...
r32269 # lines need to contain the trailing '\n' to mock the real readlines
return [l for l in mockfile(path, self).read().splitlines(True)]
Kostia Balytskyi
scmutil: add a simple key-value file helper...
r31553
def __call__(self, path, mode, atomictemp):
return mockfile(path, self)
class testsimplekeyvaluefile(unittest.TestCase):
def setUp(self):
self.vfs = mockvfs()
Kostia Balytskyi
scmutil: add simplekeyvaluefile reading test...
r32269 def testbasicwritingiandreading(self):
dw = {'key1': 'value1', 'Key2': 'value2'}
scmutil.simplekeyvaluefile(self.vfs, 'kvfile').write(dw)
Kostia Balytskyi
scmutil: add a simple key-value file helper...
r31553 self.assertEqual(sorted(self.vfs.read('kvfile').split('\n')),
['', 'Key2=value2', 'key1=value1'])
Kostia Balytskyi
scmutil: add simplekeyvaluefile reading test...
r32269 dr = scmutil.simplekeyvaluefile(self.vfs, 'kvfile').read()
self.assertEqual(dr, dw)
Kostia Balytskyi
scmutil: add a simple key-value file helper...
r31553
Augie Fackler
cleanup: polyfill assertRaisesRegex so we can avoid assertRaisesRegexp...
r37733 if not getattr(unittest.TestCase, 'assertRaisesRegex', False):
# Python 3.7 deprecates the regex*p* version, but 2.7 lacks
# the regex version.
assertRaisesRegex = (# camelcase-required
unittest.TestCase.assertRaisesRegexp)
Kostia Balytskyi
scmutil: add a simple key-value file helper...
r31553 def testinvalidkeys(self):
d = {'0key1': 'value1', 'Key2': 'value2'}
Augie Fackler
cleanup: polyfill assertRaisesRegex so we can avoid assertRaisesRegexp...
r37733 with self.assertRaisesRegex(error.ProgrammingError,
Gregory Szorc
tests: use context manager form of assertRaises...
r32279 'keys must start with a letter.*'):
scmutil.simplekeyvaluefile(self.vfs, 'kvfile').write(d)
Kostia Balytskyi
scmutil: add a simple key-value file helper...
r31553 d = {'key1@': 'value1', 'Key2': 'value2'}
Augie Fackler
cleanup: polyfill assertRaisesRegex so we can avoid assertRaisesRegexp...
r37733 with self.assertRaisesRegex(error.ProgrammingError, 'invalid key.*'):
Gregory Szorc
tests: use context manager form of assertRaises...
r32279 scmutil.simplekeyvaluefile(self.vfs, 'kvfile').write(d)
Kostia Balytskyi
scmutil: add a simple key-value file helper...
r31553
def testinvalidvalues(self):
d = {'key1': 'value1', 'Key2': 'value2\n'}
Augie Fackler
cleanup: polyfill assertRaisesRegex so we can avoid assertRaisesRegexp...
r37733 with self.assertRaisesRegex(error.ProgrammingError, 'invalid val.*'):
Gregory Szorc
tests: use context manager form of assertRaises...
r32279 scmutil.simplekeyvaluefile(self.vfs, 'kvfile').write(d)
Kostia Balytskyi
scmutil: add a simple key-value file helper...
r31553
def testcorruptedfile(self):
self.vfs.contents['badfile'] = 'ababagalamaga\n'
Augie Fackler
cleanup: polyfill assertRaisesRegex so we can avoid assertRaisesRegexp...
r37733 with self.assertRaisesRegex(error.CorruptedState,
Gregory Szorc
tests: use context manager form of assertRaises...
r32279 'dictionary.*element.*'):
scmutil.simplekeyvaluefile(self.vfs, 'badfile').read()
Kostia Balytskyi
scmutil: add a simple key-value file helper...
r31553
Kostia Balytskyi
scmutil: make simplekeyvaluefile able to have a non-key-value first line...
r32270 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'})
Kostia Balytskyi
scmutil: add a simple key-value file helper...
r31553 if __name__ == "__main__":
silenttestrunner.main(__name__)