skeleton/test/pytestconf.py
changeset 0 b97547f5f1fa
child 1977 606923dff11b
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/skeleton/test/pytestconf.py	Wed Nov 05 15:52:50 2008 +0100
@@ -0,0 +1,37 @@
+import os
+import pwd
+
+from logilab.common.pytest import PyTester
+
+def getlogin():
+    """avoid usinng os.getlogin() because of strange tty / stdin problems
+    (man 3 getlogin)
+    Another solution would be to use $LOGNAME, $USER or $USERNAME
+    """
+    return pwd.getpwuid(os.getuid())[0]
+
+
+def update_parser(parser):
+    login = getlogin()
+    parser.add_option('-r', '--rebuild-database', dest='rebuild_db',
+                      default=False, action="store_true",
+                      help="remove tmpdb and rebuilds the test database")
+    parser.add_option('-u', '--dbuser', dest='dbuser', action='store',
+                      default=login, help="database user")
+    parser.add_option('-w', '--dbpassword', dest='dbpassword', action='store',
+                      default=login, help="database name")
+    parser.add_option('-n', '--dbname', dest='dbname', action='store',
+                      default=None, help="database name")
+    parser.add_option('--euser', dest='euser', action='store',
+                      default=login, help="esuer name")
+    parser.add_option('--epassword', dest='epassword', action='store',
+                      default=login, help="euser's password' name")
+    return parser
+
+
+class CustomPyTester(PyTester):
+    def __init__(self, cvg, options):
+        super(CustomPyTester, self).__init__(cvg, options)
+        if options.rebuild_db:
+            os.unlink('tmpdb')
+            os.unlink('tmpdb-template')