chiark / gitweb /
register/confirm commands, and tests, and docs
[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 " connecting with cookie"
33     c = disorder.client()
34     c.connect(k)
35     v = c.version()
36     print " it worked"
37     print " connecting with cookie again"
38     c = disorder.client()
39     c.connect(k)
40     v = c.version()
41     print " it worked"
42     print " revoking cookie"
43     c.revoke()
44     v = c.version()
45     print " connection still works"
46     print " connecting with revoked cookie"
47     c = disorder.client()
48     try:
49       c.connect(k)
50       print "*** should not be able to connect with revoked cookie ***"
51       assert False
52     except disorder.operationError:
53       pass                              # good
54     print " revoked cookie was rejected"
55
56 if __name__ == '__main__':
57     dtest.run()