From 683f2ff4ef687d73a76c61bd8d0b94155451246f Mon Sep 17 00:00:00 2001
From: Jan Dittberner <jan@dittberner.info>
Date: Mon, 20 Feb 2006 20:42:45 +0000
Subject: [PATCH] transform psycopgtest to a unit test

git-svn-id: file:///home/www/usr01/svn/gnuviechadmin/gnuviech.info/gnuviechadmin/trunk@155 a67ec6bc-e5d5-0310-a910-815c51eb3124
---
 backend/psycopgtest.py | 73 ++++++++++++++++++++++++++----------------
 1 file changed, 46 insertions(+), 27 deletions(-)

diff --git a/backend/psycopgtest.py b/backend/psycopgtest.py
index 7628b8e..d4ffd91 100644
--- a/backend/psycopgtest.py
+++ b/backend/psycopgtest.py
@@ -1,31 +1,50 @@
 #!/usr/bin/env python
-
+import unittest
 import psycopg
 
-cnx = psycopg.connect("host=localhost user=exim4 password=CotOgigmeIk5 dbname=gnuviechadmin")
+class TestDBConnection(unittest.TestCase):
+	def setUp(self):
+		"""
+		set up database connection used in tests
+		"""
+		self.cnx = psycopg.connect("host=localhost user=gnuviech password=SIKKnsyXsV5yU dbname=gnuviechadmin")
+		self.cr = self.cnx.cursor()
+		
+	def testSelectMailAliases(self):
+		"""
+		select all mail aliases
+		"""
+		self.cr.execute('SELECT * FROM mailalias')
+		self.cnx.commit()
+		
+		print self.cr.description
+		
+		result = self.cr.fetchall()
+		for line in result:
+			print line
 
-cr = cnx.cursor()
-
-print "-- Information from mailalias --"
-cr.execute('SELECT * FROM mailalias')
-cnx.commit()
-
-print cr.description
-
-result = cr.fetchall()
-for line in result:
-	print line
-
-print "-- Information from mailpasswd --"
-cr.execute('SELECT * FROM mailpasswd')
-cnx.commit()
-result = cr.fetchall()
-for line in result:
-	print line
-
-print "-- Domains --"
-cr.execute('SELECT DISTINCT domain FROM mailalias')
-cnx.commit()
-result = cr.fetchall()
-for line in result:
-	print line
+	def testSelectMailPasswd(self):
+		"""
+		select all mail passwords
+		"""
+		self.cr.execute('SELECT * FROM mailpasswd')
+		self.cnx.commit()
+		
+		print self.cr.description
+		
+		result = cr.fetchall()
+		for line in result:
+			print line
+			
+	def testSelectDomains(self):
+		"""
+		select all domains
+		"""
+		self.cr.execute('SELECT DISTINCT domain FROM mailalias')
+		self.cnx.commit()
+		
+		print self.cr.description
+		
+		result = cr.fetchall()
+		for line in result:
+			print line
\ No newline at end of file