add more unit test

puritys · Nov 14, 2015 · 0061b3a · 0061b3a
1 parent 45af6ff
commit 0061b3a
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 2 deletions.
diff --git a/src/js/mysql.js b/src/js/mysql.js
@@ -314,3 +314,7 @@ exports.mysql_close = mysql_close;
 exports.mysql_create_db = mysql_create_db;
 exports.mysql_insert_id = mysql_insert_id;
 
+if (typeof(global.UNIT_TEST) != "undefined" && true === global.UNIT_TEST) {
+    exports.mysql_login = mysql_login;
+}
+
diff --git a/tests/mock/socket.js b/tests/mock/socket.js
@@ -1,4 +1,4 @@
-
+var cmd = "";
 if (!global['mockSocket']) {
     global.mockSocket = [];
 }
@@ -34,6 +34,7 @@ exports.fsockopen = function () {
 };
 
 exports.sendcmd = function (msg) {
+    cmd = msg;
     return msg;
 };
 
@@ -44,3 +45,10 @@ exports.recv = function () {
 exports.fread = function () {
     return recv2();
 };
+
+exports.getcmd = function () {
+    return cmd;
+};
+
+
+
diff --git a/tests/mysql/mysql.js b/tests/mysql/mysql.js
@@ -6,7 +6,9 @@ var packetReader = require('./../../src/js/mysql/packetReader.js');
 
 var serverInfo = {
     "session": "0000",
-    "protocol41": true
+    "protocol41": true,
+    "authPluginDataPart1": "xxx",
+    "authPluginDataPart2": "xxx"
 };
 
 function setResponseBuffer(data) {//{{{
@@ -186,3 +188,28 @@ describe('Mysql: method mysql_query', function() {//{{{
 
 });
 
+describe("MySql: method mysql_close", function () {
+    it("Normal", function () {
+        var ret = php.mysql_close({"connectId": 1, "session": "12"});
+        var cmd = socket.getcmd(); 
+        assert.equal(5, cmd.readIntLE(0, 1));
+        assert.equal(12, cmd.readIntLE(4, 1));
+        assert.equal(1, cmd.readIntLE(5, 1));
+    });
+
+});
+
+describe("MySql: method mysql_login", function () {
+    it("Normal", function () {
+        php.mysql_login(serverInfo, "user", "pswd", "db");
+        var cmd = socket.getcmd(); 
+        assert.equal(61, cmd.readIntLE(0, 1));
+        assert.equal(33, cmd.readIntLE(12, 1));
+
+    });
+
+});
+
+
+
+