~bzr-pqm/bzr/bzr.dev

# Copyright (C) 2009, 2010 Canonical Ltd

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

"""

Facilities to use ftp test servers.

"""

import sys

from bzrlib import tests

from bzrlib.tests import (

    features,

    )

try:

    from bzrlib.tests.ftp_server import medusa_based

    # medusa is bogus starting with python2.6, since we don't support earlier

    # pythons anymore, it's currently useless. There is hope though that the

    # unicode bugs get fixed in the future so we leave it disabled until

    # then. Keeping the framework in place means that only the following line

    # will need to be changed.  The last tests were conducted with medusa-2.0

    # -- vila 20110607

    medusa_available = False

except ImportError:

    medusa_available = False

try:

    from bzrlib.tests.ftp_server import pyftpdlib_based

    pyftpdlib_available = True

except ImportError:

    pyftpdlib_available = False

class _FTPServerFeature(features.Feature):

    """Some tests want an FTP Server, check if one is available.

    Right now, the only way this is available is if one of the following is

    installed:

    - 'medusa': http://www.amk.ca/python/code/medusa.html

    - 'pyftpdlib': http://code.google.com/p/pyftpdlib/

    """

    def _probe(self):

        return medusa_available or pyftpdlib_available

    def feature_name(self):

        return 'FTPServer'

FTPServerFeature = _FTPServerFeature()

class UnavailableFTPTestServer(object):

    """Dummy ftp test server.

    This allows the test suite report the number of tests needing that

    feature. We raise UnavailableFeature from methods before the test server is

    being used. Doing so in the setUp method has bad side-effects (tearDown is

    never called).

    """

    def start_server(self, vfs_server=None):

        pass

    def stop_server(self):

        pass

    def get_url(self):

        raise tests.UnavailableFeature(FTPServerFeature)

    def get_bogus_url(self):

        raise tests.UnavailableFeature(FTPServerFeature)

if medusa_available:

    FTPTestServer = medusa_based.FTPTestServer

elif pyftpdlib_available:

    FTPTestServer = pyftpdlib_based.FTPTestServer

else:

    FTPTestServer = UnavailableFTPTestServer