tests

scivey · Nov 30, 2015 · dca423e · dca423e
1 parent b782081
commit dca423e
Show file tree

Hide file tree

Showing 6 changed files with 418 additions and 68 deletions.
diff --git a/clients/nodejs/client/index.js b/clients/nodejs/client/index.js
@@ -55,13 +55,17 @@ var CLIENT_METHODS = [
     'createDocument',
     'createDocumentWithID',
     'deleteDocument',
+    'multiDeleteDocuments',
     'getDocumentMetadata',
+    'deleteCentroid',
+    'multiDeleteCentroids',
     'createCentroid',
     'multiCreateCentroids',
     'listAllDocumentsForCentroid',
-    'addDocumentToCentroid',
-    'removeDocumentFromCentroid',
+    'addDocumentsToCentroid',
+    'removeDocumentsFromCentroid',
     'joinCentroid',
+    'multiJoinCentroids',
     'listAllCentroids',
     'listAllDocuments',
     'listUnusedDocuments'
@@ -100,7 +104,7 @@ _.each(CLIENT_METHODS, function(methodName) {
 
 (function() {
     var original = RelevancedClient.prototype.createDocumentWithID;
-    RelevancedClient.prototype.createDocumentWithId = function(id, text, lang) {
+    RelevancedClient.prototype.createDocumentWithID = function(id, text, lang) {
         if (!_.isNumber(lang)) {
             lang = genTypes.Language.EN;
         }
@@ -157,13 +161,119 @@ _.each(CLIENT_METHODS, function(methodName) {
         if (!_.isBoolean(ignoreExisting)) {
             ignoreExisting = false;
         }
-        var request = new genTypes.CreateCentroidRequest;
+        var request = new genTypes.MultiCreateCentroidsRequest;
         request.ids = ids;
         request.ignoreExisting = ignoreExisting;
         return original.apply(this, [request]);
     };
 })();
 
+(function() {
+    var original = RelevancedClient.prototype.deleteCentroid;
+    RelevancedClient.prototype.deleteCentroid = function(id, ignoreMissing) {
+        if (!_.isBoolean(ignoreMissing)) {
+            ignoreMissing = false;
+        }
+        var request = new genTypes.DeleteCentroidRequest;
+        request.id = id;
+        request.ignoreMissing = ignoreMissing;
+        return original.apply(this, [request]);
+    };
+})();
+
+(function() {
+    var original = RelevancedClient.prototype.multiDeleteCentroids;
+    RelevancedClient.prototype.multiDeleteCentroids = function(ids, ignoreMissing) {
+        if (!_.isBoolean(ignoreMissing)) {
+            ignoreMissing = false;
+        }
+        var request = new genTypes.MultiDeleteCentroidsRequest;
+        request.ids = ids;
+        request.ignoreMissing = ignoreMissing;
+        return original.apply(this, [request]);
+    };
+})();
+
+(function() {
+    var original = RelevancedClient.prototype.addDocumentsToCentroid;
+    RelevancedClient.prototype.addDocumentsToCentroid = function(centroidId, documentIds, ignoreAlreadyInCentroid) {
+        if (!_.isBoolean(ignoreAlreadyInCentroid)) {
+            ignoreAlreadyInCentroid = false;
+        }
+        var request = new genTypes.AddDocumentsToCentroidRequest;
+        request.centroidId = centroidId;
+        request.documentIds = documentIds;
+        request.ignoreAlreadyInCentroid = ignoreAlreadyInCentroid;
+        return original.apply(this, [request]);
+    };
+})();
+
+(function() {
+    var original = RelevancedClient.prototype.removeDocumentsFromCentroid;
+    RelevancedClient.prototype.removeDocumentsFromCentroid = function(centroidId, documentIds, ignoreNotInCentroid) {
+        if (!_.isBoolean(ignoreNotInCentroid)) {
+            ignoreNotInCentroid = false;
+        }
+        var request = new genTypes.RemoveDocumentsFromCentroidRequest;
+        request.centroidId = centroidId;
+        request.documentIds = documentIds;
+        request.ignoreNotInCentroid = ignoreNotInCentroid;
+        return original.apply(this, [request]);
+    };
+})();
+
+(function() {
+    var original = RelevancedClient.prototype.deleteDocument;
+    RelevancedClient.prototype.deleteDocument = function(documentId, ignoreMissing) {
+        if (!_.isBoolean(ignoreMissing)) {
+            ignoreMissing = false;
+        }
+        var request = new genTypes.DeleteDocumentRequest;
+        request.id = documentId;
+        request.ignoreMissing = ignoreMissing;
+        return original.apply(this, [request]);
+    };
+})();
+
+(function() {
+    var original = RelevancedClient.prototype.multiDeleteDocuments;
+    RelevancedClient.prototype.multiDeleteDocuments = function(documentIds, ignoreMissing) {
+        if (!_.isBoolean(ignoreMissing)) {
+            ignoreMissing = false;
+        }
+        var request = new genTypes.MultiDeleteDocumentsRequest;
+        request.ids = documentIds;
+        request.ignoreMissing = ignoreMissing;
+        return original.apply(this, [request]);
+    };
+})();
+
+(function() {
+    var original = RelevancedClient.prototype.joinCentroid;
+    RelevancedClient.prototype.joinCentroid = function(centroidId, ignoreMissing) {
+        if (!_.isBoolean(ignoreMissing)) {
+            ignoreMissing = false;
+        }
+        var request = new genTypes.JoinCentroidRequest;
+        request.id = centroidId;
+        request.ignoreMissing = ignoreMissing;
+        return original.apply(this, [request]);
+    };
+})();
+
+(function() {
+    var original = RelevancedClient.prototype.multiJoinCentroids;
+    RelevancedClient.prototype.multiJoinCentroids = function(centroidIds, ignoreMissing) {
+        if (!_.isBoolean(ignoreMissing)) {
+            ignoreMissing = false;
+        }
+        var request = new genTypes.MultiJoinCentroidsRequest;
+        request.ids = centroidIds;
+        request.ignoreMissing = ignoreMissing;
+        return original.apply(this, [request]);
+    };
+})();
+
 
 module.exports = {
     RelevancedClient: RelevancedClient,

diff --git a/clients/nodejs/client/package.json b/clients/nodejs/client/package.json
@@ -17,5 +17,9 @@
     "bluebird": "^2.10.1",
     "lodash": "^3.10.1",
     "thrift": "^0.9.2"
+  },
+  "devDependencies": {
+    "chai": "^3.4.1",
+    "mocha": "^2.3.4"
   }
 }