/*
 * Copyright (c) 2021 Cisco and/or its affiliates.
 * 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.
 */

#include <gtest/gtest.h>
#include <gmock/gmock.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <unistd.h>
#include <netinet/in.h>

extern "C" {
#define WITH_TESTS
#include <hicn/core/strategy.h>
#include <hicn/strategies/random.h>
}

#define MAX_TESTS 10

#define NEXTHOP_ID1 NEXTHOP(28)
#define NEXTHOP_ID2 NEXTHOP(29)
#define UNKNOWN_ID1 NEXTHOP(0)
#define UNKNOWN_ID2 NEXTHOP(1)

class StrategyReplicationTest : public ::testing::Test {
 protected:
  StrategyReplicationTest() {
    /* Strategy and strategy entry */
    entry = {
        .type = STRATEGY_TYPE_REPLICATION,
        .options =
            {
                .replication = {},
            },
        .state = {.replication = {}},
    };

    strategy_initialize(&entry, nullptr);

    /* Available nexthops */
    available_nexthops_ = NEXTHOPS_EMPTY;
    EXPECT_EQ(nexthops_get_len(&available_nexthops_), (size_t)0);

    /* Message buffer */
    msgbuf_ = NULL;
    ticks_ = ticks_now();
  }

  virtual ~StrategyReplicationTest() { strategy_finalize(&entry); }

  strategy_entry_t entry;
  nexthops_t available_nexthops_;
  msgbuf_t* msgbuf_;
  Ticks ticks_;
};

TEST_F(StrategyReplicationTest, SingleNexthop) {
  off_t id;

  /* Add a single nexthop */
  id = nexthops_add(&available_nexthops_, NEXTHOP_ID1);
  EXPECT_EQ(nexthops_get_len(&available_nexthops_), (size_t)1);
  EXPECT_EQ(nexthops_get_curlen(&available_nexthops_), (size_t)1);

  strategy_add_nexthop(&entry, &available_nexthops_, id);
  EXPECT_EQ(nexthops_get_len(&available_nexthops_), (size_t)1);
  EXPECT_EQ(nexthops_get_curlen(&available_nexthops_), (size_t)1);

  EXPECT_TRUE(nexthops_contains(&available_nexthops_, NEXTHOP_ID1));
  EXPECT_FALSE(nexthops_contains(&available_nexthops_, UNKNOWN_ID1));
  EXPECT_FALSE(nexthops_contains(&available_nexthops_, UNKNOWN_ID2));

  /* Lookup */
  nexthops_t* nexthops;
  nexthops = strategy_lookup_nexthops(&entry, &available_nexthops_, msgbuf_);

  EXPECT_EQ(nexthops_get_len(nexthops), (size_t)1);
  EXPECT_EQ(nexthops_get_curlen(nexthops), (size_t)1);

  EXPECT_TRUE(nexthops_contains(nexthops, NEXTHOP_ID1));
  EXPECT_FALSE(nexthops_contains(nexthops, UNKNOWN_ID1));
  EXPECT_FALSE(nexthops_contains(nexthops, UNKNOWN_ID2));

  /* Retrieve candidate */

  unsigned nexthop;
  for (unsigned i = 0; i < MAX_TESTS; i++) {
    nexthop = nexthops_get_one(nexthops);
    EXPECT_EQ(nexthop, NEXTHOP_ID1);
  }

  /* Disable (move to nexthop unit tests) */
  nexthops_disable(nexthops, 0);

  EXPECT_EQ(nexthops_get_len(nexthops), (size_t)1);
  EXPECT_EQ(nexthops_get_curlen(nexthops), (size_t)0);

  nexthop = nexthops_get_one(nexthops);
  EXPECT_EQ(nexthop, INVALID_NEXTHOP);
}

TEST_F(StrategyReplicationTest, MultipleNexthops) {
  off_t id;

  id = nexthops_add(&available_nexthops_, NEXTHOP_ID1);
  EXPECT_EQ(nexthops_get_len(&available_nexthops_), (size_t)1);
  EXPECT_EQ(nexthops_get_curlen(&available_nexthops_), (size_t)1);

  strategy_add_nexthop(&entry, &available_nexthops_, id);
  EXPECT_EQ(nexthops_get_len(&available_nexthops_), (size_t)1);
  EXPECT_EQ(nexthops_get_curlen(&available_nexthops_), (size_t)1);

  id = nexthops_add(&available_nexthops_, NEXTHOP_ID2);
  EXPECT_EQ(nexthops_get_len(&available_nexthops_), (size_t)2);
  EXPECT_EQ(nexthops_get_curlen(&available_nexthops_), (size_t)2);

  strategy_add_nexthop(&entry, &available_nexthops_, id);
  EXPECT_EQ(nexthops_get_len(&available_nexthops_), (size_t)2);
  EXPECT_EQ(nexthops_get_curlen(&available_nexthops_), (size_t)2);

  EXPECT_TRUE(nexthops_contains(&available_nexthops_, NEXTHOP_ID1));
  EXPECT_TRUE(nexthops_contains(&available_nexthops_, NEXTHOP_ID2));
  EXPECT_FALSE(nexthops_contains(&available_nexthops_, UNKNOWN_ID1));
  EXPECT_FALSE(nexthops_contains(&available_nexthops_, UNKNOWN_ID2));

  /* Lookup */
  nexthops_t* nexthops;
  nexthops = strategy_lookup_nexthops(&entry, &available_nexthops_, msgbuf_);

  EXPECT_EQ(nexthops_get_len(nexthops), (size_t)2);
  EXPECT_EQ(nexthops_get_curlen(nexthops), (size_t)2);

  EXPECT_TRUE(nexthops_contains(nexthops, NEXTHOP_ID1));
  EXPECT_TRUE(nexthops_contains(nexthops, NEXTHOP_ID2));
  EXPECT_FALSE(nexthops_contains(nexthops, UNKNOWN_ID1));
  EXPECT_FALSE(nexthops_contains(nexthops, UNKNOWN_ID2));

  /* Retrieve candidate */

  unsigned tests = 0;
  nexthops_foreach(nexthops, nexthop, {
    EXPECT_TRUE(nexthop == NEXTHOP_ID1 || nexthop == NEXTHOP_ID2);
    tests++;
  });

  EXPECT_EQ(tests, (unsigned)2);
}