class TestUpdateStatements(unittest.TestCase):
def setUp(self):
self.db = WookieDb("localhost", "wookiedbtest", "wookiedbtest", "wookiedbtest")
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": "345", "charVar": "testing2"}
self.db.query("INSERT INTO basic_test VALUES ('', 123, 'testing');")
def tearDown(self):
self.db.query("DROP TABLE basic_test")
def testOneUpdate(self):
res = self.db.query("select * from basic_test WHERE ID = 1")
self.assertEqual(len(res), 1)
self.assertEqual(res[0][1], 123)
self.assertEqual(res[0][2], "testing")
self.db.update("basic_test", self.data, "WHERE ID = 1")
res = self.db.query("select * from basic_test")
self.assertEqual(len(res), 1)
self.assertEqual(res[0][1], 345)
self.assertEqual(res[0][2], "testing2")
def setUp(self):
self.db = WookieDb("localhost", "wookiedbtest", "wookiedbtest", "wookiedbtest")
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"}
class TestBasicCommands(unittest.TestCase):
def setUp(self):
self.db = WookieDb("localhost", "wookiedbtest", "wookiedbtest", "wookiedbtest")
def testWorking(self):
"""checks if the object is creatable and localhost accessible"""
assert True
def testQuery(self):
"""checks if running a query works"""
res = self.db.query("select 1")
self.assertEqual(res[0][0], 1)
def testCreateTable(self):
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"""
)
tables = self.db.query("show tables")
self.assertTrue(("basic_test",) in tables)
self.db.query("DROP TABLE basic_test")
def setUp(self):
self.db = WookieDb("localhost", "wookiedbtest", "wookiedbtest", "wookiedbtest")
class TestInsertStatements(unittest.TestCase):
def setUp(self):
self.db = WookieDb("localhost", "wookiedbtest", "wookiedbtest", "wookiedbtest")
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"}
def tearDown(self):
self.db.query("DROP TABLE basic_test")
def testOneInsert(self):
res = self.db.query("select * from basic_test")
self.assertEqual(len(res), 0)
self.db.insert("basic_test", self.data)
res = self.db.query("select * from basic_test")
self.assertEqual(len(res), 1)
def testManyInsert(self):
for i in xrange(100):
res = self.db.query("select * from basic_test")
self.assertEqual(len(res), i)
self.db.insert("basic_test", self.data)
class TestSelectStatements(unittest.TestCase):
def setUp(self):
self.db = WookieDb("localhost", "wookiedbtest", "wookiedbtest", "wookiedbtest")
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)
