src/test/cpp/ut/base/HostInfoTest.cpp

#include "gtest/gtest.h"
#include "HostInfo.h"
#include "rocketmq/RocketMQ.h"
#include <cstdlib>
#include "fmt/format.h"

ROCKETMQ_NAMESPACE_BEGIN

class HostInfoTest : public testing::Test {
public:
  void SetUp() override {}

  void TearDown() override {}

protected:
  std::string site_{"site"};
  std::string unit_{"unit"};
  std::string app_{"app"};
  std::string stage_{"stage"};

  void SetEnv(const char* key, const char* value) {
    int overwrite = 1;
    #ifdef _WIN32
    std::string env;
    env.append(key);
    env.push_back('=');
    env.append(value);
    _putenv(env.c_str());
    #else
    setenv(key, value, overwrite);
    #endif
  }
};

TEST_F(HostInfoTest, testQueryString) {
  SetEnv(HostInfo::ENV_LABEL_SITE, site_.c_str());
  SetEnv(HostInfo::ENV_LABEL_UNIT, unit_.c_str());
  SetEnv(HostInfo::ENV_LABEL_APP, app_.c_str());
  SetEnv(HostInfo::ENV_LABEL_STAGE, stage_.c_str());

  HostInfo host_info;
  std::string query_string = host_info.queryString();
  std::string query_string_template("labels=site:{},unit:{},app:{},stage:{}");
  EXPECT_EQ(query_string, fmt::format(query_string_template, site_, unit_, app_, stage_));
}

ROCKETMQ_NAMESPACE_END