Ignore:
Timestamp:
Jul 26, 2016, 8:43:48 PM (4 years ago)
Author:
Stefan Schwarzer <sschwarzer@…>
Branch:
default
Message:
Adapt `test_session.py` for pytest

ticket: 98
File:
1 edited

Legend:

Unmodified
Added
Removed
  • test/test_session.py

    r1564 r1653  
    1 # Copyright (C) 2014, Stefan Schwarzer <sschwarzer@sschwarzer.net>
     1# Copyright (C) 2016, Stefan Schwarzer <sschwarzer@sschwarzer.net>
    22# and ftputil contributors (see `doc/contributors.txt`)
    33# See the file LICENSE for licensing terms.
     
    6262          ftputil.session.session_factory(base_class=MockSession)
    6363        session = factory("host", "user", "password")
    64         self.assertEqual(session.calls, [("connect", "host", 21),
    65                                         ("login", "user", "password")])
     64        assert (session.calls ==
     65                [("connect", "host", 21), ("login", "user", "password")])
    6666
    6767    def test_different_port(self):
     
    7070          ftputil.session.session_factory(base_class=MockSession, port=2121)
    7171        session = factory("host", "user", "password")
    72         self.assertEqual(session.calls, [("connect", "host", 2121),
    73                                         ("login", "user", "password")])
     72        assert (session.calls ==
     73                [("connect", "host", 2121), ("login", "user", "password")])
    7474
    7575    def test_use_passive_mode(self):
     
    8282                                                  use_passive_mode=True)
    8383        session = factory("host", "user", "password")
    84         self.assertEqual(session.calls, [("connect", "host", 21),
    85                                          ("login", "user", "password"),
    86                                          ("set_pasv", True)])
     84        assert session.calls == [("connect", "host", 21),
     85                                 ("login", "user", "password"),
     86                                 ("set_pasv", True)]
    8787        # Active mode
    8888        factory = ftputil.session.session_factory(base_class=MockSession,
    8989                                                  use_passive_mode=False)
    9090        session = factory("host", "user", "password")
    91         self.assertEqual(session.calls, [("connect", "host", 21),
    92                                          ("login", "user", "password"),
    93                                          ("set_pasv", False)])
     91        assert session.calls == [("connect", "host", 21),
     92                                 ("login", "user", "password"),
     93                                 ("set_pasv", False)]
    9494
    9595    def test_encrypt_data_channel(self):
     
    9999                    base_class=EncryptedMockSession)
    100100        session = factory("host", "user", "password")
    101         self.assertEqual(session.calls, [("connect", "host", 21),
    102                                          ("login", "user", "password"),
    103                                          ("prot_p",)])
     101        assert session.calls == [("connect", "host", 21),
     102                                 ("login", "user", "password"),
     103                                 ("prot_p",)]
    104104        #
    105105        factory = ftputil.session.session_factory(
    106106                    base_class=EncryptedMockSession, encrypt_data_channel=True)
    107107        session = factory("host", "user", "password")
    108         self.assertEqual(session.calls, [("connect", "host", 21),
    109                                          ("login", "user", "password"),
    110                                          ("prot_p",)])
     108        assert session.calls == [("connect", "host", 21),
     109                                 ("login", "user", "password"),
     110                                 ("prot_p",)]
    111111        # Without encrypted data channel.
    112112        factory = ftputil.session.session_factory(
    113113                    base_class=EncryptedMockSession, encrypt_data_channel=False)
    114114        session = factory("host", "user", "password")
    115         self.assertEqual(session.calls, [("connect", "host", 21),
    116                                          ("login", "user", "password")])
     115        assert session.calls == [("connect", "host", 21),
     116                                 ("login", "user", "password")]
    117117
    118118    def test_debug_level(self):
     
    121121                                                  debug_level=1)
    122122        session = factory("host", "user", "password")
    123         self.assertEqual(session.calls, [("connect", "host", 21),
    124                                          ("set_debuglevel", 1),
    125                                          ("login", "user", "password")])
     123        assert session.calls == [("connect", "host", 21),
     124                                 ("set_debuglevel", 1),
     125                                 ("login", "user", "password")]
    126126
    127127    def test_m2crypto_session(self):
     
    137137        factory._fix_socket = lambda self: self.add_call("_fix_socket")
    138138        session = factory("host", "user", "password")
    139         self.assertEqual(session.calls, [("connect", "host", 21),
    140                                          ("auth_tls",),
    141                                          ("_fix_socket",),
    142                                          ("login", "user", "password"),
    143                                          ("prot_p",)])
     139        assert session.calls == [("connect", "host", 21),
     140                                 ("auth_tls",),
     141                                 ("_fix_socket",),
     142                                 ("login", "user", "password"),
     143                                 ("prot_p",)]
    144144
    145145
Note: See TracChangeset for help on using the changeset viewer.