/*
 * -------------------------------------------------------------------------
 * This file is part of the MindStudio project.
 * Copyright (c) 2025 Huawei Technologies Co.,Ltd.
 *
 * MindStudio is licensed under Mulan PSL v2.
 * You can use this software according to the terms and conditions of the Mulan PSL v2.
 * You may obtain a copy of Mulan PSL v2 at:
 *
 *          http://license.coscl.org.cn/MulanPSL2
 *
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
 * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
 * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
 * See the Mulan PSL v2 for more details.
 * -------------------------------------------------------------------------
 */
#include <gtest/gtest.h>
#include <utility>
#include "IECurveHandler.h"
using namespace Dic::Module::IE;
class IECurveHandlerTest : public ::testing::Test {
  protected:
    class MockRepo : public Dic::Module::IE::CurveRepo {
      public:
        std::vector<Dic::Module::ColumnAtt> QueryTableInfoByName(
            const std::string &fileId, const std::string &tableName) {
            std::vector<Dic::Module::ColumnAtt> res;
            Dic::Module::ColumnAtt col1;
            col1.key = "lll";
            res.emplace_back(col1);
            Dic::Module::ColumnAtt col2;
            col2.key = "jjj";
            res.emplace_back(col2);
            return res;
        }

        std::vector<std::map<std::string, std::string>> QueryDataByColumn(const std::string &fileId,
            const std::string &tableName, const std::vector<Dic::Module::ColumnAtt> &columns) {
            std::vector<std::map<std::string, std::string>> res;
            std::map<std::string, std::string> data1;
            data1["lll"] = "mmm1";
            data1["jjj"] = "ggg2";
            res.emplace_back(data1);
            std::map<std::string, std::string> data2;
            data2["lll"] = "mmm3";
            data2["jjj"] = "ggg4";
            res.emplace_back(data2);
            res.emplace_back(data2);
            return res;
        }
    };

    class MockIECurveHandler : public Dic::Module::IE::IECurveHandler {
      public:
        void SetRepo(std::shared_ptr<CurveRepo> mockRepo) { repo = std::move(mockRepo); }
    };
};

TEST_F(IECurveHandlerTest, TestHandler) {
    std::shared_ptr<MockRepo> mockRepo = std::make_shared<MockRepo>();
    std::shared_ptr<MockIECurveHandler> handler = std::make_shared<MockIECurveHandler>();
    std::unique_ptr<Dic::Protocol::IEUsageViewParamsRequest> request =
        std::make_unique<Dic::Protocol::IEUsageViewParamsRequest>();
    request->params.type = "llllllllll";
    handler->SetRepo(mockRepo);
    auto res = handler->HandleRequest(std::move(request));
    EXPECT_EQ(res, true);
}