test: add unit and integration tests for new discovery methods

Author: dpopp07

test/integration/test.discovery.js

@@ -22,12 +22,14 @@ describe('discovery_integration', function() {
   let configuration_id;
   let collection_id;
   let collection_id2;
+  let japanese_collection_id;
 
   before(function() {
     environment_id = auth.discovery.environment_id;
     configuration_id = auth.discovery.configuration_id;
     collection_id = auth.discovery.collection_id;
     collection_id2 = auth.discovery.collection_id_2;
+    japanese_collection_id = auth.discovery.japanese_collection_id;
 
     nock.enableNetConnect();
     discovery = new DiscoveryV1(
@@ -481,4 +483,55 @@ describe('discovery_integration', function() {
       });
     });
   });
+
+  describe('tokenization dictionary tests @slow', function() {
+    it('should createTokenizationDictionary', function(done) {
+      const params = {
+        environment_id,
+        collection_id: japanese_collection_id,
+        tokenization_rules: [
+          {
+            text: 'すしネコ',
+            tokens: ['すし', 'ネコ'],
+            readings: ['寿司', 'ネコ'],
+            part_of_speech: 'カスタム名詞',
+          },
+        ],
+      };
+
+      discovery.createTokenizationDictionary(params, (err, res) => {
+        assert.ifError(err);
+        assert(res.status);
+        assert(res.type);
+        done();
+      });
+    });
+
+    it('should getTokenizationDictionaryStatus', function(done) {
+      const params = {
+        environment_id,
+        collection_id: japanese_collection_id,
+      };
+
+      discovery.getTokenizationDictionaryStatus(params, (err, res) => {
+        assert.ifError(err);
+        assert(res.status);
+        assert(res.type);
+        done();
+      });
+    });
+
+    it('should deleteTokenizationDictionary', function(done) {
+      const params = {
+        environment_id,
+        collection_id: japanese_collection_id,
+      };
+
+      discovery.deleteTokenizationDictionary(params, (err, res) => {
+        assert.ifError(err);
+        assert.equal(res, '');
+        done();
+      });
+    });
+  });
 });

test/unit/test.discovery.v1.js

@@ -76,6 +76,8 @@ describe('discovery-v1', function() {
     events: '/v1/events',
     metrics: '/v1/metrics',
     logs: '/v1/logs',
+    tokenization_dictionaries:
+      '/v1/environments/env-guid/collections/col-guid/word_lists/tokenization_dictionary',
   };
 
   it('should generate version was not specified (negative test)', function() {
@@ -1008,6 +1010,35 @@ describe('discovery-v1', function() {
             assert.equal(req.method, 'GET');
           });
         });
+
+        describe('tokenization dictionary tests', function() {
+          it('createTokenizationDictionary', function() {
+            const req = discovery.createTokenizationDictionary(queryPayload, noop);
+            assert.equal(
+              req.uri.href,
+              service.url + paths.tokenization_dictionaries + '?version=' + service.version
+            );
+            assert.equal(req.method, 'POST');
+          });
+
+          it('deleteTokenizationDictionary', function() {
+            const req = discovery.deleteTokenizationDictionary(queryPayload, noop);
+            assert.equal(
+              req.uri.href,
+              service.url + paths.tokenization_dictionaries + '?version=' + service.version
+            );
+            assert.equal(req.method, 'DELETE');
+          });
+
+          it('getTokenizationDictionaryStatus', function() {
+            const req = discovery.getTokenizationDictionaryStatus(queryPayload, noop);
+            assert.equal(
+              req.uri.href,
+              service.url + paths.tokenization_dictionaries + '?version=' + service.version
+            );
+            assert.equal(req.method, 'GET');
+          });
+        });
       });
     });
   });