Example #1
0
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")
Example #2
0
 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"}
Example #3
0
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")
Example #4
0
 def setUp(self):
     self.db = WookieDb("localhost", "wookiedbtest", "wookiedbtest", "wookiedbtest")
Example #5
0
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)
Example #6
0
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)