/**
 * tests/t20021/test_case.h
 *
 * Copyright (c) 2021-2023 Bartek Kryza <bkryza@gmail.com>
 *
 * 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.
 */

TEST_CASE("t20021", "[test-case][sequence]")
{
    auto [config, db] = load_config("t20021");

    auto diagram = config.diagrams["t20021_sequence"];

    REQUIRE(diagram->name == "t20021_sequence");

    auto model = generate_sequence_diagram(*db, diagram);

    REQUIRE(model->name() == "t20021_sequence");
    {
        auto puml = generate_sequence_puml(diagram, *model);
        AliasMatcher _A(puml);

        REQUIRE_THAT(puml, StartsWith("@startuml"));
        REQUIRE_THAT(puml, EndsWith("@enduml\n"));

        // Check if all calls exist
        REQUIRE_THAT(puml, HasCall(_A("tmain()"), _A("A"), "a1()"));
        REQUIRE_THAT(
            puml, HasCallInControlCondition(_A("tmain()"), _A("A"), "a2()"));
        REQUIRE_THAT(puml, HasCall(_A("tmain()"), _A("A"), "a3()"));

        REQUIRE_THAT(puml, !HasCall(_A("tmain()"), _A("B"), "b1()"));
        REQUIRE_THAT(puml, HasCall(_A("tmain()"), _A("B"), "b2()"));

        REQUIRE_THAT(
            puml, HasCallInControlCondition(_A("tmain()"), _A("C"), "c1()"));
        REQUIRE_THAT(
            puml, HasCallInControlCondition(_A("tmain()"), _A("C"), "c2()"));

        // TODO: Why is this not working?
        //    REQUIRE_THAT(
        //        puml, HasCallInControlCondition(_A("tmain()"), _A("C"),
        //        "c3()"));
        REQUIRE_THAT(
            puml, HasCallInControlCondition(_A("tmain()"), _A("C"), "c4()"));
        REQUIRE_THAT(puml, HasCall(_A("C"), _A("C"), "c5()"));
        REQUIRE_THAT(puml,
            HasCallInControlCondition(_A("tmain()"), _A("C"), "contents()"));

        save_puml(
            config.output_directory() + "/" + diagram->name + ".puml", puml);
    }

    {
        auto j = generate_sequence_json(diagram, *model);

        using namespace json;

        std::vector<int> messages = {
            FindMessage(j, "tmain()", "C", "c4()"),
            FindMessage(j, "C", "C", "c5()"),
            FindMessage(j, "tmain()", "A", "a3()"),
            FindMessage(j, "tmain()", "A", "a2()"),
            FindMessage(j, "tmain()", "C", "c1()"),
            FindMessage(j, "tmain()", "C", "c2()"),
            FindMessage(j, "tmain()", "A", "a1()"),
            FindMessage(j, "tmain()", "C", "c3()"),
            FindMessage(j, "tmain()", "B", "b2()"),
            FindMessage(j, "tmain()", "C", "contents()")
            // TODO: Repeated messge gets wrong index
            // FindMessage(j, "tmain()", "B", "b2()")
        };

        REQUIRE(std::is_sorted(messages.begin(), messages.end()));

        save_json(config.output_directory() + "/" + diagram->name + ".json", j);
    }
}