module_bootstrap_test.cpp

// Copyright 2021-present StarRocks, Inc. All rights reserved.
//
// 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
//
//     https://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 "connector/connector_registry.h"
#include "connector_primitive/connector.h"
#include "gtest/gtest.h"
#include "module/connector_bootstrap.h"

namespace starrocks::connector {

TEST(ModuleBootstrapTest, BootstrapBuiltinConnectorsInstallsConnectorsIdempotently) {
    auto* registry = ConnectorRegistry::default_instance();
    ASSERT_NE(nullptr, registry);

    auto status = bootstrap_builtin_connectors();
    ASSERT_TRUE(status.ok()) << status;
    const auto* cache_stats = registry->get(Connector::CACHE_STATS);
    ASSERT_NE(nullptr, cache_stats);
    EXPECT_EQ(ConnectorType::CACHE_STATS, cache_stats->connector_type());
    const auto* hive = registry->get(Connector::HIVE);
    ASSERT_NE(nullptr, hive);
    EXPECT_EQ(ConnectorType::HIVE, hive->connector_type());
    const auto* file = registry->get(Connector::FILE);
    ASSERT_NE(nullptr, file);
    EXPECT_EQ(ConnectorType::FILE, file->connector_type());
    const auto* lake = registry->get(Connector::LAKE);
    ASSERT_NE(nullptr, lake);
    EXPECT_EQ(ConnectorType::LAKE, lake->connector_type());
#ifndef __APPLE__
    const auto* iceberg = registry->get(Connector::ICEBERG);
    ASSERT_NE(nullptr, iceberg);
    EXPECT_EQ(ConnectorType::ICEBERG, iceberg->connector_type());
#endif

    status = bootstrap_builtin_connectors();
    ASSERT_TRUE(status.ok()) << status;
    EXPECT_EQ(cache_stats, registry->get(Connector::CACHE_STATS));
    EXPECT_EQ(hive, registry->get(Connector::HIVE));
    EXPECT_EQ(file, registry->get(Connector::FILE));
    EXPECT_EQ(lake, registry->get(Connector::LAKE));
#ifndef __APPLE__
    EXPECT_EQ(iceberg, registry->get(Connector::ICEBERG));
#endif
}

} // namespace starrocks::connector