Standardize importing of queue class.

This will make it easier to enable gevent support.
This commit is contained in:
Lance Stout
2012-07-24 02:39:54 -07:00
parent 352ee2f2fd
commit 3e43b36a9d
8 changed files with 47 additions and 42 deletions

View File

@@ -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

View File

@@ -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

View File

@@ -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())