/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/*
 * Copyright (c) 2014-2022,  Regents of the University of California,
 *                           Arizona Board of Regents,
 *                           Colorado State University,
 *                           University Pierre & Marie Curie, Sorbonne University,
 *                           Washington University in St. Louis,
 *                           Beijing Institute of Technology,
 *                           The University of Memphis.
 *
 * This file is part of NFD (Named Data Networking Forwarding Daemon).
 * See AUTHORS.md for complete list of NFD authors and contributors.
 *
 * NFD is free software: you can redistribute it and/or modify it under the terms
 * of the GNU General Public License as published by the Free Software Foundation,
 * either version 3 of the License, or (at your option) any later version.
 *
 * NFD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
 * PURPOSE.  See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * NFD, e.g., in COPYING.md file.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "nfdc/cs-module.hpp"

#include "status-fixture.hpp"
#include "execute-command-fixture.hpp"

namespace nfd::tools::nfdc::tests {

BOOST_AUTO_TEST_SUITE(Nfdc)
BOOST_AUTO_TEST_SUITE(TestCsModule)

BOOST_FIXTURE_TEST_SUITE(ConfigCommand, ExecuteCommandFixture)

BOOST_AUTO_TEST_CASE(Normal)
{
  this->processInterest = [this] (const Interest& interest) {
    ControlParameters req = MOCK_NFD_MGMT_REQUIRE_COMMAND_IS("/localhost/nfd/cs/config");
    BOOST_REQUIRE(req.hasCapacity());
    BOOST_CHECK_EQUAL(req.getCapacity(), 29850);
    BOOST_CHECK(req.hasFlagBit(ndn::nfd::BIT_CS_ENABLE_ADMIT));
    BOOST_CHECK_EQUAL(req.getFlagBit(ndn::nfd::BIT_CS_ENABLE_ADMIT), false);
    BOOST_CHECK(req.hasFlagBit(ndn::nfd::BIT_CS_ENABLE_SERVE));
    BOOST_CHECK_EQUAL(req.getFlagBit(ndn::nfd::BIT_CS_ENABLE_SERVE), true);

    ControlParameters resp(req);
    resp.unsetMask();
    this->succeedCommand(interest, resp);
  };

  this->execute("cs config admit off serve on capacity 29850");
  BOOST_CHECK_EQUAL(exitCode, 0);
  BOOST_CHECK(out.is_equal("cs-config-updated capacity=29850 admit=off serve=on\n"));
  BOOST_CHECK(err.is_empty());
}

BOOST_AUTO_TEST_CASE(NoUpdate)
{
  this->processInterest = [this] (const Interest& interest) {
    ControlParameters req = MOCK_NFD_MGMT_REQUIRE_COMMAND_IS("/localhost/nfd/cs/config");
    BOOST_CHECK(!req.hasCapacity());
    BOOST_CHECK(!req.hasFlagBit(ndn::nfd::BIT_CS_ENABLE_ADMIT));
    BOOST_CHECK(!req.hasFlagBit(ndn::nfd::BIT_CS_ENABLE_SERVE));

    ControlParameters resp;
    resp.setCapacity(573599);
    resp.setFlagBit(ndn::nfd::BIT_CS_ENABLE_ADMIT, true, false);
    resp.setFlagBit(ndn::nfd::BIT_CS_ENABLE_SERVE, false, false);
    this->succeedCommand(interest, resp);
  };

  this->execute("cs config");
  BOOST_CHECK_EQUAL(exitCode, 0);
  BOOST_CHECK(out.is_equal("cs-config-updated capacity=573599 admit=on serve=off\n"));
  BOOST_CHECK(err.is_empty());
}

BOOST_AUTO_TEST_CASE(ErrorCommand)
{
  this->processInterest = nullptr; // no response to command

  this->execute("cs config capacity 19956");
  BOOST_CHECK_EQUAL(exitCode, 1);
  BOOST_CHECK(out.is_empty());
  BOOST_CHECK(err.is_equal("Error 10060 when updating CS config: request timed out\n"));
}

BOOST_AUTO_TEST_SUITE_END() // ConfigCommand

BOOST_FIXTURE_TEST_SUITE(EraseCommand, ExecuteCommandFixture)

BOOST_AUTO_TEST_CASE(WithoutCount)
{
  size_t countRequests = 0;
  this->processInterest = [this, &countRequests] (const Interest& interest) {
    ControlParameters req = MOCK_NFD_MGMT_REQUIRE_COMMAND_IS("/localhost/nfd/cs/erase");
    BOOST_REQUIRE(req.hasName());
    BOOST_CHECK_EQUAL(req.getName(), "/S2NrUoNJcQ");
    BOOST_CHECK(!req.hasCount());

    ControlParameters resp;
    resp.setName("/S2NrUoNJcQ");

    switch (countRequests) {
    case 0:
      resp.setCount(1000);
      resp.setCapacity(1000);
      break;
    case 1:
      resp.setCount(1000);
      resp.setCapacity(1000);
      break;
    case 2:
      resp.setCount(1000);
      resp.setCapacity(1000);
      break;
    case 3:
      resp.setCount(500);
      break;
    default:
      BOOST_FAIL("Exceeded allowed number of erase requests");
      break;
    }
    countRequests++;
    this->succeedCommand(interest, resp);
  };

  this->execute("cs erase /S2NrUoNJcQ");
  BOOST_CHECK_EQUAL(exitCode, 0);
  BOOST_CHECK_EQUAL(countRequests, 4);
  BOOST_CHECK(out.is_equal("cs-erased prefix=/S2NrUoNJcQ count=3500\n"));
  BOOST_CHECK(err.is_empty());
}

BOOST_AUTO_TEST_CASE(WithCount)
{
  size_t countRequests = 0;
  this->processInterest = [this, &countRequests] (const Interest& interest) {
    ControlParameters req = MOCK_NFD_MGMT_REQUIRE_COMMAND_IS("/localhost/nfd/cs/erase");
    BOOST_REQUIRE(req.hasName());
    BOOST_CHECK_EQUAL(req.getName(), "/gr7ADmIq");
    BOOST_REQUIRE(req.hasCount());

    ControlParameters resp;
    resp.setName("/gr7ADmIq");

    switch (countRequests) {
    case 0:
      BOOST_CHECK_EQUAL(req.getCount(), 3568);
      resp.setCount(1000);
      resp.setCapacity(1000);
      break;
    case 1:
      BOOST_CHECK_EQUAL(req.getCount(), 2568);
      resp.setCount(1000);
      resp.setCapacity(1000);
      break;
    case 2:
      BOOST_CHECK_EQUAL(req.getCount(), 1568);
      resp.setCount(1000);
      resp.setCapacity(1000);
      break;
    case 3:
      BOOST_CHECK_EQUAL(req.getCount(), 568);
      resp.setCount(568);
      break;
    default:
      BOOST_FAIL("Exceeded allowed number of erase requests");
      break;
    }

    countRequests++;
    this->succeedCommand(interest, resp);
  };

  this->execute("cs erase /gr7ADmIq count 3568");
  BOOST_CHECK_EQUAL(exitCode, 0);
  BOOST_CHECK_EQUAL(countRequests, 4);
  BOOST_CHECK(out.is_equal("cs-erased prefix=/gr7ADmIq count=3568\n"));
  BOOST_CHECK(err.is_empty());
}

BOOST_AUTO_TEST_CASE(WithCountError)
{
  size_t countRequests = 0;
  this->processInterest = [this, &countRequests] (const Interest& interest) {
    ControlParameters req = MOCK_NFD_MGMT_REQUIRE_COMMAND_IS("/localhost/nfd/cs/erase");
    BOOST_REQUIRE(req.hasName());
    BOOST_CHECK_EQUAL(req.getName(), "/gr7ADmIq");
    BOOST_REQUIRE(req.hasCount());

    ControlParameters resp;
    resp.setName("/gr7ADmIq");

    switch (countRequests) {
    case 0:
      BOOST_CHECK_EQUAL(req.getCount(), 3568);
      resp.setCount(1000);
      resp.setCapacity(1000);
      this->succeedCommand(interest, resp);
      break;
    case 1:
      BOOST_CHECK_EQUAL(req.getCount(), 2568);
      resp.setCount(1000);
      resp.setCapacity(1000);
      this->failCommand(interest, 500, "internal error");
      break;
    default:
      BOOST_FAIL("Exceeded allowed number of erase requests");
      break;
    }

    countRequests++;
  };

  this->execute("cs erase /gr7ADmIq count 3568");
  BOOST_CHECK_EQUAL(exitCode, 1);
  BOOST_CHECK_EQUAL(countRequests, 2);
  BOOST_CHECK(out.is_empty());
  BOOST_CHECK(err.is_equal("Error 500 when erasing cached Data: internal error\n"));
}

BOOST_AUTO_TEST_CASE(Timeout)
{
  this->processInterest = nullptr; // no response to command

  this->execute("cs erase /8Rq1Merv count 16519");
  BOOST_CHECK_EQUAL(exitCode, 1);
  BOOST_CHECK(out.is_empty());
  BOOST_CHECK(err.is_equal("Error 10060 when erasing cached Data: request timed out\n"));
}

BOOST_AUTO_TEST_SUITE_END() // EraseCommand

const std::string STATUS_XML = stripXmlSpaces(R"XML(
  <cs>
    <capacity>31807</capacity>
    <admitEnabled/>
    <nEntries>16131</nEntries>
    <nHits>14363</nHits>
    <nMisses>27462</nMisses>
  </cs>
)XML");

const std::string STATUS_TEXT = std::string(R"TEXT(
CS information:
  capacity=31807
     admit=on
     serve=off
  nEntries=16131
     nHits=14363
   nMisses=27462
)TEXT").substr(1);

BOOST_FIXTURE_TEST_CASE(Status, StatusFixture<CsModule>)
{
  this->fetchStatus();
  CsInfo payload;
  payload.setCapacity(31807)
         .setEnableAdmit(true)
         .setEnableServe(false)
         .setNEntries(16131)
         .setNHits(14363)
         .setNMisses(27462);
  this->sendDataset("/localhost/nfd/cs/info", payload);
  this->prepareStatusOutput();

  BOOST_CHECK(statusXml.is_equal(STATUS_XML));
  BOOST_CHECK(statusText.is_equal(STATUS_TEXT));
}

BOOST_AUTO_TEST_SUITE_END() // TestCsModule
BOOST_AUTO_TEST_SUITE_END() // Nfdc

} // namespace nfd::tools::nfdc::tests