chiark / gitweb /
Switch to GPL v3
[disorder] / tests / cookie.py
CommitLineData
abf54aca 1#! /usr/bin/env python
b12be54a
RK
2#
3# This file is part of DisOrder.
4# Copyright (C) 2007 Richard Kettlewell
5#
e7eb3a27 6# This program is free software: you can redistribute it and/or modify
b12be54a 7# it under the terms of the GNU General Public License as published by
e7eb3a27 8# the Free Software Foundation, either version 3 of the License, or
b12be54a
RK
9# (at your option) any later version.
10#
e7eb3a27
RK
11# This program is distributed in the hope that it will be useful,
12# but WITHOUT ANY WARRANTY; without even the implied warranty of
13# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14# GNU General Public License for more details.
15#
b12be54a 16# You should have received a copy of the GNU General Public License
e7eb3a27 17# along with this program. If not, see <http://www.gnu.org/licenses/>.
b12be54a
RK
18#
19import dtest,disorder
20
21def test():
22 """Exercise cookie protocol"""
23 dtest.start_daemon()
f0feb22e 24 dtest.create_user()
b12be54a
RK
25 print " connecting"
26 c = disorder.client()
27 v = c.version()
28 print " getting cookie"
29 k = c.make_cookie()
62ef2216 30 print " cookie value is %s" % k
b12be54a
RK
31 print " connecting with cookie"
32 c = disorder.client()
33 c.connect(k)
34 v = c.version()
35 print " it worked"
36 print " connecting with cookie again"
37 c = disorder.client()
38 c.connect(k)
39 v = c.version()
40 print " it worked"
41 print " revoking cookie"
42 c.revoke()
43 v = c.version()
44 print " connection still works"
45 print " connecting with revoked cookie"
46 c = disorder.client()
47 try:
48 c.connect(k)
49 print "*** should not be able to connect with revoked cookie ***"
50 assert False
51 except disorder.operationError:
52 pass # good
53 print " revoked cookie was rejected"
54
55if __name__ == '__main__':
56 dtest.run()