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