"""tests a downstream repo against the current working directory state of the primary suite

Multiple repos can be specified with multiple instances of the -R/--repo option. The

first specified repo is the one being tested. Further repos can be specified to either

override where suites are cloned from or to satisfy --dynamicimports.

"""

parser = ArgumentParser(prog='mx testdownstream')

parser.add_argument('-R', '--repo', dest='repos', action='append', help='URL of downstream repo to clone. First specified repo is the primary repo being tested', required=True, metavar='<url>', default=[])

parser.add_argument('--suitedir', action='store', help='relative directory of suite to test in primary repo (default: . )', default='.', metavar='<path>')

parser.add_argument('--downstream-branch', action='store', help='name of branch to look for in downstream repo(s). '

'Can be specified by DOWNSTREAM_BRANCH environment variable. If not specified, current branch of the primary suite is used.', metavar='<name>')

parser.add_argument('-C', '--mx-command', dest='mxCommands', action='append', help='arguments to an mx command run in primary repo suite (e.g., -C "-v --strict-compliance gate")', default=[], metavar='<args>')

parser.add_argument('-E', '--encoded-space', help='character used to encode a space in an mx command argument. Each instance of this character in an argument will be replaced with a space.', metavar='<char>')

args = parser.parse_args(args)

mxCommands = []

for command in [e.split() for e in args.mxCommands]:

if args.encoded_space:

command = [arg.replace(args.encoded_space, ' ') for arg in command]

mxCommands.append(command)

branch = args.downstream_branch or mx.get_env('DOWNSTREAM_BRANCH', None)

"""

Tests a downstream repo against the current working directory state of `suite`.

:param mx.Suite suite: the suite to test against the downstream repo

:param list repoUrls: URLs of downstream repos to clone, the first of which is the repo being tested

:param str relTargetSuiteDir: directory of the downstream suite to test relative to the top level

directory of the downstream repo being tested

:param list mxCommands: argument lists for the mx commands run in downstream suite being tested

:param str branch: name of branch to look for in downstream repo(s)

"""

assert len(repoUrls) > 0

workDir = join(suite.get_output_root(), 'testdownstream')

# A mirror of each suites in the same repo as `suite` is created with symlinks

rel_mirror = os.path.relpath(suite.dir, mx.SuiteModel.siblings_dir(suite.dir))

in_subdir = os.sep in rel_mirror

suites_in_repo = [suite]

if in_subdir:

base = os.path.dirname(suite.dir)

for e in os.listdir(base):

candidate = join(base, e)

if candidate != suite.dir:

mxDir = mx._is_suite_dir(candidate)

if mxDir:

matches = [s for s in mx.suites() if s.dir == candidate]

if len(matches) == 0:

suites_in_repo.append(mx.SourceSuite(mxDir, primary=False, load=False))

else:

suites_in_repo.append(matches[0])

for suite_in_repo in suites_in_repo:

rel_mirror = os.path.relpath(suite_in_repo.dir, mx.SuiteModel.siblings_dir(suite_in_repo.dir))

mirror = join(workDir, rel_mirror)

if exists(mirror):

shutil.rmtree(mirror)

mx.ensure_dir_exists(mirror)

for f in os.listdir(suite_in_repo.dir):

subDir = join(suite_in_repo.dir, f)

if subDir == suite_in_repo.get_output_root():

continue

src = join(suite_in_repo.dir, f)

dst = join(mirror, f)

mx.logv('[Creating symlink from {} to {}]'.format(dst, src))

relsrc = os.path.relpath(src, os.path.dirname(dst))

os.symlink(relsrc, dst)

targetDir = None

for repoUrl in repoUrls:

# Deduce a target name from the target URL

url = urlparse(repoUrl)

targetName = url.path

if targetName.rfind('/') != -1:

targetName = targetName[targetName.rfind('/') + 1:]

if targetName.endswith('.git'):

targetName = targetName[0:-len('.git')]

repoWorkDir = join(workDir, targetName)

git = mx.GitConfig()

if exists(repoWorkDir):

git.pull(repoWorkDir)

else:

git.clone(repoUrl, repoWorkDir)

# See if there's a matching (non-master) branch and use it if there is

if not branch:

branch = git.git_command(suite.dir, ['rev-parse', '--abbrev-ref', 'HEAD']).strip()

if branch != 'master':

git.git_command(repoWorkDir, ['checkout', branch], abortOnError=False)

if not targetDir:

targetDir = repoWorkDir

assert not isabs(relTargetSuiteDir)

targetSuiteDir = join(targetDir, relTargetSuiteDir)

assert targetSuiteDir.startswith(targetDir)

mxpy = None if suite != mx._mx_suite else join(mirror, 'mx.py')

for command in mxCommands:

mx.logv('[running "mx ' + ' '.join(command) + '" in ' + targetSuiteDir + ']')