/**
 * Copyright (c) 2021 OceanBase
 * OceanBase CE is licensed under Mulan PubL v2.
 * You can use this software according to the terms and conditions of the Mulan PubL v2.
 * You may obtain a copy of Mulan PubL v2 at:
 *          http://license.coscl.org.cn/MulanPubL-2.0
 * 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 PubL v2 for more details.
 */

#define USING_LOG_PREFIX RS
#include <gtest/gtest.h>
#include <gmock/gmock.h>
#define  private public
#include "rootserver/ob_balance_ls_primary_zone.h"
#include "share/ls/ob_ls_status_operator.h"
#include "rootserver/ob_primary_ls_service.h"
#include "share/ls/ob_ls_operator.h"
namespace oceanbase {
using namespace common;
using namespace share;
using ::testing::_;
using ::testing::Invoke;
using ::testing::Return;

namespace rootserver{
//multiclustermanage
class TestPrimaryLSService : public testing::Test
{
public:
  TestPrimaryLSService() {}
  virtual ~TestPrimaryLSService(){}
  virtual void SetUp() {};
  virtual void TearDown() {}
  virtual void TestBody() {}
}
;

TEST_F(TestPrimaryLSService, zone_balance)
{
  int ret = OB_SUCCESS;
  common::ObArray<common::ObZone> primary_zone_array;
  share::ObLSPrimaryZoneInfoArray primary_zone_infos;
  ObArray<ObZone> ls_primary_zone;
  ObSEArray<uint64_t, 3> count_group_by_zone;

  //case 1, set primary_zone to 'z1', ls to 'z2'
  ObZone z1("z1");
  ObZone z2("z2");
  ObZone z3("z3");
  ObZone z4("z4");
  ObZone z5("z5");
  ObLSPrimaryZoneInfo info;
  ObLSID ls_id(1001);
  ret = info.init(1002, 1001, ls_id, z2, z1.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_array.push_back(z1);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = ObBalanceLSPrimaryZone::set_ls_to_primary_zone_(primary_zone_array, primary_zone_infos, ls_primary_zone, count_group_by_zone);
  ASSERT_EQ(ret, OB_SUCCESS);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(z1, ls_primary_zone.at(0));
  ASSERT_EQ(1, count_group_by_zone.at(0));
  ret = ObBalanceLSPrimaryZone::balance_ls_primary_zone_(primary_zone_array, ls_primary_zone, count_group_by_zone);
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(z1, ls_primary_zone.at(0));
  ASSERT_EQ(1, count_group_by_zone.at(0));

  //case2, primary zone is z1, z2, ls to z2, z2;
  ls_primary_zone.reset();
  count_group_by_zone.reset();
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_array.push_back(z2);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = ObBalanceLSPrimaryZone::set_ls_to_primary_zone_(primary_zone_array, primary_zone_infos, ls_primary_zone, count_group_by_zone);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(z2, ls_primary_zone.at(0));
  ASSERT_EQ(z2, ls_primary_zone.at(1));
  ASSERT_EQ(0, count_group_by_zone.at(0));
  ASSERT_EQ(2, count_group_by_zone.at(1));
  ret = ObBalanceLSPrimaryZone::balance_ls_primary_zone_(primary_zone_array, ls_primary_zone, count_group_by_zone);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(z1, ls_primary_zone.at(0));
  ASSERT_EQ(z2, ls_primary_zone.at(1));
  ASSERT_EQ(1, count_group_by_zone.at(0));
  ASSERT_EQ(1, count_group_by_zone.at(1));
  
  //case 3 primary zone to z1, z3, ls to z2, z2;
  ls_primary_zone.reset();
  count_group_by_zone.reset();
  primary_zone_array.reset();
  ret = primary_zone_array.push_back(z1);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_array.push_back(z3);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = ObBalanceLSPrimaryZone::set_ls_to_primary_zone_(primary_zone_array, primary_zone_infos, ls_primary_zone, count_group_by_zone);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(z1, ls_primary_zone.at(0));
  ASSERT_EQ(z3, ls_primary_zone.at(1));
  ASSERT_EQ(1, count_group_by_zone.at(0));
  ASSERT_EQ(1, count_group_by_zone.at(1));
  ret = ObBalanceLSPrimaryZone::balance_ls_primary_zone_(primary_zone_array, ls_primary_zone, count_group_by_zone);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(z1, ls_primary_zone.at(0));
  ASSERT_EQ(z3, ls_primary_zone.at(1));
  ASSERT_EQ(1, count_group_by_zone.at(0));
  ASSERT_EQ(1, count_group_by_zone.at(1)); 

  //case 3 primary zone to z1,z3,z2, ls to z2, z2;
  ls_primary_zone.reset();
  count_group_by_zone.reset();
  ret = primary_zone_array.push_back(z2);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = ObBalanceLSPrimaryZone::set_ls_to_primary_zone_(primary_zone_array, primary_zone_infos, ls_primary_zone, count_group_by_zone);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(z2, ls_primary_zone.at(0));
  ASSERT_EQ(z2, ls_primary_zone.at(1));
  ASSERT_EQ(0, count_group_by_zone.at(0));
  ASSERT_EQ(0, count_group_by_zone.at(1));
  ASSERT_EQ(2, count_group_by_zone.at(2));
  ret = ObBalanceLSPrimaryZone::balance_ls_primary_zone_(primary_zone_array, ls_primary_zone, count_group_by_zone);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(z1, ls_primary_zone.at(0));
  ASSERT_EQ(z2, ls_primary_zone.at(1));
  ASSERT_EQ(1, count_group_by_zone.at(0));
  ASSERT_EQ(0, count_group_by_zone.at(1)); 
  ASSERT_EQ(1, count_group_by_zone.at(2));

  //case 4 pirmary zone to  z1,z3,z2, ls to z1, z2, z2, z2
  ls_primary_zone.reset();
  count_group_by_zone.reset();
  ret = info.init(1002, 1001, ls_id, z2, z1.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = info.init(1002, 1001, ls_id, z1, z1.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = ObBalanceLSPrimaryZone::set_ls_to_primary_zone_(primary_zone_array, primary_zone_infos, ls_primary_zone, count_group_by_zone);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(z2, ls_primary_zone.at(0));
  ASSERT_EQ(z2, ls_primary_zone.at(1));
  ASSERT_EQ(z2, ls_primary_zone.at(2));
  ASSERT_EQ(z1, ls_primary_zone.at(3));
  ASSERT_EQ(1, count_group_by_zone.at(0));
  ASSERT_EQ(0, count_group_by_zone.at(1));
  ASSERT_EQ(3, count_group_by_zone.at(2));
  ret = ObBalanceLSPrimaryZone::balance_ls_primary_zone_(primary_zone_array, ls_primary_zone, count_group_by_zone);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(z3, ls_primary_zone.at(0));
  ASSERT_EQ(z2, ls_primary_zone.at(1));
  ASSERT_EQ(z2, ls_primary_zone.at(2));
  ASSERT_EQ(z1, ls_primary_zone.at(3));
  ASSERT_EQ(1, count_group_by_zone.at(0));
  ASSERT_EQ(1, count_group_by_zone.at(1)); 
  ASSERT_EQ(2, count_group_by_zone.at(2));

  //case 5 pirmary zone to  z1,z3,z2, ls to z1, z1, z1, z5, z5
  ls_primary_zone.reset();
  count_group_by_zone.reset();
  primary_zone_infos.reset();
  ret = info.init(1002, 1001, ls_id, z1, z1.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = info.init(1002, 1001, ls_id, z5, z1.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = ObBalanceLSPrimaryZone::set_ls_to_primary_zone_(primary_zone_array, primary_zone_infos, ls_primary_zone, count_group_by_zone);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(z1, ls_primary_zone.at(0));
  ASSERT_EQ(z1, ls_primary_zone.at(1));
  ASSERT_EQ(z1, ls_primary_zone.at(2));
  ASSERT_EQ(z3, ls_primary_zone.at(3));
  ASSERT_EQ(z2, ls_primary_zone.at(4));
  ASSERT_EQ(3, count_group_by_zone.at(0));
  ASSERT_EQ(1, count_group_by_zone.at(1));
  ASSERT_EQ(1, count_group_by_zone.at(2));
  ret = ObBalanceLSPrimaryZone::balance_ls_primary_zone_(primary_zone_array, ls_primary_zone, count_group_by_zone);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(z3, ls_primary_zone.at(0));
  ASSERT_EQ(z1, ls_primary_zone.at(1));
  ASSERT_EQ(z1, ls_primary_zone.at(2));
  ASSERT_EQ(z3, ls_primary_zone.at(3));
  ASSERT_EQ(z2, ls_primary_zone.at(4));
  ASSERT_EQ(2, count_group_by_zone.at(0));
  ASSERT_EQ(2, count_group_by_zone.at(1));
  ASSERT_EQ(1, count_group_by_zone.at(2));

  //case 6 pirmary zone to  z1,z3,z2, ls to z1, z1, z1,z1,z2,z2,z2,z2

  ls_primary_zone.reset();
  count_group_by_zone.reset();
  primary_zone_infos.reset();
  ret = info.init(1002, 1001, ls_id, z1, z1.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);

  ret = info.init(1002, 1001, ls_id, z2, z1.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = ObBalanceLSPrimaryZone::set_ls_to_primary_zone_(primary_zone_array, primary_zone_infos, ls_primary_zone, count_group_by_zone);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(z1, ls_primary_zone.at(0));
  ASSERT_EQ(z1, ls_primary_zone.at(1));
  ASSERT_EQ(z1, ls_primary_zone.at(2));
  ASSERT_EQ(z1, ls_primary_zone.at(3));
  ASSERT_EQ(z2, ls_primary_zone.at(4));
  ASSERT_EQ(z2, ls_primary_zone.at(5));
  ASSERT_EQ(z2, ls_primary_zone.at(6));
  ASSERT_EQ(z2, ls_primary_zone.at(7));
  ASSERT_EQ(4, count_group_by_zone.at(0));
  ASSERT_EQ(0, count_group_by_zone.at(1));
  ASSERT_EQ(4, count_group_by_zone.at(2));
  ret = ObBalanceLSPrimaryZone::balance_ls_primary_zone_(primary_zone_array, ls_primary_zone, count_group_by_zone);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(z3, ls_primary_zone.at(0));
  ASSERT_EQ(z1, ls_primary_zone.at(1));
  ASSERT_EQ(z1, ls_primary_zone.at(2));
  ASSERT_EQ(z1, ls_primary_zone.at(3));
  ASSERT_EQ(z3, ls_primary_zone.at(4));
  ASSERT_EQ(z2, ls_primary_zone.at(5));
  ASSERT_EQ(z2, ls_primary_zone.at(6));
  ASSERT_EQ(z2, ls_primary_zone.at(7));
  ASSERT_EQ(3, count_group_by_zone.at(0));
  ASSERT_EQ(2, count_group_by_zone.at(1));
  ASSERT_EQ(3, count_group_by_zone.at(2));

  //case 8 pirmary zone to  z1,z2, z3. ls to z1, z1, z2, z2
  ls_primary_zone.reset();
  count_group_by_zone.reset();
  primary_zone_infos.reset();
  primary_zone_array.reset();
  ret = primary_zone_array.push_back(z1);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_array.push_back(z2);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_array.push_back(z3);
  ASSERT_EQ(ret, OB_SUCCESS);

  ret = info.init(1002, 1001, ls_id, z1, z1.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = info.init(1002, 1001, ls_id, z1, z1.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = info.init(1002, 1001, ls_id, z2, z1.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = info.init(1002, 1001, ls_id, z2, z1.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ret = primary_zone_infos.push_back(info);
  ASSERT_EQ(ret, OB_SUCCESS);

  ret = ObBalanceLSPrimaryZone::set_ls_to_primary_zone_(primary_zone_array, primary_zone_infos, ls_primary_zone, count_group_by_zone);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(z1, ls_primary_zone.at(0));
  ASSERT_EQ(z1, ls_primary_zone.at(1));
  ASSERT_EQ(z2, ls_primary_zone.at(2));
  ASSERT_EQ(z2, ls_primary_zone.at(3));
  ASSERT_EQ(2, count_group_by_zone.at(0));
  ASSERT_EQ(2, count_group_by_zone.at(1));
  ASSERT_EQ(0, count_group_by_zone.at(2));
  ret = ObBalanceLSPrimaryZone::balance_ls_primary_zone_(primary_zone_array, ls_primary_zone, count_group_by_zone);
  LOG_INFO("set ls to primary zone", K(ls_primary_zone), K(count_group_by_zone));
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(z3, ls_primary_zone.at(0));
  ASSERT_EQ(z1, ls_primary_zone.at(1));
  ASSERT_EQ(z2, ls_primary_zone.at(2));
  ASSERT_EQ(z2, ls_primary_zone.at(3));
  ASSERT_EQ(1, count_group_by_zone.at(0));
  ASSERT_EQ(2, count_group_by_zone.at(1));
  ASSERT_EQ(1, count_group_by_zone.at(2));

}
TEST_F(TestPrimaryLSService, LS_FLAG)
{
  int ret = OB_SUCCESS;
  ObLSFlag flag;
  ObLSFlagStr str;
  ObLSFlagStr empty_str;
  ObLSFlagStr str0("DUPLICATE");
  ObLSFlagStr str1("DUPLICATE ");
  ObLSFlagStr str2(" DUPLICATE ");
  ObLSFlagStr str3("BLOCK_TABLET_IN");
  ObLSFlagStr str4("BLOCK_TABLET_IN ");
  ObLSFlagStr str5("BLOCK_TABLET_IN|DUPLICATE");
  ObLSFlagStr str6("DUPLICATE|BLOCK_TABLET_IN");
  ObLSFlagStr str7("BLOCK_TABLET_IN  | DUPLICATE");
  ObLSFlagStr str8("BLOCK_TABLET_IN,DUPLICATE");
  ObLSFlagStr str9("BLOCK_TABLET_IN DUPLICATE");

  ret = flag.flag_to_str(str);
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(empty_str, str);
  ASSERT_EQ(0, flag.flag_);
  LOG_INFO("test", K(flag), K(str));

  flag.set_block_tablet_in();
  ASSERT_EQ(2, flag.flag_);
  ret = flag.flag_to_str(str);
  ASSERT_EQ(str3, str);
  LOG_INFO("test", K(flag), K(str));

  flag.clear_block_tablet_in();
  ASSERT_EQ(0, flag.flag_);
  ret = flag.flag_to_str(str);
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(empty_str, str);
  LOG_INFO("test", K(flag), K(str));

  flag.set_duplicate();
  ASSERT_EQ(1, flag.flag_);
  ret = flag.flag_to_str(str);
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(str0, str);
  LOG_INFO("test", K(flag), K(str));

  flag.set_block_tablet_in();
  ASSERT_EQ(3, flag.flag_);
  ret = flag.flag_to_str(str);
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(str6, str);
  LOG_INFO("test", K(flag), K(str));

  flag.clear_block_tablet_in();
  ASSERT_EQ(1, flag.flag_);
  ret = flag.flag_to_str(str);
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(str0, str);
  LOG_INFO("test", K(flag), K(str));

  ret = flag.str_to_flag(empty_str.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(0, flag.flag_);
  LOG_INFO("test", K(flag));

  ret = flag.str_to_flag(str0.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(1, flag.flag_);
  LOG_INFO("test", K(flag));

  ret = flag.str_to_flag(str1.str());
  ASSERT_EQ(OB_ERR_UNEXPECTED, ret);

  ret = flag.str_to_flag(str2.str());
  ASSERT_EQ(OB_ERR_UNEXPECTED, ret);

  ret = flag.str_to_flag(str3.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(2, flag.flag_);
  LOG_INFO("test", K(flag));

  ret = flag.str_to_flag(str4.str());
  ASSERT_EQ(OB_ERR_UNEXPECTED, ret);

  ret = flag.str_to_flag(str5.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(3, flag.flag_);
  LOG_INFO("test", K(flag));

  ret = flag.str_to_flag(str6.str());
  ASSERT_EQ(ret, OB_SUCCESS);
  ASSERT_EQ(3, flag.flag_);

  LOG_INFO("test", K(flag));
  ret = flag.str_to_flag(str7.str());
  ASSERT_EQ(OB_ERR_UNEXPECTED, ret);

  LOG_INFO("test", K(flag));

  ret = flag.str_to_flag(str8.str());
  ASSERT_EQ(OB_ERR_UNEXPECTED, ret);

  ret = flag.str_to_flag(str9.str());
  ASSERT_EQ(OB_ERR_UNEXPECTED, ret);


}

}
}
int main(int argc, char **argv)
{
  oceanbase::common::ObLogger::get_logger().set_log_level("INFO");
  ::testing::InitGoogleTest(&argc, argv);
  return RUN_ALL_TESTS();
}
#undef private