def test_ctor(connector, key, request, transport, protocol, loop):
    conn = Connection(connector, key, request,
                      transport, protocol, loop)
    assert conn.loop is loop
    assert conn.protocol is protocol
    assert conn.writer is protocol.writer
    conn.close()
Example #2
0
def test_detach_closed(connector, key, protocol, loop):
    conn = Connection(connector, key, protocol, loop)
    conn.release()
    conn.detach()

    assert not connector._release_acquired.called
    assert conn._protocol is None
def test_close(connector, key, protocol, loop) -> None:
    conn = Connection(connector, key, protocol, loop)
    assert not conn.closed
    conn.close()
    assert conn._protocol is None
    connector._release.assert_called_with(key, protocol, should_close=True)
    assert conn.closed
Example #4
0
 def test_close(self):
     conn = Connection(
         self.connector, self.key, self.request,
         self.transport, self.protocol)
     conn.close()
     self.assertTrue(self.transport.close.called)
     self.assertIsNone(conn._transport)
Example #5
0
def test_detach(connector, key, request, transport, protocol, loop):
    conn = Connection(connector, key, request,
                      transport, protocol, loop)
    assert not conn.closed
    conn.detach()
    assert conn._transport is None
    assert not connector._release.called
    assert conn.closed
def test_release_released(connector, key, protocol, loop) -> None:
    conn = Connection(connector, key, protocol, loop)
    conn.release()
    connector._release.reset_mock()
    conn.release()
    assert not protocol.transport.close.called
    assert conn._protocol is None
    assert not connector._release.called
def test_detach(connector, key, protocol, loop) -> None:
    conn = Connection(connector, key, protocol, loop)
    assert not conn.closed
    conn.detach()
    assert conn._protocol is None
    assert connector._release_acquired.called
    assert not connector._release.called
    assert conn.closed
def test_release(connector, key, protocol, loop) -> None:
    conn = Connection(connector, key, protocol, loop)
    assert not conn.closed
    conn.release()
    assert not protocol.transport.close.called
    assert conn._protocol is None
    connector._release.assert_called_with(key, protocol, should_close=False)
    assert conn.closed
Example #9
0
def test_release_released(connector, key, request, transport, protocol, loop):
    conn = Connection(connector, key, request,
                      transport, protocol, loop)
    conn.release()
    connector._release.reset_mock()
    conn.release()
    assert not transport.close.called
    assert conn._transport is None
    assert not connector._release.called
Example #10
0
    def test_no_share_cookies(self):
        connector = aiohttp.BaseConnector(share_cookies=False, loop=self.loop)

        conn = Connection(
            connector, self.key, self.request,
            self.transport, self.protocol, self.loop)
        self.assertEqual(connector.cookies, {})
        conn.share_cookies({'c1': 'cookie1'})
        self.assertEqual(connector.cookies, {})
Example #11
0
 def test_close(self):
     conn = Connection(
         self.connector, self.key, self.request,
         self.transport, self.protocol, self.loop)
     conn.close()
     self.assertIsNone(conn._transport)
     self.connector._release.assert_called_with(
         self.key, self.request, self.transport, self.protocol,
         should_close=True)
Example #12
0
 def test_release(self):
     conn = Connection(
         self.connector, self.key, self.request,
         self.transport, self.protocol)
     conn.release()
     self.assertFalse(self.transport.close.called)
     self.assertIsNone(conn._transport)
     self.connector._release.assert_called_with(
         self.key, self.request, self.transport, self.protocol)
Example #13
0
 def test_detach(self):
     conn = Connection(
         self.connector, self.key, self.request,
         self.transport, self.protocol, self.loop)
     self.assertFalse(conn.closed)
     conn.detach()
     self.assertIsNone(conn._transport)
     self.assertFalse(self.connector._release.called)
     self.assertTrue(conn.closed)
Example #14
0
def test_close(connector, key, request, transport, protocol, loop):
    conn = Connection(connector, key, request,
                      transport, protocol, loop)
    assert not conn.closed
    conn.close()
    assert conn._transport is None
    connector._release.assert_called_with(
        key, request, transport, protocol,
        should_close=True)
    assert conn.closed
Example #15
0
 def test_release_released(self):
     conn = Connection(
         self.connector, self.key, self.request,
         self.transport, self.protocol, self.loop)
     conn.release()
     self.connector._release.reset_mock()
     conn.release()
     self.assertFalse(self.transport.close.called)
     self.assertIsNone(conn._transport)
     self.assertFalse(self.connector._release.called)
def test_callbacks_on_release(connector, key, protocol, loop) -> None:
    conn = Connection(connector, key, protocol, loop)
    notified = False

    def cb():
        nonlocal notified
        notified = True

    conn.add_callback(cb)
    conn.release()
    assert notified
Example #17
0
def test_callbacks_on_detach(connector, key, protocol, loop):
    conn = Connection(connector, key, protocol, loop)
    notified = False

    def cb():
        nonlocal notified
        notified = True

    conn.add_callback(cb)
    conn.detach()
    assert notified
def test_callbacks_exception(connector, key, protocol, loop) -> None:
    conn = Connection(connector, key, protocol, loop)
    notified = False

    def cb1():
        raise Exception

    def cb2():
        nonlocal notified
        notified = True

    conn.add_callback(cb1)
    conn.add_callback(cb2)
    conn.close()
    assert notified
Example #19
0
def test_detach_closed(connector, key, request, transport, protocol, loop):
    conn = Connection(connector, key, request, transport, protocol, loop)
    conn.release()
    conn.detach()

    assert not connector._release_acquired.called
    assert conn._transport is None
def function2169(function130, function1068, function214, function1949):
    var2360 = Connection(function130, function1068, function214, function1949)
    var380 = False

    def function1565():
        raise Exception

    def function787():
        nonlocal notified
        var380 = True

    var2360.add_callback(function1565)
    var2360.add_callback(function787)
    var2360.close()
    assert notified
def function546(function130, function1068, function214, function1949):
    var1045 = Connection(function130, function1068, function214, function1949)
    var1045.release()
    function130._release.reset_mock()
    var1045.release()
    assert (not function214.transport.close.called)
    assert (var1045._protocol is None)
    assert (not function130._release.called)
Example #22
0
def test_release_released(connector, key, protocol, loop) -> None:
    conn = Connection(connector, key, protocol, loop)
    conn.release()
    connector._release.reset_mock()
    conn.release()
    assert not protocol.transport.close.called
    assert conn._protocol is None
    assert not connector._release.called
Example #23
0
def test_release_released(connector, key, request, transport, protocol, loop):
    conn = Connection(connector, key, request, transport, protocol, loop)
    conn.release()
    connector._release.reset_mock()
    conn.release()
    assert not transport.close.called
    assert conn._transport is None
    assert not connector._release.called
Example #24
0
 def test_release_released(self):
     conn = Connection(self.connector, self.key, self.request,
                       self.transport, self.protocol, self.loop)
     conn.release()
     self.connector._release.reset_mock()
     conn.release()
     self.assertFalse(self.transport.close.called)
     self.assertIsNone(conn._transport)
     self.assertFalse(self.connector._release.called)
Example #25
0
def test_del(connector, key, protocol, loop) -> None:
    loop.is_closed.return_value = False
    conn = Connection(connector, key, protocol, loop)
    exc_handler = mock.Mock()
    loop.set_exception_handler(exc_handler)

    with pytest.warns(ResourceWarning):
        del conn
        gc.collect()

    connector._release.assert_called_with(key, protocol, should_close=True)
    msg = {'client_connection': mock.ANY,  # conn was deleted
           'message': 'Unclosed connection'}
    if loop.get_debug():
        msg['source_traceback'] = mock.ANY
    loop.call_exception_handler.assert_called_with(msg)
def function1525(function130, function1068, function214, function1949):
    var263 = Connection(function130, function1068, function214, function1949)
    var1921 = False

    def function806():
        nonlocal notified
        var1921 = True

    var263.add_callback(function806)
    var263.close()
    assert notified
def function2437(function130, function1068, function214, function1949):
    var374 = Connection(function130, function1068, function214, function1949)
    var1711 = False

    def function806():
        nonlocal notified
        var1711 = True

    var374.add_callback(function806)
    var374.release()
    assert notified
def function1717(function130, function1068, function214, function1949):
    var950 = Connection(function130, function1068, function214, function1949)
    var1691 = False

    def function806():
        nonlocal notified
        var1691 = True

    var950.add_callback(function806)
    var950.detach()
    assert notified
Example #29
0
def test_callbacks_on_release(connector, key, protocol, loop) -> None:
    conn = Connection(connector, key, protocol, loop)
    notified = False

    def cb():
        nonlocal notified
        notified = True

    conn.add_callback(cb)
    conn.release()
    assert notified
Example #30
0
def test_callbacks_on_detach(connector, key, protocol, loop):
    conn = Connection(connector, key, protocol, loop)
    notified = False

    def cb():
        nonlocal notified
        notified = True

    conn.add_callback(cb)
    conn.detach()
    assert notified
Example #31
0
def test_callbacks_on_close(connector: Any, key: Any, protocol: Any,
                            loop: Any) -> None:
    conn = Connection(connector, key, protocol, loop)
    notified = False

    def cb() -> None:
        nonlocal notified
        notified = True

    conn.add_callback(cb)
    conn.close()
    assert notified
def function543(function130, function1068, function214, function1949):
    function1949.is_closed.return_value = False
    var309 = Connection(function130, function1068, function214, function1949)
    var570 = mock.Mock()
    function1949.set_exception_handler(var570)
    with pytest.warns(ResourceWarning):
        del conn
        gc.collect()
    function130._release.assert_called_with(function1068,
                                            function214,
                                            should_close=True)
    var4015 = {
        'client_connection': mock.ANY,
        'message': 'Unclosed connection',
    }
    if function1949.get_debug():
        var4015['source_traceback'] = mock.ANY
    function1949.call_exception_handler.assert_called_with(var4015)
def test_del(connector, key, request, transport, protocol, loop, warning):
    conn = Connection(connector, key, request,
                      transport, protocol, loop)
    exc_handler = mock.Mock()
    loop.set_exception_handler(exc_handler)

    with warning(ResourceWarning):
        del conn
        gc.collect()

    connector._release.assert_called_with(key,
                                          request,
                                          transport,
                                          protocol,
                                          should_close=True)
    msg = {'client_connection': mock.ANY,  # conn was deleted
           'message': 'Unclosed connection'}
    if loop.get_debug():
        msg['source_traceback'] = mock.ANY
    exc_handler.assert_called_with(loop, msg)
Example #34
0
def test_callbacks_exception(connector, key, protocol, loop) -> None:
    conn = Connection(connector, key, protocol, loop)
    notified = False

    def cb1():
        raise Exception

    def cb2():
        nonlocal notified
        notified = True

    conn.add_callback(cb1)
    conn.add_callback(cb2)
    conn.close()
    assert notified
Example #35
0
    def test_del(self):
        conn = Connection(
            self.connector, self.key, self.request,
            self.transport, self.protocol, self.loop)
        exc_handler = unittest.mock.Mock()
        self.loop.set_exception_handler(exc_handler)

        with self.assertWarns(ResourceWarning):
            del conn
            gc.collect()

        self.connector._release.assert_called_with(self.key,
                                                   self.request,
                                                   self.transport,
                                                   self.protocol,
                                                   should_close=True)
        msg = {'client_connection': unittest.mock.ANY,  # conn was deleted
               'message': 'Unclosed connection'}
        if self.loop.get_debug():
            msg['source_traceback'] = unittest.mock.ANY
        exc_handler.assert_called_with(self.loop, msg)
def test_ctor(connector, key, protocol, loop) -> None:
    conn = Connection(connector, key, protocol, loop)
    assert conn.loop is loop
    assert conn.protocol is protocol
    assert conn.writer is protocol.writer
    conn.close()
Example #37
0
def test_ctor(connector, key, protocol, loop) -> None:
    conn = Connection(connector, key, protocol, loop)
    with pytest.warns(DeprecationWarning):
        assert conn.loop is loop
    assert conn.protocol is protocol
    conn.close()
Example #38
0
def test_ctor(connector, key, protocol, loop) -> None:
    conn = Connection(connector, key, protocol, loop)
    assert conn.loop is loop
    assert conn.protocol is protocol
    conn.close()
Example #39
0
def test_ctor(connector, key, protocol, loop):
    conn = Connection(connector, key, protocol, loop)
    assert conn.loop is loop
    assert conn.protocol is protocol
    assert conn.writer is protocol.writer
    conn.close()
Example #40
0
 def test_del(self):
     conn = Connection(self.connector, self.key, self.request,
                       self.transport, self.protocol)
     del conn
     self.assertTrue(self.transport.close.called)
def function2180(function130, function1068, function214, function1949):
    var3947 = Connection(function130, function1068, function214, function1949)
    var3947.release()
    var3947.detach()
    assert (not function130._release_acquired.called)
    assert (var3947._protocol is None)
def function1252(function130, function1068, function214, function1949):
    var2745 = Connection(function130, function1068, function214, function1949)
    assert (var2745.function1949 is function1949)
    assert (var2745.function214 is function214)
    assert (var2745.writer is function214.writer)
    var2745.close()
Example #43
0
 def test_close(self):
     conn = Connection(self.connector, self.key, self.request,
                       self.transport, self.protocol)
     conn.close()
     self.assertTrue(self.transport.close.called)
     self.assertIsNone(conn._transport)
Example #44
0
def test_ctor(connector: Any, key: Any, protocol: Any, loop: Any) -> None:
    conn = Connection(connector, key, protocol, loop)
    assert conn.protocol is protocol
    conn.close()