tests/heredoctest.py

tests: unset environment variable P in test-revset2.t (issue6109)

The test tests/test-revset2.t fails the test case

"Test repo.anyrevs with customized revset overrides" (line 1609)

if the environment variable P is set. The test implicitly expects that the
environment, in which it is started, does not export the variable 'P'.

To solve this issue, unset 'P' right before the test commands are run.

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

from __future__ import absolute_import, print_function

import sys

def flush():
    sys.stdout.flush()
    sys.stderr.flush()

globalvars = {}
lines = sys.stdin.readlines()
while lines:
    l = lines.pop(0)
    if l.startswith('SALT'):
        print(l[:-1])
    elif l.startswith('>>> '):
        snippet = l[4:]
        while lines and lines[0].startswith('... '):
            l = lines.pop(0)
            snippet += l[4:]
        c = compile(snippet, '<heredoc>', 'single')
        try:
            flush()
            exec(c, globalvars)
            flush()
        except Exception as inst:
            flush()
            print(repr(inst))