test-simplekeyvaluefile.py
90 lines
| 3.0 KiB
| text/x-python
|
PythonLexer
/ tests / test-simplekeyvaluefile.py
|
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): | ||||
|
r32269 | # lines need to contain the trailing '\n' to mock the real readlines | ||
return [l for l in mockfile(path, self).read().splitlines(True)] | ||||
|
r31553 | |||
def __call__(self, path, mode, atomictemp): | ||||
return mockfile(path, self) | ||||
class testsimplekeyvaluefile(unittest.TestCase): | ||||
def setUp(self): | ||||
self.vfs = mockvfs() | ||||
|
r32269 | def testbasicwritingiandreading(self): | ||
|
r37958 | 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() | ||||
|
r32269 | self.assertEqual(dr, dw) | ||
|
r31553 | |||
|
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) | ||||
|
r31553 | def testinvalidkeys(self): | ||
|
r37958 | d = {b'0key1': b'value1', b'Key2': b'value2'} | ||
|
r37733 | with self.assertRaisesRegex(error.ProgrammingError, | ||
|
r32279 | 'keys must start with a letter.*'): | ||
|
r37958 | scmutil.simplekeyvaluefile(self.vfs, b'kvfile').write(d) | ||
|
r32279 | |||
|
r37958 | d = {b'key1@': b'value1', b'Key2': b'value2'} | ||
|
r37733 | with self.assertRaisesRegex(error.ProgrammingError, 'invalid key.*'): | ||
|
r37958 | scmutil.simplekeyvaluefile(self.vfs, b'kvfile').write(d) | ||
|
r31553 | |||
def testinvalidvalues(self): | ||||
|
r37958 | d = {b'key1': b'value1', b'Key2': b'value2\n'} | ||
|
r37733 | with self.assertRaisesRegex(error.ProgrammingError, 'invalid val.*'): | ||
|
r37958 | scmutil.simplekeyvaluefile(self.vfs, b'kvfile').write(d) | ||
|
r31553 | |||
def testcorruptedfile(self): | ||||
|
r37958 | self.vfs.contents[b'badfile'] = b'ababagalamaga\n' | ||
|
r37733 | with self.assertRaisesRegex(error.CorruptedState, | ||
|
r32279 | 'dictionary.*element.*'): | ||
|
r37958 | scmutil.simplekeyvaluefile(self.vfs, b'badfile').read() | ||
|
r31553 | |||
|
r32270 | def testfirstline(self): | ||
|
r37958 | 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')\ | ||||
|
r32270 | .read(firstlinenonkeyval=True) | ||
|
r37958 | self.assertEqual(dr, {b'__firstline': b'1.0', b'key1': b'value1'}) | ||
|
r32270 | |||
|
r31553 | if __name__ == "__main__": | ||
silenttestrunner.main(__name__) | ||||