← Back to branch summary

~azzar1/unity/add-show-desktop-key

« back to all changes in this revision

Viewing changes to ivle/tests/test_chat.py

  • Committer: William Grant
  • Date: 2010-02-18 06:43:57 UTC
  • Revision ID: grantw@unimelb.edu.au-20100218064357-a43eajvau1gpzw29
Add ivle.tests, move ivle.test_chat into there, and be slightly more idiomatic.

Show diffs side-by-side

added added

removed removed

Lines of Context:
ivle/tests/test_chat.py
16
16
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17
17
 
18
18
import socket
19
 
import os
20
 
import random
21
 
 
22
 
try:
23
 
    import json
24
 
except ImportError:
25
 
    import simplejson as json
26
 
 
 
19
 
 
20
from nose import with_setup
27
21
from nose.tools import assert_equal, raises
28
22
 
29
23
import ivle.chat
68
62
        assert_equal(ivle.chat.recv_netstring(self.s2), SIMPLESTRING)
69
63
 
70
64
    @raises(socket.timeout)
71
 
    def test_invalid_short_netstring(self):
 
65
    def test_short_netstring(self):
72
66
        self.s1.sendall("1234:not that long!,")
73
67
        assert ivle.chat.recv_netstring(self.s2) is None
74
68
 
75
69
    @raises(ivle.chat.ProtocolError)
76
 
    def test_invalid_long_netstring(self):
 
70
    def test_long_netstring(self):
77
71
        self.s1.sendall("5:not that short!,")
78
72
        assert ivle.chat.recv_netstring(self.s2) is None
79
73
 
80
 
    def test_long_netstring(self):
81
 
        # XXX: send() may block if this is too big
82
 
        msg = os.urandom(50000)
83
 
        ivle.chat.send_netstring(self.s1, msg)
84
 
        assert ivle.chat.recv_netstring(self.s2) == msg
85
 
 
86
 
    def test_multiple_netstrings(self):
87
 
        messages = []
88
 
        for i in range(10):
89
 
            message = os.urandom(random.randint(0,20))
90
 
            messages.append(message)
91
 
            ivle.chat.send_netstring(self.s1, message)
92
 
        for i in range(10):
93
 
            assert_equal(ivle.chat.recv_netstring(self.s2), messages[i])
94
 
 
95
 
    def test_encode(self):
96
 
        """Check that we correctly encode a basic object
97
 
        """
98
 
        MESSAGE = {}
99
 
        MAGIC = "3EE"
100
 
        content = json.dumps(MESSAGE)
101
 
        # Digest can be formed with `echo -n "${content}${MAGIC}" | md5sum`
102
 
        DIGEST = '2b59b68e1ac0852b87fb7e64946f2658'
103
 
        expected = {'digest': DIGEST,
104
 
                'content': content}
105
 
        encoded = ivle.chat.encode(MESSAGE, MAGIC)
106
 
        assert_equal(json.loads(encoded), expected)
107
 
 
108
 
    def test_encode_decode(self):
109
 
        """Check that a round trip encoding and decoding works
110
 
        """
111
 
        MESSAGE = {'message': 'Hello, world'}
112
 
        MAGIC = "MagicString"
113
 
        encoded = ivle.chat.encode(MESSAGE, MAGIC)
114
 
        decoded = ivle.chat.decode(encoded, MAGIC)
115
 
        assert_equal(decoded, MESSAGE)
116
 
 
117
 
    @raises(ivle.chat.ProtocolError)
118
 
    def test_decode_bad_magic(self):
119
 
        """Check that a bad digest causes a ProtocolError to be raised
120
 
        """
121
 
        CHATMESSAGE = ('{"content": "{\\"a\\": \\"b\\"}", "digest": ' +
122
 
                '"eb860a5fe8fdbef19ffb79e3a5c47113"}')
123
 
        CORRECTMAGIC = "AEIOU"
124
 
        INCORRECTMAGIC = "ABCDE"
125
 
 
126
 
        # Check our "correct" string decodes without a ProtocolError
127
 
        try:
128
 
            ivle.chat.decode(CHATMESSAGE, CORRECTMAGIC)
129
 
        except ivle.chat.ProtocolError:
130
 
            raise AssertionError("ProtocolError with 'correct' magic")
131
 
 
132
 
        # This should raise the ProtocolError
133
 
        ivle.chat.decode(CHATMESSAGE, INCORRECTMAGIC)