chiark / gitweb /
tidy up upgrade/recovery flags a bit
[disorder] / tests / dbversion.py
... / ...
CommitLineData
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#
21import dtest,time,disorder,sys,re,subprocess
22
23def 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 print "Checking daemon won't start with an old database"
35 dtest.copyfile(configsave, config)
36 dtest.start_daemon()
37 time.sleep(2)
38 c = disorder.client()
39 try:
40 v = c.version()
41 print "unexpected success"
42 ok = False
43 except disorder.communicationError, e:
44 if re.search("connection refused", str(e)):
45 print "unexpected error: %s" % e
46 ok = False
47 else:
48 ok = True
49 dtest.stop_daemon()
50 if not ok:
51 sys.exit(1)
52 # Try running the upgrade tool
53 print "Attempting an upgrade..."
54 rc = subprocess.call(["disorder-dbupgrade",
55 "--config", "%s/config" % dtest.testroot])
56 if rc != 0:
57 print "disorder-dbupgrade: FAILED: exit code %s" % rc
58 sys.exit(1)
59 print "Testing daemon after upgrade..."
60 dtest.start_daemon()
61 time.sleep(2)
62 assert dtest.check_files() == 0
63
64if __name__ == '__main__':
65 dtest.run()