Ejemplo n.º 1
0
    def test_repair(self, connection, close):
        url = 'test-url'

        # test
        sender = Sender(url)
        sender.repair()

        # validation
        close.assert_called_once_with()
        connection.return_value.close.assert_called_once_with()
        connection.return_value.open.assert_called_once_with()
        connection.return_value.session.assert_called_once_with()
        self.assertEqual(sender.session, connection.return_value.session.return_value)
Ejemplo n.º 2
0
    def test_repair(self, connection, close):
        url = 'test-url'

        # test
        sender = Sender(url)
        sender.repair()

        # validation
        close.assert_called_once_with()
        connection.return_value.close.assert_called_once_with()
        connection.return_value.open.assert_called_once_with()
        connection.return_value.session.assert_called_once_with()
        self.assertEqual(sender.session,
                         connection.return_value.session.return_value)