Más tests para #23

codenautas · Nov 7, 2016 · 289bb91 · 289bb91
1 parent 7576e03
commit 289bb91
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/lib/txt-to-sql.js b/lib/txt-to-sql.js
@@ -667,10 +667,12 @@ function finalizeStats(info) {
     info.stats.columns = info.columnsInfo.length;
     info.stats.textColumns = 0;
     info.stats.nullColumns = 0;
-    info.columnsInfo.forEach(function(column) {
-        //console.log("column", column)
+    info.stats.primaryKey = [];
+    var names = info.headers.split(info.opts.separator);
+    info.columnsInfo.forEach(function(column, index) {
         if(column.typeInfo.isTextColumn) { ++info.stats.textColumns; }
         if(column.hasNullValues) { ++info.stats.nullColumns; }
+        if(column.inPrimaryKey) { info.stats.primaryKey.push(names[index]); }
     });
     return info;
 }

diff --git a/test/fixtures/pk-custom.result.yaml b/test/fixtures/pk-custom.result.yaml
@@ -33,3 +33,4 @@ stats:
   columns: 4
   textColumns: 4
   nullColumns: 0
+  primaryKey: [X,Y]
diff --git a/web/txt-to-sql.js b/web/txt-to-sql.js
@@ -667,10 +667,12 @@ function finalizeStats(info) {
     info.stats.columns = info.columnsInfo.length;
     info.stats.textColumns = 0;
     info.stats.nullColumns = 0;
-    info.columnsInfo.forEach(function(column) {
-        //console.log("column", column)
+    info.stats.primaryKey = [];
+    var names = info.headers.split(info.opts.separator);
+    info.columnsInfo.forEach(function(column, index) {
         if(column.typeInfo.isTextColumn) { ++info.stats.textColumns; }
         if(column.hasNullValues) { ++info.stats.nullColumns; }
+        if(column.inPrimaryKey) { info.stats.primaryKey.push(names[index]); }
     });
     return info;
 }