← Back to branch summary

~ubuntu-branches/ubuntu/precise/pyzmq/precise

« back to all changes in this revision

Viewing changes to zmq/tests/__init__.py

  • Committer: Bazaar Package Importer
  • Author(s): Piotr Ożarowski
  • Date: 2011-02-15 09:08:36 UTC
  • mfrom: (2.1.2 experimental)
  • Revision ID: james.westby@ubuntu.com-20110215090836-phh4slym1g6muucn
Tags: 2.0.10.1-2
* Team upload.
* Upload to unstable
* Add Breaks: ${python:Breaks}

Show diffs side-by-side

added added

removed removed

Lines of Context:
zmq/tests/__init__.py
21
21
# Imports
22
22
#-----------------------------------------------------------------------------
23
23
 
 
24
import sys
 
25
 
24
26
from unittest import TestCase
25
27
 
26
28
import zmq
27
29
 
 
30
try:
 
31
    from nose import SkipTest
 
32
except ImportError:
 
33
    class SkipTest(Exception):
 
34
        pass
 
35
 
28
36
#-----------------------------------------------------------------------------
29
37
# Utilities
30
38
#-----------------------------------------------------------------------------
31
39
 
32
40
 
33
41
class BaseZMQTestCase(TestCase):
34
 
 
 
42
    _pgm = None
 
43
    
35
44
    def setUp(self):
36
45
        self.context = zmq.Context()
37
 
 
 
46
        self.sockets = []
 
47
    
 
48
    def tearDown(self):
 
49
        while self.sockets:
 
50
            sock = self.sockets.pop()
 
51
            sock.close()
 
52
        self.context.term()
 
53
        del self.context
 
54
    
 
55
    def has_pgm(self):
 
56
        if self._pgm is None:
 
57
            s = self.context.socket(zmq.PUB)
 
58
            try:
 
59
                s.bind('epgm://:12345')
 
60
            except zmq.ZMQError:
 
61
                e = sys.exc_info()[1]
 
62
                if e.errno == zmq.EPROTONOSUPPORT:
 
63
                    self._pgm = False
 
64
                else:
 
65
                    raise e
 
66
            else:
 
67
                self._pgm = True
 
68
        return self._pgm
 
69
    
 
70
    def skip_if_pgm(self):
 
71
        major,minor = zmq.zmq_version().split('.',2)[:2]
 
72
        if self.has_pgm() and major == '2' and minor == '0':
 
73
            raise SkipTest("Known Failure on 0MQ 2.0.x+PGM")
 
74
    
38
75
    def create_bound_pair(self, type1, type2, interface='tcp://127.0.0.1'):
39
76
        """Create a bound socket pair using a random port."""
40
77
        s1 = zmq.Socket(self.context, type1)
41
78
        port = s1.bind_to_random_port(interface)
42
79
        s2 = zmq.Socket(self.context, type2)
43
80
        s2.connect('%s:%s' % (interface, port))
 
81
        self.sockets.extend([s1,s2])
44
82
        return s1, s2
45
83
 
46
84
    def ping_pong(self, s1, s2, msg):
64
102
        o3 = s1.recv_pyobj()
65
103
        return o3
66
104
 
67
 
    def assertRaisesErrno(self, errno, func, *args):
 
105
    def assertRaisesErrno(self, errno, func, *args, **kwargs):
68
106
        try:
69
 
            func(*args)
70
 
        except zmq.ZMQError, e:
 
107
            func(*args, **kwargs)
 
108
        except zmq.ZMQError:
 
109
            e = sys.exc_info()[1]
71
110
            self.assertEqual(e.errno, errno, "wrong error raised, expected '%s' \
72
111
got '%s'" % (zmq.ZMQError(errno), zmq.ZMQError(e.errno)))
73
112
        else: