EmbeddedPkg: Extend NvVarStoreFormattedLib LIBRARY_CLASS

[mirror_edk2.git] / AppPkg / Applications / Python / Python-2.7.2 / Lib / distutils / tests / test_config.py

"""Tests for distutils.pypirc.pypirc."""\r
import sys\r
import os\r
import unittest\r
import tempfile\r
import shutil\r
\r
from distutils.core import PyPIRCCommand\r
from distutils.core import Distribution\r
from distutils.log import set_threshold\r
from distutils.log import WARN\r
\r
from distutils.tests import support\r
from test.test_support import run_unittest\r
\r
PYPIRC = """\\r
[distutils]\r
\r
index-servers =\r
    server1\r
    server2\r
\r
[server1]\r
username:me\r
password:secret\r
\r
[server2]\r
username:meagain\r
password: secret\r
realm:acme\r
repository:http://another.pypi/\r
"""\r
\r
PYPIRC_OLD = """\\r
[server-login]\r
username:tarek\r
password:secret\r
"""\r
\r
WANTED = """\\r
[distutils]\r
index-servers =\r
    pypi\r
\r
[pypi]\r
username:tarek\r
password:xxx\r
"""\r
\r
\r
class PyPIRCCommandTestCase(support.TempdirManager,\r
                            support.LoggingSilencer,\r
                            support.EnvironGuard,\r
                            unittest.TestCase):\r
\r
    def setUp(self):\r
        """Patches the environment."""\r
        super(PyPIRCCommandTestCase, self).setUp()\r
        self.tmp_dir = self.mkdtemp()\r
        os.environ['HOME'] = self.tmp_dir\r
        self.rc = os.path.join(self.tmp_dir, '.pypirc')\r
        self.dist = Distribution()\r
\r
        class command(PyPIRCCommand):\r
            def __init__(self, dist):\r
                PyPIRCCommand.__init__(self, dist)\r
            def initialize_options(self):\r
                pass\r
            finalize_options = initialize_options\r
\r
        self._cmd = command\r
        self.old_threshold = set_threshold(WARN)\r
\r
    def tearDown(self):\r
        """Removes the patch."""\r
        set_threshold(self.old_threshold)\r
        super(PyPIRCCommandTestCase, self).tearDown()\r
\r
    def test_server_registration(self):\r
        # This test makes sure PyPIRCCommand knows how to:\r
        # 1. handle several sections in .pypirc\r
        # 2. handle the old format\r
\r
        # new format\r
        self.write_file(self.rc, PYPIRC)\r
        cmd = self._cmd(self.dist)\r
        config = cmd._read_pypirc()\r
\r
        config = config.items()\r
        config.sort()\r
        waited = [('password', 'secret'), ('realm', 'pypi'),\r
                  ('repository', 'http://pypi.python.org/pypi'),\r
                  ('server', 'server1'), ('username', 'me')]\r
        self.assertEqual(config, waited)\r
\r
        # old format\r
        self.write_file(self.rc, PYPIRC_OLD)\r
        config = cmd._read_pypirc()\r
        config = config.items()\r
        config.sort()\r
        waited = [('password', 'secret'), ('realm', 'pypi'),\r
                  ('repository', 'http://pypi.python.org/pypi'),\r
                  ('server', 'server-login'), ('username', 'tarek')]\r
        self.assertEqual(config, waited)\r
\r
    def test_server_empty_registration(self):\r
        cmd = self._cmd(self.dist)\r
        rc = cmd._get_rc_file()\r
        self.assertTrue(not os.path.exists(rc))\r
        cmd._store_pypirc('tarek', 'xxx')\r
        self.assertTrue(os.path.exists(rc))\r
        f = open(rc)\r
        try:\r
            content = f.read()\r
            self.assertEqual(content, WANTED)\r
        finally:\r
            f.close()\r
\r
def test_suite():\r
    return unittest.makeSuite(PyPIRCCommandTestCase)\r
\r
if __name__ == "__main__":\r
    run_unittest(test_suite())\r