// Copyright 2019 Google LLC
//
// 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
//
//     https://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.

#include "socket.h"

#include "gmock/gmock.h"
#include "gtest/gtest.h"

#include <chrono>
#include <vector>

TEST(Socket, ConnectTimeout) {
  const char* port = "19021";
  const int timeoutMillis = 200;
  const int maxAttempts = 1024;

  using namespace std::chrono;

  auto server = dap::Socket("localhost", port);

  std::vector<std::shared_ptr<dap::ReaderWriter>> connections;

  for (int i = 0; i < maxAttempts; i++) {
    auto start = system_clock::now();
    auto connection = dap::Socket::connect("localhost", port, timeoutMillis);
    auto end = system_clock::now();

    if (!connection) {
      auto timeTakenMillis = duration_cast<milliseconds>(end - start).count();
      ASSERT_GE(timeTakenMillis + 20,  // +20ms for a bit of timing wiggle room
                timeoutMillis);
      return;
    }

    // Keep hold of the connections to saturate any incoming socket buffers.
    connections.emplace_back(std::move(connection));
  }

  FAIL() << "Failed to test timeout after " << maxAttempts << " attempts";
}