/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to brzlib/tests/transport_util.py

  • Committer: Jelmer Vernooij
  • Date: 2017-05-21 12:41:27 UTC
  • mto: This revision was merged to the branch mainline in revision 6623.
  • Revision ID: jelmer@jelmer.uk-20170521124127-iv8etg0vwymyai6y
s/bzr/brz/ in apport config.

Show diffs side-by-side

added added

removed removed

Lines of Context:
brzlib/tests/transport_util.py
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
 
from ..transport import Transport
18
 
 
19
 
# SFTPTransport is the only bundled transport that properly counts connections
20
 
# at the moment.
21
 
from . import test_sftp_transport
22
 
 
23
 
 
24
 
class TestCaseWithConnectionHookedTransport(test_sftp_transport.TestCaseWithSFTPServer):
 
17
from brzlib.tests import features
 
18
 
 
19
# SFTPTransport offers better performances but relies on paramiko, if paramiko
 
20
# is not available, we fallback to FtpTransport
 
21
if features.paramiko.available():
 
22
    from brzlib.tests import test_sftp_transport
 
23
    from brzlib.transport import sftp, Transport
 
24
    _backing_scheme = 'sftp'
 
25
    _backing_transport_class = sftp.SFTPTransport
 
26
    _backing_test_class = test_sftp_transport.TestCaseWithSFTPServer
 
27
else:
 
28
    from brzlib.transport import ftp, Transport
 
29
    from brzlib.tests import test_ftp_transport
 
30
    _backing_scheme = 'ftp'
 
31
    _backing_transport_class = ftp.FtpTransport
 
32
    _backing_test_class = test_ftp_transport.TestCaseWithFTPServer
 
33
 
 
34
 
 
35
class TestCaseWithConnectionHookedTransport(_backing_test_class):
25
36
 
26
37
    def setUp(self):
27
38
        super(TestCaseWithConnectionHookedTransport, self).setUp()
29
40
 
30
41
    def start_logging_connections(self):
31
42
        Transport.hooks.install_named_hook('post_connect',
32
 
                                           self.connections.append, None)
 
43
            self.connections.append, None)
33
44
 
34
45
    def reset_connections(self):
35
46
        self.connections = []
 
47