def test_response_lose_connection(self): def connection_made(transport): transport.close() obj = object() self.protocol.connection_made = connection_made self._run_test( self.t, [TransportMock.Close(response=TransportMock.LoseConnection(obj))]) self.assertSequenceEqual(self.protocol.mock_calls, [unittest.mock.call.connection_lost(obj)])