class TestSelectStatementsDict(unittest.TestCase):
def setUp(self):
self.db = WookieDb("localhost", "wookiedbtest", "wookiedbtest", "wookiedbtest", select_type="dict")
self.db.query("""CREATE TABLE `basic_test` (
`id` int(11) NOT NULL AUTO_INCREMENT,
`intVar` int(11) DEFAULT NULL,
`charVar` varchar(45) DEFAULT NULL,
PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=latin1""")
self.data = {"intVar": "123", "charVar": "testing"}
self.db.query("INSERT INTO basic_test VALUES ('', 123, 'testing');")
self.db.query("INSERT INTO basic_test VALUES ('', 123, 'testing');")
self.db.query("INSERT INTO basic_test VALUES ('', 123, 'testing');")
def tearDown(self):
self.db.query("DROP TABLE basic_test")
def testBasicSelect(self):
res = self.db.select("basic_test", "*")
self.assertEqual(len(res), 3)
for r in res:
self.assertEqual(len(r), 3)
self.db.query("INSERT INTO basic_test VALUES ('',123, 'testing');")
res = self.db.select("basic_test", "*")
self.assertEqual(len(res), 4)
for r in res:
self.assertEqual(len(r), 3)
def testConditionSelect(self):
res = self.db.select("basic_test", "*", "where id > 1")
self.assertEqual(len(res), 2)
res = self.db.select("basic_test", "*", "where id = 3")
self.assertEqual(len(res), 1)
def testDictGet(self):
res = self.db.select("basic_test", "*", "where id = 1")
self.assertTrue(isinstance(res[0], dict))
self.assertTrue("id" in res[0])
self.assertTrue("intVar" in res[0])
self.assertTrue("charVar" in res[0])
self.assertFalse("nothere" in res[0])
