/*
 * Copyright 2023 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 grpc from "@grpc/grpc-js";
import { afterAll as after, afterEach, beforeAll as before, describe, it } from "bun:test";
import assert from "node:assert";

import { TestClient, TestServer } from "./common";

describe("Channel idle timer", () => {
  let server: TestServer;
  let client: TestClient | null = null;
  before(() => {
    server = new TestServer(false);
    return server.start();
  });
  afterEach(() => {
    if (client) {
      client.close();
      client = null;
    }
  });
  after(() => {
    server.shutdown();
  });
  it("Should go idle after the specified time after a request ends", function (done) {
    client = TestClient.createFromServer(server, {
      "grpc.client_idle_timeout_ms": 1000,
    });
    client.sendRequest(error => {
      assert.ifError(error);
      assert.strictEqual(client!.getChannelState(), grpc.connectivityState.READY);
      setTimeout(() => {
        assert.strictEqual(client!.getChannelState(), grpc.connectivityState.IDLE);
        done();
      }, 1100);
    });
  });
  it("Should be able to make a request after going idle", function (done) {
    client = TestClient.createFromServer(server, {
      "grpc.client_idle_timeout_ms": 1000,
    });
    client.sendRequest(error => {
      assert.ifError(error);
      assert.strictEqual(client!.getChannelState(), grpc.connectivityState.READY);
      setTimeout(() => {
        assert.strictEqual(client!.getChannelState(), grpc.connectivityState.IDLE);
        client!.sendRequest(error => {
          assert.ifError(error);
          done();
        });
      }, 1100);
    });
  });
  it("Should go idle after the specified time after waitForReady ends", function (done) {
    client = TestClient.createFromServer(server, {
      "grpc.client_idle_timeout_ms": 1000,
    });
    const deadline = new Date();
    deadline.setSeconds(deadline.getSeconds() + 3);
    client.waitForReady(deadline, error => {
      assert.ifError(error);
      assert.strictEqual(client!.getChannelState(), grpc.connectivityState.READY);
      setTimeout(() => {
        assert.strictEqual(client!.getChannelState(), grpc.connectivityState.IDLE);
        done();
      }, 1100);
    });
  });
  it("Should ensure that the timeout is at least 1 second", function (done) {
    client = TestClient.createFromServer(server, {
      "grpc.client_idle_timeout_ms": 50,
    });
    client.sendRequest(error => {
      assert.ifError(error);
      assert.strictEqual(client!.getChannelState(), grpc.connectivityState.READY);
      setTimeout(() => {
        // Should still be ready after 100ms
        assert.strictEqual(client!.getChannelState(), grpc.connectivityState.READY);
        setTimeout(() => {
          // Should go IDLE after another second
          assert.strictEqual(client!.getChannelState(), grpc.connectivityState.IDLE);
          done();
        }, 1000);
      }, 100);
    });
  });
});

describe("Channel idle timer with UDS", () => {
  let server: TestServer;
  let client: TestClient | null = null;
  before(() => {
    server = new TestServer(false);
    return server.startUds();
  });
  afterEach(() => {
    if (client) {
      client.close();
      client = null;
    }
  });
  after(() => {
    server.shutdown();
  });
  it("Should be able to make a request after going idle", function (done) {
    client = TestClient.createFromServer(server, {
      "grpc.client_idle_timeout_ms": 1000,
    });
    client.sendRequest(error => {
      assert.ifError(error);
      assert.strictEqual(client!.getChannelState(), grpc.connectivityState.READY);
      setTimeout(() => {
        assert.strictEqual(client!.getChannelState(), grpc.connectivityState.IDLE);
        client!.sendRequest(error => {
          assert.ifError(error);
          done();
        });
      }, 1100);
    });
  });
});

describe("Server idle timer", () => {
  let server: TestServer;
  let client: TestClient | null = null;
  before(() => {
    server = new TestServer(false, {
      "grpc.max_connection_idle_ms": 500, // small for testing purposes
    });
    return server.start();
  });
  afterEach(() => {
    if (client) {
      client.close();
      client = null;
    }
  });
  after(() => {
    server.shutdown();
  });

  it("Should go idle after the specified time after a request ends", function (done) {
    client = TestClient.createFromServer(server);
    client.sendRequest(error => {
      assert.ifError(error);
      assert.strictEqual(client!.getChannelState(), grpc.connectivityState.READY);
      client?.waitForClientState(Date.now() + 1500, grpc.connectivityState.IDLE, done);
    });
  });

  it("Should be able to make a request after going idle", function (done) {
    client = TestClient.createFromServer(server);
    client.sendRequest(error => {
      assert.ifError(error);
      assert.strictEqual(client!.getChannelState(), grpc.connectivityState.READY);
      client!.waitForClientState(Date.now() + 1500, grpc.connectivityState.IDLE, err => {
        if (err) return done(err);

        assert.strictEqual(client!.getChannelState(), grpc.connectivityState.IDLE);
        client!.sendRequest(error => {
          assert.ifError(error);
          done();
        });
      });
    });
  });

  it("Should go idle after the specified time after waitForReady ends", function (done) {
    client = TestClient.createFromServer(server);
    const deadline = new Date();
    deadline.setSeconds(deadline.getSeconds() + 3);
    client.waitForReady(deadline, error => {
      assert.ifError(error);
      assert.strictEqual(client!.getChannelState(), grpc.connectivityState.READY);

      client!.waitForClientState(Date.now() + 1500, grpc.connectivityState.IDLE, done);
    });
  });
});