"""Tests for distutils.pypirc.pypirc."""

import sys

import os

import unittest

from distutils.core import PyPIRCCommand

from distutils.core import Distribution

from distutils.tests import support

PYPIRC = """\

[distutils]

index-servers =

    server1

    server2

[server1]

username:me

password:secret

[server2]

username:meagain

password: secret

realm:acme

repository:http://another.pypi/

"""

PYPIRC_OLD = """\

[server-login]

username:tarek

password:secret

"""

class PyPIRCCommandTestCase(support.TempdirManager, unittest.TestCase):

    def setUp(self):

        """Patches the environment."""

        if os.environ.has_key('HOME'):

            self._old_home = os.environ['HOME']

        else:

            self._old_home = None

        curdir = os.path.dirname(__file__)

        os.environ['HOME'] = curdir

        self.rc = os.path.join(curdir, '.pypirc')

        self.dist = Distribution()

        class command(PyPIRCCommand):

            def __init__(self, dist):

                PyPIRCCommand.__init__(self, dist)

            def initialize_options(self):

                pass

            finalize_options = initialize_options

        self._cmd = command

    def tearDown(self):

        """Removes the patch."""

        if self._old_home is None:

            del os.environ['HOME']

        else:

            os.environ['HOME'] = self._old_home

        if os.path.exists(self.rc):

            os.remove(self.rc)

    def test_server_registration(self):

        # This test makes sure PyPIRCCommand knows how to:

        # 1. handle several sections in .pypirc

        # 2. handle the old format

        # new format

        f = open(self.rc, 'w')

        try:

            f.write(PYPIRC)

        finally:

            f.close()

        cmd = self._cmd(self.dist)

        config = cmd._read_pypirc()

        config = config.items()

        config.sort()

        waited = [('password', 'secret'), ('realm', 'pypi'),

                  ('repository', 'http://pypi.python.org/pypi'),

                  ('server', 'server1'), ('username', 'me')]

        self.assertEquals(config, waited)

        # old format

        f = open(self.rc, 'w')

        f.write(PYPIRC_OLD)

        f.close()

        config = cmd._read_pypirc()

        config = config.items()

        config.sort()

        waited = [('password', 'secret'), ('realm', 'pypi'),

                  ('repository', 'http://pypi.python.org/pypi'),

                  ('server', 'server-login'), ('username', 'tarek')]

        self.assertEquals(config, waited)

def test_suite():

    return unittest.makeSuite(PyPIRCCommandTestCase)

if __name__ == "__main__":

    unittest.main(defaultTest="test_suite")