1 files changed, 106 insertions, 0 deletions

diff --git a/test/test_config/test_db.py b/test/test_config/test_db.py
new file mode 100644
index 0000000..ed46eae
--- /dev/null
+++ b/test/test_config/test_db.py
@@ -0,0 +1,106 @@
+# Copyright (C) 2010  Leonard Thomas
+#
+# This file is part of Dodai.
+#
+# Dodai is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Dodai is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Dodai.  If not, see <http://www.gnu.org/licenses/>.
+
+import unittest
+import os
+from dodai.config.db import ConfigDb
+from dodai.config.db import ConfigDbFile
+from dodai.config.db import BaseConfigDb
+from dodai.config.db import NotConfigParserObject
+from dodai.config.db import InvalidProtocolException
+from dodai.config.db import InvalidPortException
+from dodai.config.db import UnknownHandlerException
+from dodai.config.db import UnknownConnectionException
+from dodai.config.db.sa import Sa
+from dodai.config.file import ConfigFile
+from dodai.db import Db
+
+class TestConfigDb(unittest.TestCase):
+
+
+    def setUp(self):
+        self.config_db = ConfigDb()
+        config = ConfigFile()
+        config.set_directory(os.path.dirname(os.path.abspath(__file__)))
+        config.load('config.cfg')
+        self.parser = config.parser()
+
+    def test_setup(self):
+        obj = self.config_db._handlers['sa'][0]
+        self.assertTrue(obj == Sa)
+
+    def test_register_handler(self):
+        self.config_db.register_handler('foo', Exception)
+        self.assertTrue('foo' in self.config_db._handlers.keys())
+
+    def test_add_config_one(self):
+        self.config_db.add_config(config_parser=self.parser)
+        self.assertTrue('test_db' in self.config_db.connections.keys())
+
+    def test_add_config_two(self):
+        self.failUnlessRaises(NotConfigParserObject, self.config_db.add_config,
+                              config_parser='blah')
+
+    def test_load_one(self):
+        self.config_db.add_config(config_parser=self.parser)
+        obj = self.config_db.load('test_db')
+        self.assertTrue(isinstance(obj, Db))
+
+    def test_load_two(self):
+        self.config_db.add_config(config_parser=self.parser)
+        obj = self.config_db.load('test_db')
+        obj = self.config_db.load('test_db')
+        self.assertTrue(isinstance(obj, Db))
+
+    def test_load_handler(self):
+        self.failUnlessRaises(UnknownHandlerException,
+                              self.config_db._load_handler, 'test')
+
+    def test_clean_protocol_one(self):
+        self.config_db.add_config(config_parser=self.parser)
+        obj = self.config_db.load('test_db_two')
+        self.assertTrue(isinstance(obj, Db))
+
+    def test_clean_protocol_one(self):
+        self.config_db.add_config(config_parser=self.parser)
+        obj = self.config_db.load('test_db_two')
+        self.assertTrue(isinstance(obj, Db))
+
+    def test_clean_protocol_two(self):
+        self.config_db.add_config(config_parser=self.parser)
+        self.failUnlessRaises(InvalidProtocolException, self.config_db.load,
+                             'test_db_three')
+
+    def test_clean_port_one(self):
+        obj = BaseConfigDb()
+        data = obj._clean_port('ad')
+        self.assertTrue(data == None)
+
+    def test_clean_port_two(self):
+        obj = BaseConfigDb()
+        data = obj._clean_port(None)
+        self.assertTrue(data == None)
+
+    def test_clean_port_three(self):
+        obj = BaseConfigDb()
+        self.failUnlessRaises(InvalidPortException, obj._clean_port, 66666)
+
+
+    def test_file_load_one(self):
+        self.config_db.add_config(config_parser=self.parser)
+        obj = self.config_db.load('test_db_six')
+        self.assertTrue(isinstance(obj, Db))