chiark / gitweb /
database upgrade tool. needs to be run manually.
[disorder] / tests / dbversion.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,time,disorder,sys,re,subprocess
22
23 def test():
24     """Database version tests"""
25     # Start up with dbversion 1
26     config = "%s/config" % dtest.testroot
27     configsave = "%s.save" % config
28     dtest.copyfile(config, configsave)
29     open(config, "a").write("dbversion 1\n")
30     dtest.start_daemon()
31     time.sleep(2)
32     dtest.stop_daemon()
33     # Revert to default configuration
34     dtest.copyfile(configsave, config)
35     dtest.start_daemon()
36     time.sleep(2)
37     c = disorder.client()
38     try:
39         v = c.version()
40         print "unexpected success"
41         ok = False
42     except disorder.communicationError, e:
43         if re.search("connection refused", str(e)):
44             print "unexpected error: %s" % e
45             ok = False
46         else:
47             ok = True
48     dtest.stop_daemon()
49     if not ok:
50         sys.exit(1)
51     # Try running the upgrade tool
52     print "Attempting an upgrade..."
53     rc = subprocess.call(["disorder-dbupgrade",
54                           "--config", "%s/config" % dtest.testroot])
55     if rc != 0:
56         print "disorder-dbupgrade: FAILED: exit code %s" % rc
57         sys.exit(1)
58
59 if __name__ == '__main__':
60     dtest.run()