1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.
"""
Tests for L{twisted.words.protocols.jabber.sasl_mechanisms}.
"""
from twisted.trial import unittest
from twisted.words.protocols.jabber import sasl_mechanisms
class PlainTest(unittest.TestCase):
def test_getInitialResponse(self):
"""
Test the initial response.
"""
m = sasl_mechanisms.Plain(None, 'test', 'secret')
self.assertEqual(m.getInitialResponse(), '\x00test\x00secret')
class AnonymousTest(unittest.TestCase):
"""
Tests for L{twisted.words.protocols.jabber.sasl_mechanisms.Anonymous}.
"""
def test_getInitialResponse(self):
"""
Test the initial response to be empty.
"""
m = sasl_mechanisms.Anonymous()
self.assertEqual(m.getInitialResponse(), None)
class DigestMD5Test(unittest.TestCase):
def setUp(self):
self.mechanism = sasl_mechanisms.DigestMD5('xmpp', 'example.org', None,
'test', 'secret')
def test_getInitialResponse(self):
"""
Test that no initial response is generated.
"""
self.assertIdentical(self.mechanism.getInitialResponse(), None)
def test_getResponse(self):
"""
Partially test challenge response.
Does not actually test the response-value, yet.
"""
challenge = 'realm="localhost",nonce="1234",qop="auth",charset=utf-8,algorithm=md5-sess'
directives = self.mechanism._parse(self.mechanism.getResponse(challenge))
self.assertEqual(directives['username'], 'test')
self.assertEqual(directives['nonce'], '1234')
self.assertEqual(directives['nc'], '00000001')
self.assertEqual(directives['qop'], ['auth'])
self.assertEqual(directives['charset'], 'utf-8')
self.assertEqual(directives['digest-uri'], 'xmpp/example.org')
self.assertEqual(directives['realm'], 'localhost')
def test_getResponseNoRealm(self):
"""
Test that we accept challenges without realm.
The realm should default to the host part of the JID.
"""
challenge = 'nonce="1234",qop="auth",charset=utf-8,algorithm=md5-sess'
directives = self.mechanism._parse(self.mechanism.getResponse(challenge))
self.assertEqual(directives['realm'], 'example.org')
def test__parse(self):
"""
Test challenge decoding.
Specifically, check for multiple values for the C{qop} and C{cipher}
directives.
"""
challenge = 'nonce="1234",qop="auth,auth-conf",charset=utf-8,' \
'algorithm=md5-sess,cipher="des,3des"'
directives = self.mechanism._parse(challenge)
self.assertEqual('1234', directives['nonce'])
self.assertEqual('utf-8', directives['charset'])
self.assertIn('auth', directives['qop'])
self.assertIn('auth-conf', directives['qop'])
self.assertIn('des', directives['cipher'])
self.assertIn('3des', directives['cipher'])
|