Add functionality to retrieve corpora for a give account_id #145

Author: germanattanasio

services/concept_insights/v2.js

@@ -205,9 +205,13 @@ ConceptInsightsGraphs.prototype.getRelationScores = function(params, callback) {
  * Retrieves the available corpora
  */
 ConceptInsightsCorpora.prototype.listCorpora = function(params, callback) {
+  var path = '/v2/corpora';
+  if (params && params.account_id)
+    path += '/' + params.account_id;
+
   var parameters = {
     options: {
-      url: '/v2/corpora',
+      url: path,
       method: 'GET',
       json: true
     },

test/test.integration-all-services.js

@@ -135,15 +135,134 @@ describe('integration-all-services', function() {
 
   describe('functional_concept_insights', function() {
     this.timeout(TEN_SECONDS);
+
+    var sample = {
+      concept: '/graphs/wikipedia/en-20120601/concepts/IBM',
+      document: '/corpora/public/ibmresearcher/documents/il-AHARONA',
+      corpus : '/corpora/public/ibmresearcher',
+      graph: '/graphs/wikipedia/en-20120601'
+    };
+
     var concept_insights = watson.concept_insights(auth.concept_insights);
 
     it('listCorpora()', function(done) {
       concept_insights.corpora.listCorpora({}, failIfError.bind(failIfError, done));
     });
 
+    it('listCorporaWithAccountId()', function(done) {
+      concept_insights.corpora.listCorpora({
+        account_id: auth.concept_insights.account_id
+      }, failIfError.bind(failIfError, done));
+    });
+
     it('getAccountsInfo()', function(done) {
       concept_insights.accounts.getAccountsInfo({}, failIfError.bind(failIfError, done));
     });
+
+    // to update
+    it('getConcept()', function(done) {
+      concept_insights.graphs.getConcept({
+        id: sample.concept
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('searchConceptByLabel()', function(done) {
+      concept_insights.graphs.searchConceptByLabel({
+        graph: sample.graph,
+        query: 'ibm'
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('getRelatedConcepts()', function(done) {
+      concept_insights.graphs.getRelatedConcepts({
+        graph: sample.graph,
+        concepts: [sample.concept]
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('annotateText()', function(done) {
+      concept_insights.graphs.annotateText({
+        graph: sample.graph,
+        text: 'Nizar Magboul Alseddeg is currently living in Austin Texas'
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('getRelationScores()', function(done) {
+      concept_insights.graphs.getRelationScores({
+        id: sample.concept,
+        concepts: ['/graphs/wikipedia/en-20120601/concepts/Cloud_computing',
+        '/graphs/wikipedia/en-20120601/concepts/Web_services']
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('getCorpus()', function(done) {
+      concept_insights.corpora.getCorpus({
+        corpus: sample.corpus
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('listDocuments()', function(done) {
+      concept_insights.corpora.listDocuments({
+        corpus: sample.corpus
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('getDocument()', function(done) {
+      concept_insights.corpora.getDocument({
+        id: sample.document
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('getDocumentAnnotations()', function(done) {
+      concept_insights.corpora.getDocumentAnnotations({
+        id: sample.document
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('getCorpusProcessingState()', function(done) {
+      concept_insights.corpora.getCorpusProcessingState({
+        corpus: sample.corpus
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('getCorpusStats()', function(done) {
+      concept_insights.corpora.getCorpusStats({
+        corpus: sample.corpus
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('searchByLabel()', function(done) {
+      concept_insights.corpora.searchByLabel({
+        corpus: sample.corpus,
+        query: 'ibm'
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('getRelatedDocuments()', function(done) {
+      concept_insights.corpora.getRelatedDocuments({
+        corpus: sample.corpus,
+        ids: [sample.concept]
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('getRelatedConcepts()', function(done) {
+      concept_insights.corpora.getRelatedConcepts({
+        id: sample.concept
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('getRelatedConcepts()', function(done) {
+      concept_insights.corpora.getRelatedConcepts({
+        concepts: [sample.concept],
+        id: sample.concept
+      }, failIfError.bind(failIfError, done));
+    });
+
+    it('getDocumentProcessingState()', function(done) {
+      concept_insights.corpora.getDocumentProcessingState({
+        id: sample.document
+      }, failIfError.bind(failIfError, done));
+    });
   });
 
   describe('functional_text_to_speech', function() {