/**
 *    Copyright 2014 MongoDB Inc.
 *
 *    This program is free software: you can redistribute it and/or  modify
 *    it under the terms of the GNU Affero General Public License, version 3,
 *    as published by the Free Software Foundation.
 *
 *    This program 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 Affero General Public License for more details.
 *
 *    You should have received a copy of the GNU Affero General Public License
 *    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 *    As a special exception, the copyright holders give permission to link the
 *    code of portions of this program with the OpenSSL library under certain
 *    conditions as described in each individual source file and distribute
 *    linked combinations including the program with the OpenSSL library. You
 *    must comply with the GNU Affero General Public License in all respects for
 *    all of the code used other than as permitted herein. If you modify file(s)
 *    with this exception, you may extend this exception to your version of the
 *    file(s), but you are not obligated to do so. If you do not wish to do so,
 *    delete this exception statement from your version. If you delete this
 *    exception statement from all source files in the program, then also delete
 *    it in the license file.
 */

#include <algorithm>
#include <vector>

#include "mongo/db/jsobj.h"
#include "mongo/db/repl/member_config.h"
#include "mongo/unittest/unittest.h"
#include "mongo/util/time_support.h"

namespace mongo {
namespace repl {
namespace {

TEST(MemberConfig, ParseMinimalMemberConfigAndCheckDefaults) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "localhost:12345"),
                            &tagConfig));
    ASSERT_EQUALS(0, mc.getId());
    ASSERT_EQUALS(HostAndPort("localhost", 12345), mc.getHostAndPort());
    ASSERT_EQUALS(1.0, mc.getPriority());
    ASSERT_EQUALS(Seconds(0), mc.getSlaveDelay());
    ASSERT_TRUE(mc.isVoter());
    ASSERT_FALSE(mc.isHidden());
    ASSERT_FALSE(mc.isArbiter());
    ASSERT_TRUE(mc.shouldBuildIndexes());
    ASSERT_EQUALS(3U, mc.getNumTags());
    ASSERT_OK(mc.validate());
}

TEST(MemberConfig, ParseFailsWithIllegalFieldName) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_EQUALS(ErrorCodes::BadValue,
                  mc.initialize(BSON("_id" << 0 << "host"
                                           << "localhost"
                                           << "frim"
                                           << 1),
                                &tagConfig));
}

TEST(MemberConfig, ParseFailsWithMissingIdField) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_EQUALS(ErrorCodes::NoSuchKey,
                  mc.initialize(BSON("host"
                                     << "localhost:12345"),
                                &tagConfig));
}

TEST(MemberConfig, ParseFailsWithBadIdField) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_EQUALS(ErrorCodes::NoSuchKey,
                  mc.initialize(BSON("host"
                                     << "localhost:12345"),
                                &tagConfig));
    ASSERT_EQUALS(ErrorCodes::TypeMismatch,
                  mc.initialize(BSON("_id"
                                     << "0"
                                     << "host"
                                     << "localhost:12345"),
                                &tagConfig));
    ASSERT_EQUALS(ErrorCodes::TypeMismatch,
                  mc.initialize(BSON("_id" << Date_t() << "host"
                                           << "localhost:12345"),
                                &tagConfig));
}

TEST(MemberConfig, ParseFailsWithMissingHostField) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_EQUALS(ErrorCodes::NoSuchKey, mc.initialize(BSON("_id" << 0), &tagConfig));
}


TEST(MemberConfig, ParseFailsWithBadHostField) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_EQUALS(ErrorCodes::TypeMismatch,
                  mc.initialize(BSON("_id" << 0 << "host" << 0), &tagConfig));
    ASSERT_EQUALS(ErrorCodes::FailedToParse,
                  mc.initialize(BSON("_id" << 0 << "host"
                                           << ""),
                                &tagConfig));
    ASSERT_EQUALS(ErrorCodes::FailedToParse,
                  mc.initialize(BSON("_id" << 0 << "host"
                                           << "myhost:zabc"),
                                &tagConfig));
}

TEST(MemberConfig, ParseArbiterOnly) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "arbiterOnly"
                                       << 1.0),
                            &tagConfig));
    ASSERT_TRUE(mc.isArbiter());
    ASSERT_EQUALS(0.0, mc.getPriority());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "arbiterOnly"
                                       << false),
                            &tagConfig));
    ASSERT_TRUE(!mc.isArbiter());
    ASSERT_EQUALS(1.0, mc.getPriority());
}

TEST(MemberConfig, ParseHidden) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "hidden"
                                       << 1.0),
                            &tagConfig));
    ASSERT_TRUE(mc.isHidden());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "hidden"
                                       << false),
                            &tagConfig));
    ASSERT_TRUE(!mc.isHidden());
    ASSERT_EQUALS(ErrorCodes::TypeMismatch,
                  mc.initialize(BSON("_id" << 0 << "host"
                                           << "h"
                                           << "hidden"
                                           << "1.0"),
                                &tagConfig));
}

TEST(MemberConfig, ParseBuildIndexes) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "buildIndexes"
                                       << 1.0),
                            &tagConfig));
    ASSERT_TRUE(mc.shouldBuildIndexes());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "buildIndexes"
                                       << false),
                            &tagConfig));
    ASSERT_TRUE(!mc.shouldBuildIndexes());
}

TEST(MemberConfig, ParseVotes) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << 1.0),
                            &tagConfig));
    ASSERT_TRUE(mc.isVoter());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << 0
                                       << "priority"
                                       << 0),
                            &tagConfig));
    ASSERT_FALSE(mc.isVoter());

    // For backwards compatibility, truncate 1.X to 1, and 0.X to 0 (and -0.X to 0).
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << 1.5),
                            &tagConfig));
    ASSERT_TRUE(mc.isVoter());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << 0.5),
                            &tagConfig));
    ASSERT_FALSE(mc.isVoter());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << -0.5),
                            &tagConfig));
    ASSERT_FALSE(mc.isVoter());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << 2),
                            &tagConfig));

    ASSERT_EQUALS(ErrorCodes::TypeMismatch,
                  mc.initialize(BSON("_id" << 0 << "host"
                                           << "h"
                                           << "votes"
                                           << Date_t::fromMillisSinceEpoch(2)),
                                &tagConfig));
}

TEST(MemberConfig, ParsePriority) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 1),
                            &tagConfig));
    ASSERT_EQUALS(1.0, mc.getPriority());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 0),
                            &tagConfig));
    ASSERT_EQUALS(0.0, mc.getPriority());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 100.8),
                            &tagConfig));
    ASSERT_EQUALS(100.8, mc.getPriority());

    ASSERT_EQUALS(ErrorCodes::TypeMismatch,
                  mc.initialize(BSON("_id" << 0 << "host"
                                           << "h"
                                           << "priority"
                                           << Date_t::fromMillisSinceEpoch(2)),
                                &tagConfig));
}

TEST(MemberConfig, ParseSlaveDelay) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "slaveDelay"
                                       << 100),
                            &tagConfig));
    ASSERT_EQUALS(Seconds(100), mc.getSlaveDelay());
}

TEST(MemberConfig, ParseTags) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "tags"
                                       << BSON("k1"
                                               << "v1"
                                               << "k2"
                                               << "v2")),
                            &tagConfig));
    ASSERT_EQUALS(5U, mc.getNumTags());
    ASSERT_EQUALS(5, std::distance(mc.tagsBegin(), mc.tagsEnd()));
    ASSERT_EQUALS(1, std::count(mc.tagsBegin(), mc.tagsEnd(), tagConfig.findTag("k1", "v1")));
    ASSERT_EQUALS(1, std::count(mc.tagsBegin(), mc.tagsEnd(), tagConfig.findTag("k2", "v2")));
    ASSERT_EQUALS(1, std::count(mc.tagsBegin(), mc.tagsEnd(), tagConfig.findTag("$voter", "0")));
    ASSERT_EQUALS(1,
                  std::count(mc.tagsBegin(), mc.tagsEnd(), tagConfig.findTag("$electable", "0")));
    ASSERT_EQUALS(1, std::count(mc.tagsBegin(), mc.tagsEnd(), tagConfig.findTag("$all", "0")));
}

TEST(MemberConfig, ValidateFailsWithIdOutOfRange) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << -1 << "host"
                                       << "localhost:12345"),
                            &tagConfig));
    ASSERT_EQUALS(ErrorCodes::BadValue, mc.validate());
    ASSERT_OK(mc.initialize(BSON("_id" << 256 << "host"
                                       << "localhost:12345"),
                            &tagConfig));
    ASSERT_EQUALS(ErrorCodes::BadValue, mc.validate());
}

TEST(MemberConfig, ValidateVotes) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;

    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << 1.0),
                            &tagConfig));
    ASSERT_OK(mc.validate());
    ASSERT_TRUE(mc.isVoter());

    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << 0
                                       << "priority"
                                       << 0),
                            &tagConfig));
    ASSERT_OK(mc.validate());
    ASSERT_FALSE(mc.isVoter());

    // For backwards compatibility, truncate 1.X to 1, and 0.X to 0 (and -0.X to 0).
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << 1.5),
                            &tagConfig));
    ASSERT_OK(mc.validate());
    ASSERT_TRUE(mc.isVoter());

    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << 0.5
                                       << "priority"
                                       << 0),
                            &tagConfig));
    ASSERT_OK(mc.validate());
    ASSERT_FALSE(mc.isVoter());

    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << -0.5
                                       << "priority"
                                       << 0),
                            &tagConfig));
    ASSERT_OK(mc.validate());
    ASSERT_FALSE(mc.isVoter());

    // Invalid values
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << 2),
                            &tagConfig));
    ASSERT_EQUALS(ErrorCodes::BadValue, mc.validate());

    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << -1),
                            &tagConfig));
    ASSERT_EQUALS(ErrorCodes::BadValue, mc.validate());
}

TEST(MemberConfig, ValidatePriorityRanges) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 0),
                            &tagConfig));
    ASSERT_OK(mc.validate());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 1000),
                            &tagConfig));
    ASSERT_OK(mc.validate());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << -1),
                            &tagConfig));
    ASSERT_EQUALS(ErrorCodes::BadValue, mc.validate());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 1001),
                            &tagConfig));
    ASSERT_EQUALS(ErrorCodes::BadValue, mc.validate());
}

TEST(MemberConfig, ValidateSlaveDelays) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 0
                                       << "slaveDelay"
                                       << 0),
                            &tagConfig));
    ASSERT_OK(mc.validate());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 0
                                       << "slaveDelay"
                                       << 3600 * 10),
                            &tagConfig));
    ASSERT_OK(mc.validate());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 0
                                       << "slaveDelay"
                                       << -1),
                            &tagConfig));
    ASSERT_EQUALS(ErrorCodes::BadValue, mc.validate());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 0
                                       << "slaveDelay"
                                       << 3600 * 24 * 400),
                            &tagConfig));
    ASSERT_EQUALS(ErrorCodes::BadValue, mc.validate());
}

TEST(MemberConfig, ValidatePriorityAndSlaveDelayRelationship) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 1
                                       << "slaveDelay"
                                       << 60),
                            &tagConfig));
    ASSERT_EQUALS(ErrorCodes::BadValue, mc.validate());
}

TEST(MemberConfig, ValidatePriorityAndHiddenRelationship) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 1
                                       << "hidden"
                                       << true),
                            &tagConfig));
    ASSERT_EQUALS(ErrorCodes::BadValue, mc.validate());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 1
                                       << "hidden"
                                       << false),
                            &tagConfig));
    ASSERT_OK(mc.validate());
}

TEST(MemberConfig, ValidatePriorityAndBuildIndexesRelationship) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 1
                                       << "buildIndexes"
                                       << false),
                            &tagConfig));

    ASSERT_EQUALS(ErrorCodes::BadValue, mc.validate());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "priority"
                                       << 1
                                       << "buildIndexes"
                                       << true),
                            &tagConfig));
    ASSERT_OK(mc.validate());
}

TEST(MemberConfig, ValidateArbiterVotesRelationship) {
    ReplSetTagConfig tagConfig;
    MemberConfig mc;
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << 1
                                       << "arbiterOnly"
                                       << true),
                            &tagConfig));
    ASSERT_OK(mc.validate());

    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << 0
                                       << "priority"
                                       << 0
                                       << "arbiterOnly"
                                       << false),
                            &tagConfig));
    ASSERT_OK(mc.validate());
    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << 1
                                       << "arbiterOnly"
                                       << false),
                            &tagConfig));
    ASSERT_OK(mc.validate());

    ASSERT_OK(mc.initialize(BSON("_id" << 0 << "host"
                                       << "h"
                                       << "votes"
                                       << 0
                                       << "arbiterOnly"
                                       << true),
                            &tagConfig));
    ASSERT_EQUALS(ErrorCodes::BadValue, mc.validate());
}

}  // namespace
}  // namespace repl
}  // namespace mongo