Added support for Pkg.test()

Author: nkottary

README.md

@@ -64,6 +64,14 @@ make sure LD_LIBRARY_PATH contains the MariaDB/MySQL .so file directory path. Us
 * OSX: Same as above. In this case the file will be something like libmysqlclient.dylib.
 * Windows: There is no `@windows_only lib_choices` currently. Please add one and send a pull request.
 
+# Tests
+
+To run the tests you must have MySQL server running on the host. Set the constants HOST and ROOTPASS 
+in test/runtests.jl to the host and root password on your test setup. Run the tests using:
+```
+Pkg.test("MySQL")
+```
+
 # Performance
 
 A total of 67,000 insert queries were executed batch wise in batch sizes of 50, 100, 150 ... so on.

test/runtests.jl

@@ -0,0 +1,10 @@
+using MySQL
+using Base.Test
+
+# Change the below values to whatever it is on your test setup.
+const HOST = "127.0.0.1"
+const ROOTPASS = "root"
+
+for file in ["test_basic.jl", ]
+    include(file)
+end

test/test.jl test/test_basic.jl

@@ -1,20 +1,42 @@
-using MySQL
+# A basic  test that includes creating  a database, a user,  a table and
+# then inserting, updating, selecting ,  etc. And finally cleaning up by
+# dropping table, user and database.
 
-const HOST = "127.0.0.1"
-const USER = "root"
-const PASSWD = "root"
-const DBNAME = "mysqltest"
-
-function run_query_helper(command, successmsg, failmsg)
+function run_query_helper(command, msg)
     response = MySQL.mysql_query(con.ptr, command)
 
     if (response == 0)
-        println(successmsg)
+        println("SUCCESS: " * msg)
+        return true
     else
-        println(failmsg)
+        println("FAILED: " * msg)
+        return false
     end
 end
 
+function connect_as_root()
+    global con = MySQL.mysql_init_and_connect(HOST, "root", ROOTPASS, "")
+end
+
+function create_test_database()
+    command = """CREATE DATABASE mysqltest;"""
+    @test run_query_helper(command, "Create database")
+end
+
+function create_test_user()
+    command = "CREATE USER test@$HOST IDENTIFIED BY 'test';"
+    @test run_query_helper(command, "Create user")
+end
+
+function grant_test_user_privilege()
+    command = "GRANT ALL ON mysqltest.* TO test@$HOST;"
+    @test run_query_helper(command, "Grant privilege")
+end
+
+function connect_as_test_user()
+    global con = MySQL.mysql_init_and_connect(HOST, "test", "test", "mysqltest")
+end
+
 function create_table()
     command = """CREATE TABLE Employee
                  (
@@ -26,12 +48,7 @@ function create_table()
                      LunchTime TIME,
                      PRIMARY KEY (ID)
                  );"""
-    run_query_helper(command, "Create table succeeded", "Create table failed")
-end
-
-function drop_table()
-    command = """DROP TABLE Employee;"""
-    run_query_helper(command, "Drop table succeeded", "Drop table failed")
+    @test run_query_helper(command, "Create table")
 end
 
 function insert_values()
@@ -42,20 +59,25 @@ function insert_values()
                  ('Jim', 30000.00, '2015-6-2', '2015-9-5 10:05:10', '12:30:00'),
                  ('Tim', 15000.50, '2015-7-25', '2015-10-10 12:12:25', '12:30:00');
               """
-    run_query_helper(command, "Insert succeeded", "Insert failed")
+    @test run_query_helper(command, "Insert")
 end
 
 function update_values()
     command = """UPDATE Employee SET Salary = 25000.00 WHERE ID > 2;"""
-    run_query_helper(command, "Update success", "Update failed")
+    @test run_query_helper(command, "Update")
+end
+
+function drop_table()
+    command = """DROP TABLE Employee;"""
+    @test run_query_helper(command, "Drop table")
 end
 
 function do_multi_statement()
     command = """INSERT INTO Employee (Name, Salary, JoinDate, LastLogin, LunchTime)
                  VALUES
                  ('Donald', 30000.00, '2014-2-2', '2015-8-8 13:14:15', '14:01:02');
                  UPDATE Employee SET LunchTime = '15:00:00' WHERE LENGTH(Name) > 5;"""
-    run_query_helper(command, "Multi statement success", "Multi statement failed")
+    @test run_query_helper(command, "Multi statement")
 end
 
 function show_as_dataframe()
@@ -80,22 +102,43 @@ function show_as_dataframe()
     println(dframe)
 end
 
+function drop_test_user()
+    command = """DROP USER test@$HOST;"""
+    @test run_query_helper(command, "Drop user")
+end
+
+function drop_test_database()
+    command = """DROP DATABASE mysqltest;"""
+    @test run_query_helper(command, "Drop database")
+end
+
 function run_test()
-    global con = MySQL.mysql_init_and_connect(HOST, USER, PASSWD, DBNAME)
-    create_table()
 
-    insert_values()
-    show_as_dataframe()
+    # Connect as root and setup database, user and privilege
+    # for the user.
+    connect_as_root()
+    create_test_database()
+    create_test_user()
+    grant_test_user_privilege()
+    MySQL.mysql_disconnect(con)
 
+    # Connect as test user and do insert, update etc.
+    # and finally drop the table.
+    connect_as_test_user()
+    create_table()
+    insert_values()
     update_values()
-    show_as_dataframe()
-
 #   Subsequent queries fail after multi statement, need to debug.
 #    do_multi_statement()
-#    show_as_dataframe()
-
+    show_as_dataframe()
     drop_table()
     MySQL.mysql_disconnect(con)
+
+    # Drop the test user and database.
+    connect_as_root()
+    drop_test_user()
+    drop_test_database()
+    MySQL.mysql_disconnect(con)
 end
 
 run_test()