tests/fakemergerecord.py

graft: start using the cmdstate class to read and write data to graftstate

This patch replaces the logic to read and write data to graftstate file to use
the state.cmdstate() class.

The previous graftstate format didn't had any version number on top of that, so
we have to catch the CorruptedState error and then read the graftstate in case
of old state files.

This will help us to implement nice additions to graft commands like
`--no-commit`, `--abort`, `--stop` flags.

Passing on test-graft.t shows that things are working fine.

Differential Revision: https://phab.mercurial-scm.org/D3654

# Extension to write out fake unsupported records into the merge state
#
#

from __future__ import absolute_import

from mercurial import (
    merge,
    registrar,
)

cmdtable = {}
command = registrar.command(cmdtable)

@command(b'fakemergerecord',
         [(b'X', b'mandatory', None, b'add a fake mandatory record'),
          (b'x', b'advisory', None, b'add a fake advisory record')], '')
def fakemergerecord(ui, repo, *pats, **opts):
    with repo.wlock():
        ms = merge.mergestate.read(repo)
        records = ms._makerecords()
        if opts.get('mandatory'):
            records.append((b'X', b'mandatory record'))
        if opts.get('advisory'):
            records.append((b'x', b'advisory record'))
        ms._writerecords(records)