Standardize importing of queue class.
This will make it easier to enable gevent support.
This commit is contained in:
@@ -8,10 +8,8 @@
|
||||
|
||||
import socket
|
||||
import threading
|
||||
try:
|
||||
import queue
|
||||
except ImportError:
|
||||
import Queue as queue
|
||||
|
||||
from sleekxmpp.util import Queue
|
||||
|
||||
|
||||
class TestLiveSocket(object):
|
||||
@@ -39,8 +37,8 @@ class TestLiveSocket(object):
|
||||
"""
|
||||
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
self.recv_buffer = []
|
||||
self.recv_queue = queue.Queue()
|
||||
self.send_queue = queue.Queue()
|
||||
self.recv_queue = Queue()
|
||||
self.send_queue = Queue()
|
||||
self.send_queue_lock = threading.Lock()
|
||||
self.recv_queue_lock = threading.Lock()
|
||||
self.is_live = True
|
||||
|
@@ -7,10 +7,8 @@
|
||||
"""
|
||||
|
||||
import socket
|
||||
try:
|
||||
import queue
|
||||
except ImportError:
|
||||
import Queue as queue
|
||||
|
||||
from sleekxmpp.util import Queue
|
||||
|
||||
|
||||
class TestSocket(object):
|
||||
@@ -36,8 +34,8 @@ class TestSocket(object):
|
||||
Same as arguments for socket.socket
|
||||
"""
|
||||
self.socket = socket.socket(*args, **kwargs)
|
||||
self.recv_queue = queue.Queue()
|
||||
self.send_queue = queue.Queue()
|
||||
self.recv_queue = Queue()
|
||||
self.send_queue = Queue()
|
||||
self.is_live = False
|
||||
self.disconnected = False
|
||||
|
||||
|
@@ -8,13 +8,10 @@
|
||||
|
||||
import unittest
|
||||
from xml.parsers.expat import ExpatError
|
||||
try:
|
||||
import Queue as queue
|
||||
except:
|
||||
import queue
|
||||
|
||||
import sleekxmpp
|
||||
from sleekxmpp import ClientXMPP, ComponentXMPP
|
||||
from sleekxmpp.util import Queue
|
||||
from sleekxmpp.stanza import Message, Iq, Presence
|
||||
from sleekxmpp.test import TestSocket, TestLiveSocket
|
||||
from sleekxmpp.exceptions import XMPPError, IqTimeout, IqError
|
||||
@@ -338,7 +335,7 @@ class SleekTest(unittest.TestCase):
|
||||
|
||||
# We will use this to wait for the session_start event
|
||||
# for live connections.
|
||||
skip_queue = queue.Queue()
|
||||
skip_queue = Queue()
|
||||
|
||||
if socket == 'mock':
|
||||
self.xmpp.set_socket(TestSocket())
|
||||
|
Reference in New Issue
Block a user