/*
 * Copyright 2020 gRPC authors.
 *
 * Licensed 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 * as assert from 'assert';
import * as uriParser from '../src/uri-parser';
import * as resolver from '../src/resolver';

describe('URI Parser', function () {
  describe('parseUri', function () {
    const expectationList: {
      target: string;
      result: uriParser.GrpcUri | null;
    }[] = [
      {
        target: 'localhost',
        result: { scheme: undefined, authority: undefined, path: 'localhost' },
      },
      /* This looks weird, but it's OK because the resolver selection code will handle it */
      {
        target: 'localhost:80',
        result: { scheme: 'localhost', authority: undefined, path: '80' },
      },
      {
        target: 'dns:localhost',
        result: { scheme: 'dns', authority: undefined, path: 'localhost' },
      },
      {
        target: 'dns:///localhost',
        result: { scheme: 'dns', authority: '', path: 'localhost' },
      },
      {
        target: 'dns://authority/localhost',
        result: { scheme: 'dns', authority: 'authority', path: 'localhost' },
      },
      {
        target: '//authority/localhost',
        result: {
          scheme: undefined,
          authority: 'authority',
          path: 'localhost',
        },
      },
      // Regression test for https://github.com/grpc/grpc-node/issues/1359
      {
        target:
          'dns:foo-internal.aws-us-east-2.tracing.staging-edge.foo-data.net:443:443',
        result: {
          scheme: 'dns',
          authority: undefined,
          path: 'foo-internal.aws-us-east-2.tracing.staging-edge.foo-data.net:443:443',
        },
      },
    ];
    for (const { target, result } of expectationList) {
      it(target, function () {
        assert.deepStrictEqual(uriParser.parseUri(target), result);
      });
    }
  });

  describe('parseUri + mapUriDefaultScheme', function () {
    const expectationList: {
      target: string;
      result: uriParser.GrpcUri | null;
    }[] = [
      {
        target: 'localhost',
        result: { scheme: 'dns', authority: undefined, path: 'localhost' },
      },
      {
        target: 'localhost:80',
        result: { scheme: 'dns', authority: undefined, path: 'localhost:80' },
      },
      {
        target: 'dns:localhost',
        result: { scheme: 'dns', authority: undefined, path: 'localhost' },
      },
      {
        target: 'dns:///localhost',
        result: { scheme: 'dns', authority: '', path: 'localhost' },
      },
      {
        target: 'dns://authority/localhost',
        result: { scheme: 'dns', authority: 'authority', path: 'localhost' },
      },
      {
        target: 'unix:socket',
        result: { scheme: 'unix', authority: undefined, path: 'socket' },
      },
      {
        target: 'bad:path',
        result: { scheme: 'dns', authority: undefined, path: 'bad:path' },
      },
    ];
    for (const { target, result } of expectationList) {
      it(target, function () {
        assert.deepStrictEqual(
          resolver.mapUriDefaultScheme(
            uriParser.parseUri(target) ?? { path: 'null' }
          ),
          result
        );
      });
    }
  });

  describe('splitHostPort', function () {
    const expectationList: {
      path: string;
      result: uriParser.HostPort | null;
    }[] = [
      { path: 'localhost', result: { host: 'localhost' } },
      { path: 'localhost:123', result: { host: 'localhost', port: 123 } },
      { path: '12345:6789', result: { host: '12345', port: 6789 } },
      { path: '[::1]:123', result: { host: '::1', port: 123 } },
      { path: '[::1]', result: { host: '::1' } },
      { path: '[', result: null },
      { path: '[123]', result: null },
      // Regression test for https://github.com/grpc/grpc-node/issues/1359
      {
        path: 'foo-internal.aws-us-east-2.tracing.staging-edge.foo-data.net:443:443',
        result: {
          host: 'foo-internal.aws-us-east-2.tracing.staging-edge.foo-data.net:443:443',
        },
      },
    ];
    for (const { path, result } of expectationList) {
      it(path, function () {
        assert.deepStrictEqual(uriParser.splitHostPort(path), result);
      });
    }
  });
});