val/irctest
1
0
Fork 0
mirror of https://github.com/progval/irctest.git synced 2025-04-08 00:09:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add test for authstrings whose size is a multiple of 400.

Browse Source
This commit is contained in:
Valentin Lorentz
2015-12-19 19:03:52 +01:00
parent aa6ab3a77d
commit 9f5cdbc067
1 changed files with 28 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
irctest/clienttests/test_sasl.py
View File

@ -34,7 +34,6 @@ class CapTestCase(cases.BaseClientTestCase, cases.ClientNegociationHelper):
self.assertEqual(m.command, 'AUTHENTICATE', m) self.assertEqual(m.command, 'AUTHENTICATE', m)
self.sendLine('AUTHENTICATE +') self.sendLine('AUTHENTICATE +')
m = self.getMessage() m = self.getMessage()
print(authstring[0:400])
self.assertEqual(m, Message([], None, 'AUTHENTICATE', self.assertEqual(m, Message([], None, 'AUTHENTICATE',
[authstring[0:400]]), m) [authstring[0:400]]), m)
m = self.getMessage() m = self.getMessage()
@ -43,7 +42,34 @@ class CapTestCase(cases.BaseClientTestCase, cases.ClientNegociationHelper):
m = self.getMessage() m = self.getMessage()
self.assertEqual(m, Message([], None, 'AUTHENTICATE', self.assertEqual(m, Message([], None, 'AUTHENTICATE',
[authstring[800:]])) [authstring[800:]]))
self.sendLine('900 * * {} :You are now logged in.'.format('foo'*100)) self.sendLine('900 * * {} :You are now logged in.'.format('foo'))
self.sendLine('903 * :SASL authentication successful')
m = self.negotiateCapabilities(['sasl'], False)
self.assertEqual(m, Message([], None, 'CAP', ['END']))
def testPlainLargeMultiple(self):
# TODO: authzid is optional
auth = authentication.Authentication(
mechanisms=[authentication.Mechanisms.plain],
username='foo',
password='quux'*148,
)
authstring = base64.b64encode(b'\x00'.join(
[b'foo', b'foo', b'quux'*148])).decode()
print(len(authstring))
m = self.negotiateCapabilities(['sasl'], auth=auth)
self.assertEqual(m.command, 'AUTHENTICATE', m)
self.sendLine('AUTHENTICATE +')
m = self.getMessage()
self.assertEqual(m, Message([], None, 'AUTHENTICATE',
[authstring[0:400]]), m)
m = self.getMessage()
self.assertEqual(m, Message([], None, 'AUTHENTICATE',
[authstring[400:800]]))
m = self.getMessage()
self.assertEqual(m, Message([], None, 'AUTHENTICATE',
['+']))
self.sendLine('900 * * {} :You are now logged in.'.format('foo'))
self.sendLine('903 * :SASL authentication successful') self.sendLine('903 * :SASL authentication successful')
m = self.negotiateCapabilities(['sasl'], False) m = self.negotiateCapabilities(['sasl'], False)
self.assertEqual(m, Message([], None, 'CAP', ['END'])) self.assertEqual(m, Message([], None, 'CAP', ['END']))