tests/test-simplekeyvaluefile.py

obsolescence: add test for the "branch replacement" logic during push, case B3

Mercurial checks for the introduction of new heads on push. Evolution comes
into play to detect if existing branches on the server are being replaced by
some of the new one we push.

The current code for this logic is very basic (eg: issue4354) and was poorly
tested. We have a better implementation coming in the evolve extension fixing
these issues and with more serious tests coverage. In the process of upstreaming
this improved logic, we start with adding the test case that are already passing
with the current implementation. Once they are all in, we'll upstream the better
implementation and the extra test cases.

See inline documentation for details about the test case added in this
changeset.

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):
        return mockfile(path, self).read().split('\n')

    def __call__(self, path, mode, atomictemp):
        return mockfile(path, self)

class testsimplekeyvaluefile(unittest.TestCase):
    def setUp(self):
        self.vfs = mockvfs()

    def testbasicwriting(self):
        d = {'key1': 'value1', 'Key2': 'value2'}
        scmutil.simplekeyvaluefile(self.vfs, 'kvfile').write(d)
        self.assertEqual(sorted(self.vfs.read('kvfile').split('\n')),
                         ['', 'Key2=value2', 'key1=value1'])

    def testinvalidkeys(self):
        d = {'0key1': 'value1', 'Key2': 'value2'}
        self.assertRaises(error.ProgrammingError,
                          scmutil.simplekeyvaluefile(self.vfs, 'kvfile').write,
                          d)
        d = {'key1@': 'value1', 'Key2': 'value2'}
        self.assertRaises(error.ProgrammingError,
                          scmutil.simplekeyvaluefile(self.vfs, 'kvfile').write,
                          d)

    def testinvalidvalues(self):
        d = {'key1': 'value1', 'Key2': 'value2\n'}
        self.assertRaises(error.ProgrammingError,
                          scmutil.simplekeyvaluefile(self.vfs, 'kvfile').write,
                          d)

    def testcorruptedfile(self):
        self.vfs.contents['badfile'] = 'ababagalamaga\n'
        self.assertRaises(error.CorruptedState,
                          scmutil.simplekeyvaluefile(self.vfs, 'badfile').read)

if __name__ == "__main__":
    silenttestrunner.main(__name__)