|
|
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 = {b'key1': b'value1', b'Key2': b'value2'}
|
|
|
scmutil.simplekeyvaluefile(self.vfs, b'kvfile').write(dw)
|
|
|
self.assertEqual(sorted(self.vfs.read(b'kvfile').split(b'\n')),
|
|
|
[b'', b'Key2=value2', b'key1=value1'])
|
|
|
dr = scmutil.simplekeyvaluefile(self.vfs, b'kvfile').read()
|
|
|
self.assertEqual(dr, dw)
|
|
|
|
|
|
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)
|
|
|
|
|
|
def testinvalidkeys(self):
|
|
|
d = {b'0key1': b'value1', b'Key2': b'value2'}
|
|
|
with self.assertRaisesRegex(error.ProgrammingError,
|
|
|
'keys must start with a letter.*'):
|
|
|
scmutil.simplekeyvaluefile(self.vfs, b'kvfile').write(d)
|
|
|
|
|
|
d = {b'key1@': b'value1', b'Key2': b'value2'}
|
|
|
with self.assertRaisesRegex(error.ProgrammingError, 'invalid key.*'):
|
|
|
scmutil.simplekeyvaluefile(self.vfs, b'kvfile').write(d)
|
|
|
|
|
|
def testinvalidvalues(self):
|
|
|
d = {b'key1': b'value1', b'Key2': b'value2\n'}
|
|
|
with self.assertRaisesRegex(error.ProgrammingError, 'invalid val.*'):
|
|
|
scmutil.simplekeyvaluefile(self.vfs, b'kvfile').write(d)
|
|
|
|
|
|
def testcorruptedfile(self):
|
|
|
self.vfs.contents[b'badfile'] = b'ababagalamaga\n'
|
|
|
with self.assertRaisesRegex(error.CorruptedState,
|
|
|
'dictionary.*element.*'):
|
|
|
scmutil.simplekeyvaluefile(self.vfs, b'badfile').read()
|
|
|
|
|
|
def testfirstline(self):
|
|
|
dw = {b'key1': b'value1'}
|
|
|
scmutil.simplekeyvaluefile(self.vfs, b'fl').write(dw, firstline=b'1.0')
|
|
|
self.assertEqual(self.vfs.read(b'fl'), b'1.0\nkey1=value1\n')
|
|
|
dr = scmutil.simplekeyvaluefile(
|
|
|
self.vfs, b'fl').read(firstlinenonkeyval=True)
|
|
|
self.assertEqual(dr, {b'__firstline': b'1.0', b'key1': b'value1'})
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
silenttestrunner.main(__name__)
|
|
|
|
