← 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: 2009-12-08 03:50:24 UTC
  • mfrom: (1294.2.143 ui-the-third)
  • Revision ID: grantw@unimelb.edu.au-20091208035024-wjx8zp54gth15ph8
Merge ui-the-third. This is another UI revamp.

The header is now thin! Thin! The yellow bar is gone. The tabs are gone.
Breadcrumbs are here. Routes is replaced (with an object publishing
mechanism). Views are less repetitive. etc.

Show diffs side-by-side

added added

removed removed

Lines of Context:
ivle/tests/test_chat.py
1
 
# IVLE - Informatics Virtual Learning Environment
2
 
# Copyright (C) 2007-2010 The University of Melbourne
3
 
#
4
 
# This program is free software; you can redistribute it and/or modify
5
 
# it under the terms of the GNU General Public License as published by
6
 
# the Free Software Foundation; either version 2 of the License, or
7
 
# (at your option) any later version.
8
 
#
9
 
# This program is distributed in the hope that it will be useful,
10
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 
# GNU General Public License for more details.
13
 
#
14
 
# You should have received a copy of the GNU General Public License
15
 
# along with this program; if not, write to the Free Software
16
 
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17
 
 
18
 
import socket
19
 
import os
20
 
import random
21
 
 
22
 
import cjson
23
 
from nose.tools import assert_equal, raises
24
 
 
25
 
import ivle.chat
26
 
 
27
 
 
28
 
TIMEOUT = 0.1
29
 
NULLNETSTRING = "0:,"
30
 
SIMPLESTRING = "Hello world!"
31
 
SIMPLENETSTRING = "12:Hello world!,"
32
 
 
33
 
 
34
 
class TestChat(object):
35
 
    def setUp(self):
36
 
        """Creates a socket pair for testing"""
37
 
        self.s1, self.s2 = socket.socketpair()
38
 
        self.s1.settimeout(TIMEOUT)
39
 
        self.s2.settimeout(TIMEOUT)
40
 
 
41
 
    def tearDown(self):
42
 
        """Closes the socket pair"""
43
 
        self.s1.close()
44
 
        self.s2.close()
45
 
 
46
 
    def test_send_null_netstring(self):
47
 
        """Check that we construct a empty Netstring correctly"""
48
 
        ivle.chat.send_netstring(self.s1, "")
49
 
        assert_equal(self.s2.recv(1024), NULLNETSTRING)
50
 
 
51
 
    def test_send_simple_netstring(self):
52
 
        """Check that we construct a simple Netstring correctly"""
53
 
        ivle.chat.send_netstring(self.s1, SIMPLESTRING)
54
 
        assert_equal(self.s2.recv(1024), SIMPLENETSTRING)
55
 
 
56
 
    def test_recv_null_netstring(self):
57
 
        """Check that we can decode a null Netstring"""
58
 
        self.s1.sendall(NULLNETSTRING)
59
 
        assert_equal(ivle.chat.recv_netstring(self.s2), "")
60
 
 
61
 
    def test_recv_null_netstring(self):
62
 
        """Check that we can decode a simple Netstring"""
63
 
        self.s1.sendall(SIMPLENETSTRING)
64
 
        assert_equal(ivle.chat.recv_netstring(self.s2), SIMPLESTRING)
65
 
 
66
 
    @raises(socket.timeout)
67
 
    def test_invalid_short_netstring(self):
68
 
        self.s1.sendall("1234:not that long!,")
69
 
        assert ivle.chat.recv_netstring(self.s2) is None
70
 
 
71
 
    @raises(ivle.chat.ProtocolError)
72
 
    def test_invalid_long_netstring(self):
73
 
        self.s1.sendall("5:not that short!,")
74
 
        assert ivle.chat.recv_netstring(self.s2) is None
75
 
 
76
 
    def test_long_netstring(self):
77
 
        # XXX: send() may block if this is too big
78
 
        msg = os.urandom(50000)
79
 
        ivle.chat.send_netstring(self.s1, msg)
80
 
        assert ivle.chat.recv_netstring(self.s2) == msg
81
 
 
82
 
    def test_multiple_netstrings(self):
83
 
        messages = []
84
 
        for i in range(10):
85
 
            message = os.urandom(random.randint(0,20))
86
 
            messages.append(message)
87
 
            ivle.chat.send_netstring(self.s1, message)
88
 
        for i in range(10):
89
 
            assert_equal(ivle.chat.recv_netstring(self.s2), messages[i])
90
 
 
91
 
    def test_encode(self):
92
 
        """Check that we correctly encode a basic object
93
 
        """
94
 
        MESSAGE = {}
95
 
        MAGIC = "3EE"
96
 
        content = cjson.encode(MESSAGE)
97
 
        # Digest can be formed with `echo -n "${content}${MAGIC}" | md5sum`
98
 
        DIGEST = '2b59b68e1ac0852b87fb7e64946f2658'
99
 
        expected = {'digest': DIGEST,
100
 
                'content': content}
101
 
        encoded = ivle.chat.encode(MESSAGE, MAGIC)
102
 
        assert_equal(cjson.decode(encoded), expected)
103
 
 
104
 
    def test_encode_decode(self):
105
 
        """Check that a round trip encoding and decoding works
106
 
        """
107
 
        MESSAGE = {'message': 'Hello, world'}
108
 
        MAGIC = "MagicString"
109
 
        encoded = ivle.chat.encode(MESSAGE, MAGIC)
110
 
        decoded = ivle.chat.decode(encoded, MAGIC)
111
 
        assert_equal(decoded, MESSAGE)
112
 
 
113
 
    @raises(ivle.chat.ProtocolError)
114
 
    def test_decode_bad_magic(self):
115
 
        """Check that a bad digest causes a ProtocolError to be raised
116
 
        """
117
 
        CHATMESSAGE = ('{"content": "{\\"a\\": \\"b\\"}", "digest": ' +
118
 
                '"eb860a5fe8fdbef19ffb79e3a5c47113"}')
119
 
        CORRECTMAGIC = "AEIOU"
120
 
        INCORRECTMAGIC = "ABCDE"
121
 
 
122
 
        # Check our "correct" string decodes without a ProtocolError
123
 
        try:
124
 
            ivle.chat.decode(CHATMESSAGE, CORRECTMAGIC)
125
 
        except ivle.chat.ProtocolError:
126
 
            raise AssertionError("ProtocolError with 'correct' magic")
127
 
 
128
 
        # This should raise the ProtocolError
129
 
        ivle.chat.decode(CHATMESSAGE, INCORRECTMAGIC)