import os
from mercurial import (
    hg,
    ui as uimod,
)

ui = uimod.ui.load()

repo = hg.repository(ui, b'test1', create=True)
os.chdir('test1')

# Add a commit with file "foo"
with open('foo', 'wb') as f:
    f.write(b'foo\n')
repo[None].add([b'foo'])
node1 = repo.commit(text=b'commit1', date=b"0 0")

# Modify file "foo" in a second commit
with open('foo', 'wb') as f:
    f.write(b'foo2\n')
repo.commit(text=b'commit2', date=b"0 0")

# Simulate `hg checkout` of commit 1 by reverting the contents on disk
# and calling dirstate.set_possibly_dirty().
with open('foo', 'wb') as f:
    f.write(b'foo\n')
with repo.wlock(), repo.lock(), repo.transaction(b'test') as tr:
    with repo.dirstate.parentchange():
        repo.dirstate.setparents(node1, repo.nullid)
    repo.dirstate.set_possibly_dirty(b'foo')
    # repo.dirstate.normallookup(b'foo')
    repo.dirstate.write(tr)

repo = hg.repository(ui, b'.')
print("status: %r" % repo.status())