# -*- coding: utf-8 -*-

#!/usr/bin/env python

import os

import sys

# This import is not the real run of ah_bootstrap for the purposes of the test,

# so we need to preserve the command-line arguments otherwise these get eaten

# up by this import

args = sys.argv[:]

import ah_bootstrap

sys.argv = args

{extra}

# reset the name of the package installed by ah_boostrap to

# _astropy_helpers_test_--this will prevent any confusion by pkg_resources with

# any already installed packages named astropy_helpers

# We also disable auto-upgrade by default

ah_bootstrap.DIST_NAME = 'astropy-helpers-test'

ah_bootstrap.PACKAGE_NAME = '_astropy_helpers_test_'

ah_bootstrap.AUTO_UPGRADE = False

ah_bootstrap.DOWNLOAD_IF_NEEDED = False

try:

    ah_bootstrap.BOOTSTRAPPER = ah_bootstrap._Bootstrapper.main()

    ah_bootstrap.use_astropy_helpers({args})

finally:

    ah_bootstrap.DIST_NAME = 'astropy-helpers'

    ah_bootstrap.PACKAGE_NAME = 'astropy_helpers'

    ah_bootstrap.AUTO_UPGRADE = True

    ah_bootstrap.DOWNLOAD_IF_NEEDED = True

# Kind of a hacky way to do this, but this assertion is specifically

# for test_check_submodule_no_git

# TODO: Rework the tests in this module so that it's easier to test specific

# behaviors of ah_bootstrap for each test

assert '--no-git' not in sys.argv

import _astropy_helpers_test_

filename = os.path.abspath(_astropy_helpers_test_.__file__)

filename = filename.replace('.pyc', '.py')  # More consistent this way

# We print out variables that are needed in tests below in JSON

import json

data = {{}}

data['filename'] = filename

data['ah_bootstrap.BOOTSTRAPPER.use_git'] = ah_bootstrap.BOOTSTRAPPER.use_git

print(json.dumps(data))

"""

# The behavior checked in some of the tests depends on the version of

# setuptools

    # We need to use LooseVersion here instead of StrictVersion since developer

    # versions of setuptools ('35.0.2.post20170530') don't satisfy the

    # StrictVersion criteria even though they satisfy PEP440

    # Broken setuptools? ¯\_(ツ)_/¯

    """

    Tests importing _astropy_helpers_test_ from a submodule in a git

    repository.  This tests actually performing a fresh clone of the repository

    without the submodule initialized, and that importing astropy_helpers in

    that context works transparently after calling

    `ah_boostrap.use_astropy_helpers`.

    """

        # Write a test setup.py that uses ah_bootstrap; it also ensures that

        # any previous reference to astropy_helpers is first wiped from

        # sys.modules

        # Add our own clone of the astropy_helpers repo as a submodule named

        # astropy_helpers

                        '_astropy_helpers_test_'])

        # Creates a clone of our test repo in the directory 'clone'

        # Ensure that the astropy_helpers used by the setup.py is the one that

        # was imported from git submodule

                                             '_astropy_helpers_test_',

                                             '__init__.py')))

                                            monkeypatch):

    """

    Regression test for https://github.com/astropy/astropy/issues/2749

    Runs test_bootstrap_from_submodule but with missing locale/language

    settings.

    """

                                             monkeypatch):

    """

    Additional regression test for

    https://github.com/astropy/astropy/issues/2749

    """

    # Test also with bad LC_CTYPE a la http://bugs.python.org/issue18378

class UpgradeError(Exception):

    pass

def _do_upgrade(*args, **kwargs):

    raise UpgradeError()

ah_bootstrap._Bootstrapper._do_upgrade = _do_upgrade

"""

    """

    Tests that when importing astropy_helpers from a submodule, it is still

    recognized as a submodule even when using the --no-git option.

    In particular this ensures that the auto-upgrade feature is not activated.

    """

        # Write a test setup.py that uses ah_bootstrap; it also ensures that

        # any previous reference to astropy_helpers is first wiped from

        # sys.modules

        # Add our own clone of the astropy_helpers repo as a submodule named

        # astropy_helpers

                        '_astropy_helpers_test_'])

                        '_astropy_helpers_test_ from a git submodule')

        # Ensure that the no-git option was in fact set

    """

    Tests simply bundling a copy of the astropy_helpers source code in its

    entirety bundled directly in the source package and not in an archive.

    """

        # Ensure that the astropy_helpers used by the setup.py is the one that

        # was imported from git submodule

                                             '_astropy_helpers_test_',

                                             '__init__.py')))

    """

    Tests importing _astropy_helpers_test_ from a .tar.gz source archive

    shipped alongside the package that uses it.

    """

    # Make a source distribution of the test package

                  ['sdist', '--dist-dir=dist', '--formats=gztar'])

        # Write a test setup.py that uses ah_bootstrap; it also ensures that

        # any previous reference to astropy_helpers is first wiped from

        # sys.modules

        # Installation from the .tar.gz should have resulted in a .egg

        # directory that the _astropy_helpers_test_ package was imported from

                                          '__init__.py')))

    """

    Tests the case where astropy_helpers was not actually included in a

    package, or is otherwise missing, and we need to "download" it.

    This does not test actually downloading from the internet--this is normally

    done through setuptools' easy_install command which can also install from a

    source archive.  From the point of view of ah_boostrap the two actions are

    equivalent, so we can just as easily simulate this by providing a setup.cfg

    giving the path to a source archive to "download" (as though it were a

    URL).

    """

    # Ensure ah_bootstrap is imported from the local directory

    # Make a source distribution of the test package

                  ['sdist', '--dist-dir=dist', '--formats=gztar'])

            args='download_if_needed=True', extra=''))

            [easy_install]

            find_links = {find_links}

        """.format(find_links=str(dist_dir))))

        # easy_install should have worked by 'installing' astropy_helpers as a

        # .egg in the current directory

                                          '__init__.py')))

from setuptools.package_index import PackageIndex

class FakePackageIndex(PackageIndex):

    def __init__(self, *args, **kwargs):

        PackageIndex.__init__(self, *args, **kwargs)

        self.to_scan = {dists}

    def find_packages(self, requirement):

        # no-op

        pass

ah_bootstrap.PackageIndex = FakePackageIndex

"""

    # Make a test package that uses _astropy_helpers_test_

        # This will be used to later to fake downloading the upgrade package

            [easy_install]

            find_links = {find_links}

        """.format(find_links=str(dist_dir))))

    # Make additional "upgrade" versions of the _astropy_helpers_test_

    # package--one of them is version 0.2 and the other is version 0.1.1.  The

    # auto-upgrade should ignore version 0.2 but use version 0.1.1.

    # For each upgrade package go ahead and build a source distribution of it

    # and copy that source distribution to a dist directory we'll use later to

    # simulate a 'download'

                      ['sdist', '--dist-dir=dist', '--formats=gztar'])

                                        extra=EXTRA_PACKAGE_INDEX.format(dists=dists))

        # Now run the source setup.py; this test is similar to

        # test_download_if_needed, but we explicitly check that the correct

        # *version* of _astropy_helpers_test_ was used

                                          '__init__.py')))

    """

    Helper utility used by some tests to get the list of egg archive files

    in a local directory.

    """

    else: