Python 3: comment tester les exceptions avec?

Question

J'ai des problèmes pour tester les exceptions qui seraient levées dans un avec python 3.4. Je ne peux pas obtenir l'exécution des tests pour cette paix de code:

import logging 
... 
class Foo(object): 
    ... 
    def foo(self, src, dst): 
     try: 
      with pysftp.Connection(self._host, username=self._username, password=self._password) as connection: 
       connection.put(src, dst) 
       connection.close() 
     except (
       ConnectionException, 
       CredentialException, 
       SSHException, 
       AuthenticationException, 
       HostKeysException, 
       PasswordRequiredException 
     ) as e: 
      self._log.error(e) 

Et voilà comment je veux le tester:

import logging 
... 
class TestFoo(TestCase): 
    @parameterized.expand([ 
     ('ConnectionException', ConnectionException), 
     ('CredentialException', CredentialException), 
     ('SSHException', SSHException), 
     ('AuthenticationException', AuthenticationException), 
     ('HostKeysException', HostKeysException), 
     ('PasswordRequiredException', PasswordRequiredException), 
    ]) 
    @patch('pysftp.Connection', spec_set=pysftp.Connection) 
    def test_foo_exceptions(self, _, ex, sftp_mock): 
     """ 
     NOTE: take a look at: 
       http://stackoverflow.com/questions/37014904/mocking-python-class-in-unit-test-and-verifying-an-instance 
       to get an understanding of __enter__ and __exit__ 
     """ 
     sftp_mock.return_value = Mock(
      spec=pysftp.Connection, 
      side_effect=ex, 
      __enter__ = lambda self: self, 
      __exit__ = lambda *args: None 
     ) 
     foo = Foo('host', 'user', 'pass', Mock(spec_set=logging.Logger)) 
     foo.foo('src', 'dst') 
     self.assertEqual(foo._log.error.call_count, 1) 

Mais il échoue - Sortie:

Failure 
... 
AssertionError: 0 != 1 

Answer 1

Votre objet sftp_mock.return_value n'est jamais appelé, donc le side_effect n'est jamais déclenché et aucune exception n'est déclenchée. Il ne serait appelé que si la valeur de retour de pysftp.Connection(...) était elle-même appelée à nouveau.

Définir l'effet secondaire directement sur la maquette:

sftp_mock.side_effect = ex 

Notez que maintenant l'expression pysftp.Connection(...) soulève l'exception et il n'importe plus que la valeur de retour de cette expression aurait été utilisée comme contexte gestionnaire dans une instruction with.

Notez que vos exceptions se plaindront de ne pas avoir d'arguments; passer cas de vos exceptions, pas le type:

@parameterized.expand([ 
    ('ConnectionException', ConnectionException('host', 1234)), 
    # ... etc. 
])