def test_eq(self):
self.assertEqual(ODBCConfig(), ODBCConfig())
configA = ODBCConfig(data=self.simple_data)
configB = ODBCConfig()
configB.sources = {'testdb': self.ACCESS_CONFIGSTR}
configB.source_specs = {'testdb': {'Database': 'testdb.mdb'}}
self.assertEqual(configA, configB)
def test_init_simple_data(self):
config = ODBCConfig(data=self.simple_data)
self.assertEqual({'testdb': self.ACCESS_CONFIGSTR}, config.sources)
self.assertEqual({'testdb': {
'Database': "testdb.mdb"
}}, config.source_specs)
self.assertEqual({}, config.odbc_info)
def test_del_source_dependent(self):
config = ODBCConfig()
config.add_source("testdb", self.ACCESS_CONFIGSTR)
config.add_source_spec("testdb", {'Database': "testdb.mdb"})
config.del_source("testdb")
self.assertEqual({}, config.sources)
self.assertEqual({}, config.source_specs)
def test_add_source_spec(self):
config = ODBCConfig()
config.add_source("testdb", self.ACCESS_CONFIGSTR)
config.add_source_spec("testdb", {'Database': "testdb.mdb"})
self.assertEqual({'testdb': {
'Database': "testdb.mdb"
}}, config.source_specs)
def test_baselines(self):
filenames = ['simple_odbc', 'diet']
basePath = os.path.split(os.path.abspath(__file__))[0]
for fn in filenames:
iniPath = os.path.join(basePath, 'baselines', '{0}.ini'.format(fn))
outPath = os.path.join(basePath, 'baselines', '{0}.out'.format(fn))
config = ODBCConfig(filename=iniPath)
config.write(outPath)
written = ODBCConfig(filename=outPath)
self.assertEqual(config, written)
try:
os.remove(outPath)
except:
pass
def test_init_complex_data(self):
config = ODBCConfig(data=self.complex_data)
self.assertEqual(
{
'test1': self.ACCESS_CONFIGSTR,
'test2': self.EXCEL_CONFIGSTR
}, config.sources)
self.assertEqual(
{
'test1': {
'Database': "test1.db",
'LogonID': "Admin",
'pwd': "secret_pass"
},
'test2': {
'Database': "test2.xls"
}
}, config.source_specs)
self.assertEqual({'UNICODE': "UTF-8"}, config.odbc_info)
def test_add_source(self):
config = ODBCConfig()
config.add_source("testdb", self.ACCESS_CONFIGSTR)
self.assertEqual({'testdb': self.ACCESS_CONFIGSTR}, config.sources)
self.assertEqual({}, config.source_specs)
self.assertEqual({}, config.odbc_info)
def test_init_empty_data(self):
config = ODBCConfig()
self.assertEqual({}, config.sources)
self.assertEqual({}, config.source_specs)
self.assertEqual({}, config.odbc_info)
def test_create(self):
config = ODBCConfig()
self.assertIsNone(config.file)
def test_odbc_repr(self):
config = ODBCConfig(data=self.simple_data)
self.assertMultiLineEqual(config.odbc_repr(), self.simple_data)
def test_set_odbc_info(self):
config = ODBCConfig()
config.set_odbc_info("UNICODE", "UTF-8")
self.assertEqual({'UNICODE': "UTF-8"}, config.odbc_info)
def test_add_spec_bad(self):
config = ODBCConfig()
with self.assertRaises(ODBCError):
config.add_source_spec("testdb", {'Database': "testdb.mdb"})
def test_add_source_reserved(self):
config = ODBCConfig()
with self.assertRaises(ODBCError):
config.add_source("ODBC Data Sources", self.ACCESS_CONFIGSTR)
with self.assertRaises(ODBCError):
config.add_source("ODBC", self.ACCESS_CONFIGSTR)
def test_del_source(self):
config = ODBCConfig(data=self.simple_data)
config.del_source('testdb')
self.assertEqual({}, config.sources)