gremlin-js/gremlin-javascript/test/unit/client-test.js

/*
 *  Licensed to the Apache Software Foundation (ASF) under one
 *  or more contributor license agreements.  See the NOTICE file
 *  distributed with this work for additional information
 *  regarding copyright ownership.  The ASF licenses this file
 *  to you under the Apache License, Version 2.0 (the
 *  "License"); you may not use this file except in compliance
 *  with the License.  You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing,
 *  software distributed under the License is distributed on an
 *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 *  KIND, either express or implied.  See the License for the
 *  specific language governing permissions and limitations
 *  under the License.
 */

import assert from 'assert';
import Client from '../../lib/driver/client.js';

describe('Client', function () {
  const query = 'customQuery';

  it('should submit request with default traversalSource', function () {
    const connectionMock = {
      submit: function (requestMessage) {
        assert.strictEqual(requestMessage.getGremlin(), query);
        assert.strictEqual(requestMessage.getG(), 'g');
        assert.strictEqual(requestMessage.getLanguage(), 'gremlin-lang');
        return Promise.resolve();
      }
    };

    const customClient = new Client('http://localhost:9321', {traversalSource: 'g', connectOnStartup: false});
    customClient._connection = connectionMock;
    customClient.submit(query)
  });

  it('should submit request with custom traversalSource', function () {
    const connectionMock = {
      submit: function (requestMessage) {
        assert.strictEqual(requestMessage.getGremlin(), query);
        assert.strictEqual(requestMessage.getG(), 'gCustom');
        return Promise.resolve();
      }
    };

    const customClient = new Client('http://localhost:9321', {traversalSource: 'gCustom', connectOnStartup: false});
    customClient._connection = connectionMock;
    customClient.submit(query)
  });
});