|
1 # copyright 2003-2010 LOGILAB S.A. (Paris, FRANCE), all rights reserved. |
|
2 # contact http://www.logilab.fr/ -- mailto:contact@logilab.fr |
|
3 # |
|
4 # This file is part of CubicWeb. |
|
5 # |
|
6 # CubicWeb is free software: you can redistribute it and/or modify it under the |
|
7 # terms of the GNU Lesser General Public License as published by the Free |
|
8 # Software Foundation, either version 2.1 of the License, or (at your option) |
|
9 # any later version. |
|
10 # |
|
11 # CubicWeb is distributed in the hope that it will be useful, but WITHOUT |
|
12 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS |
|
13 # FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more |
|
14 # details. |
|
15 # |
|
16 # You should have received a copy of the GNU Lesser General Public License |
|
17 # along with CubicWeb. If not, see <http://www.gnu.org/licenses/>. |
|
18 """ |
|
19 |
|
20 """ |
|
21 import os |
|
22 import sys |
|
23 |
|
24 from logilab.common.pytest import PyTester |
|
25 |
|
26 |
|
27 def getlogin(): |
|
28 """avoid usinng os.getlogin() because of strange tty / stdin problems |
|
29 (man 3 getlogin) |
|
30 Another solution would be to use $LOGNAME, $USER or $USERNAME |
|
31 """ |
|
32 if sys.platform == 'win32': |
|
33 return os.environ.get('USERNAME') or 'cubicweb' |
|
34 import pwd |
|
35 return pwd.getpwuid(os.getuid())[0] |
|
36 |
|
37 |
|
38 def update_parser(parser): |
|
39 login = getlogin() |
|
40 parser.add_option('-r', '--rebuild-database', dest='rebuild_db', |
|
41 default=False, action="store_true", |
|
42 help="remove tmpdb and rebuilds the test database") |
|
43 parser.add_option('-u', '--dbuser', dest='dbuser', action='store', |
|
44 default=login, help="database user") |
|
45 parser.add_option('-w', '--dbpassword', dest='dbpassword', action='store', |
|
46 default=login, help="database user's password") |
|
47 parser.add_option('-n', '--dbname', dest='dbname', action='store', |
|
48 default=None, help="database name") |
|
49 parser.add_option('--euser', dest='euser', action='store', |
|
50 default=login, help="euser name") |
|
51 parser.add_option('--epassword', dest='epassword', action='store', |
|
52 default=login, help="euser's password' name") |
|
53 return parser |
|
54 |
|
55 |
|
56 class CustomPyTester(PyTester): |
|
57 def __init__(self, cvg, options): |
|
58 super(CustomPyTester, self).__init__(cvg, options) |
|
59 if options.rebuild_db: |
|
60 os.unlink('tmpdb') |
|
61 os.unlink('tmpdb-template') |