/*
 * Copyright 2020 Google Inc.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met: redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer;
 * redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution;
 * neither the name of the copyright holders nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <gtest/gtest.h>

// For EXPECT_THAT and HasSubstr
#include <gmock/gmock.h>

#include "args.hh"
#include "call_type.hh"

// Simple substitute definitions for Args and DispatchTable, with an int so
// we can tell instances apart.
class DispatchTable { public: int i; };

class TestCallType : public CallType
{
  protected:
    bool isDefault() const override { return testIsDefault; }
    void
    init() override
    {
        testInitHappened = true;
        CallType::init();
    }

    void printBrief(std::ostream &os) const override { os << testBrief; }
    void printDesc(std::ostream &os) const override { os << testDesc; }
  public:
    TestCallType(const std::string &_name) : CallType(_name) {}

    static std::map<std::string, CallType &> &testGetMap() { return map(); }

    // Usage strings to return.
    std::string testBrief;
    std::string testDesc;

    // Return this dispatch table when requested.
    DispatchTable testDt = { 0 };
    const DispatchTable &getDispatch() const override { return testDt; }

    // Whether this call type should be considered default.
    bool testIsDefault = false;

    // Whether init has been called.
    bool testInitHappened = false;
};

TEST(CallTypeTest, Constructor)
{
    auto &map = TestCallType::testGetMap();

    // There should be no call types yet.
    EXPECT_EQ(map.size(), 0);

    // Create one.
    TestCallType test_ct("test_ct");

    // Set the dispatch table to something we'll recognize.
    test_ct.testDt.i = 0xaa55;

    // Verify that the list of all call types has one in it now.
    EXPECT_EQ(map.size(), 1);

    // Verify that that was our call type by verifying that the dispatch table
    // has our signature.
    EXPECT_EQ(map.begin()->second.getDispatch().i, 0xaa55);
}

TEST(CallTypeTest, DetectOne)
{
    auto &map = TestCallType::testGetMap();

    // One option selected.
    TestCallType option1("option1");
    option1.testIsDefault = true;
    option1.testDt.i = 1;

    EXPECT_EQ(map.size(), 1);

    Args args1({"--option1"});

    EXPECT_FALSE(option1.testInitHappened);

    auto *ct = CallType::detect(args1);

    // Verify that we selected the only option.
    EXPECT_TRUE(option1.testInitHappened);
    EXPECT_EQ(ct, &option1);

    // One option, selecting the default.
    option1.testInitHappened = false;

    // Args will not match.
    Args args2({"--option2"});

    auto *def_ct = CallType::detect(args2);

    // Verify that the one option was defaulted to.
    EXPECT_TRUE(option1.testInitHappened);
    EXPECT_EQ(def_ct, &option1);
}

TEST(CallTypeTest, DetectTwo)
{
    auto &map = TestCallType::testGetMap();

    // One of two options selected.
    TestCallType option1("option1");
    option1.testIsDefault = true;
    option1.testDt.i = 1;

    TestCallType option2("option2");
    option2.testIsDefault = false;
    option2.testDt.i = 2;

    EXPECT_EQ(map.size(), 2);

    // Select the first option.
    Args args1({"--option1"});

    EXPECT_FALSE(option1.testInitHappened);
    EXPECT_FALSE(option2.testInitHappened);

    auto *ct1 = CallType::detect(args1);

    // Verify that we selected the first option.
    EXPECT_TRUE(option1.testInitHappened);
    EXPECT_FALSE(option2.testInitHappened);
    EXPECT_EQ(ct1, &option1);

    option1.testInitHappened = false;
    option2.testInitHappened = false;

    // Select the second option.
    Args args2({"--option2"});

    auto *ct2 = CallType::detect(args2);

    // Verify that we selected the second option.
    EXPECT_FALSE(option1.testInitHappened);
    EXPECT_TRUE(option2.testInitHappened);
    EXPECT_EQ(ct2, &option2);

    option1.testInitHappened = false;
    option2.testInitHappened = false;

    // Default to the first option.
    Args args3({"--option3"});

    auto *def_ct1 = CallType::detect(args3);

    // Verify that we selected the first option.
    EXPECT_TRUE(option1.testInitHappened);
    EXPECT_FALSE(option2.testInitHappened);
    EXPECT_EQ(def_ct1, &option1);

    option1.testInitHappened = false;
    option2.testInitHappened = false;

    // Default to the second option.
    option1.testIsDefault = false;
    option2.testIsDefault = true;

    auto *def_ct2 = CallType::detect(args3);

    // Verify that we selected the second option.
    EXPECT_FALSE(option1.testInitHappened);
    EXPECT_TRUE(option2.testInitHappened);
    EXPECT_EQ(def_ct2, &option2);

    option1.testInitHappened = false;
    option2.testInitHappened = false;
}

TEST(CallTypeTest, Usage)
{
    auto &map = TestCallType::testGetMap();

    TestCallType ct1("ct1");
    ct1.testBrief = "brief 1";
    ct1.testDesc = "A longer description of call type 1, which is long.";

    TestCallType ct2("ct2");
    ct2.testBrief = "short 2";
    ct2.testDesc = "Very verbose text saying what call type 2 is, "
        "and is different from 1.";

    EXPECT_EQ(map.size(), 2);

    auto summary = CallType::usageSummary();

    // For now, just expect that the brief and full descriptive text shows up
    // in the summary somewhere.
    //
    // More strict checks might test that things were placed in the right
    // order, were on their own lines when appropriate, etc.
    EXPECT_THAT(summary, testing::HasSubstr(ct1.testBrief));
    EXPECT_THAT(summary, testing::HasSubstr(ct1.testDesc));
    EXPECT_THAT(summary, testing::HasSubstr(ct2.testBrief));
    EXPECT_THAT(summary, testing::HasSubstr(ct2.testDesc));
    EXPECT_THAT(summary, testing::HasSubstr(ct2.testDesc));
}