287 lines
9.0 KiB
Go
287 lines
9.0 KiB
Go
/**
|
|
* 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.
|
|
*/
|
|
|
|
package server
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/oceanbase/configserver/config"
|
|
"github.com/oceanbase/configserver/ent"
|
|
)
|
|
|
|
const testRootServiceJson = "{\"Type\":\"PRIMARY\",\"ObClusterId\":1,\"ObRegionId\":1,\"ObCluster\":\"c1\",\"ObRegion\":\"c1\",\"ReadonlyRsList\":[],\"RsList\":[{\"address\":\"1.1.1.1:2882\",\"role\":\"LEADER\",\"sql_port\":2881}],\"timestamp\":1649435362283000}"
|
|
|
|
func TestGetRootServiceInfoParamOldVersion(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
|
|
c.Request, _ = http.NewRequest("GET", "http://1.1.1.1:8080/services?Action=ObRootServiceInfo&ObRegion=c1&ObRegionId=1", nil)
|
|
|
|
obRootServiceInfoParam, err := getCommonParam(c)
|
|
require.Equal(t, "c1", obRootServiceInfoParam.ObCluster)
|
|
require.Equal(t, int64(1), obRootServiceInfoParam.ObClusterId)
|
|
require.Equal(t, 0, obRootServiceInfoParam.Version)
|
|
require.True(t, err == nil)
|
|
|
|
}
|
|
|
|
func TestGetRootServiceInfoParamVersion2(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
|
|
c.Request, _ = http.NewRequest("GET", "http://1.1.1.1:8080/services?Action=ObRootServiceInfo&ObCluster=c1&ObClusterId=1&version=2", nil)
|
|
|
|
obRootServiceInfoParam, err := getCommonParam(c)
|
|
require.Equal(t, "c1", obRootServiceInfoParam.ObCluster)
|
|
require.Equal(t, int64(1), obRootServiceInfoParam.ObClusterId)
|
|
require.Equal(t, 2, obRootServiceInfoParam.Version)
|
|
require.True(t, err == nil)
|
|
|
|
}
|
|
|
|
func TestGetObRootServiceInfo(t *testing.T) {
|
|
// test gin
|
|
gin.SetMode(gin.TestMode)
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Request, _ = http.NewRequest("GET", "http://1.1.1.1:8080/services?Action=ObRootServiceInfo&ObCluster=c1", nil)
|
|
|
|
// mock db client
|
|
client, _ := ent.Open("sqlite3", "file:ent?mode=memory&cache=shared&_fk=1")
|
|
client.Schema.Create(context.Background())
|
|
|
|
configServerConfig, _ := config.ParseConfigServerConfig("../etc/config.yaml")
|
|
configServer = &ConfigServer{
|
|
Config: configServerConfig,
|
|
Client: client,
|
|
}
|
|
|
|
client.ObCluster.
|
|
Create().
|
|
SetName("c1").
|
|
SetObClusterID(1).
|
|
SetType("PRIMARY").
|
|
SetRootserviceJSON(testRootServiceJson).
|
|
OnConflict().
|
|
SetRootserviceJSON(testRootServiceJson).
|
|
Exec(context.Background())
|
|
|
|
response := getObRootServiceInfo(context.Background(), c)
|
|
require.Equal(t, http.StatusOK, response.Code)
|
|
}
|
|
|
|
func TestGetObRootServiceInfoV2(t *testing.T) {
|
|
// test gin
|
|
gin.SetMode(gin.TestMode)
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Request, _ = http.NewRequest("GET", "http://1.1.1.1:8080/services?Action=ObRootServiceInfo&ObCluster=c1&version=2", nil)
|
|
|
|
// mock db client
|
|
client, _ := ent.Open("sqlite3", "file:ent?mode=memory&cache=shared&_fk=1")
|
|
client.Schema.Create(context.Background())
|
|
|
|
configServerConfig, _ := config.ParseConfigServerConfig("../etc/config.yaml")
|
|
configServer = &ConfigServer{
|
|
Config: configServerConfig,
|
|
Client: client,
|
|
}
|
|
client.ObCluster.
|
|
Create().
|
|
SetName("c1").
|
|
SetObClusterID(1).
|
|
SetType("PRIMARY").
|
|
SetRootserviceJSON(testRootServiceJson).
|
|
OnConflict().
|
|
SetRootserviceJSON(testRootServiceJson).
|
|
Exec(context.Background())
|
|
|
|
response := getObRootServiceInfo(context.Background(), c)
|
|
require.Equal(t, http.StatusOK, response.Code)
|
|
}
|
|
|
|
func TestGetObRootServiceInfoV2WithObClusterId(t *testing.T) {
|
|
// test gin
|
|
gin.SetMode(gin.TestMode)
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Request, _ = http.NewRequest("GET", "http://1.1.1.1:8080/services?Action=ObRootServiceInfo&ObCluster=c1&ObClusterId=1&version=2", nil)
|
|
|
|
// mock db client
|
|
client, _ := ent.Open("sqlite3", "file:ent?mode=memory&cache=shared&_fk=1")
|
|
client.Schema.Create(context.Background())
|
|
|
|
configServerConfig, _ := config.ParseConfigServerConfig("../etc/config.yaml")
|
|
configServer = &ConfigServer{
|
|
Config: configServerConfig,
|
|
Client: client,
|
|
}
|
|
|
|
client.ObCluster.
|
|
Create().
|
|
SetName("c1").
|
|
SetObClusterID(1).
|
|
SetType("PRIMARY").
|
|
SetRootserviceJSON(testRootServiceJson).
|
|
OnConflict().
|
|
SetRootserviceJSON(testRootServiceJson).
|
|
Exec(context.Background())
|
|
|
|
response := getObRootServiceInfo(context.Background(), c)
|
|
require.Equal(t, http.StatusOK, response.Code)
|
|
}
|
|
|
|
func TestGetObRootServiceInfoNoResult(t *testing.T) {
|
|
// test gin
|
|
gin.SetMode(gin.TestMode)
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Request, _ = http.NewRequest("GET", "http://1.1.1.1:8080/services?Action=ObRootServiceInfo&ObCluster=c2&ObClusterId=2&version=2", nil)
|
|
|
|
// mock db client
|
|
client, _ := ent.Open("sqlite3", "file:ent?mode=memory&cache=shared&_fk=1")
|
|
client.Schema.Create(context.Background())
|
|
|
|
configServerConfig, _ := config.ParseConfigServerConfig("../etc/config.yaml")
|
|
configServer = &ConfigServer{
|
|
Config: configServerConfig,
|
|
Client: client,
|
|
}
|
|
|
|
response := getObRootServiceInfo(context.Background(), c)
|
|
require.Equal(t, http.StatusNotFound, response.Code)
|
|
}
|
|
|
|
func TestCreateOrUpdateObRootServiceInfo(t *testing.T) {
|
|
// test gin
|
|
gin.SetMode(gin.TestMode)
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Request, _ = http.NewRequest("GET", "http://1.1.1.1:8080/services?Action=ObRootServiceInfo&ObCluster=c1&ObClusterId=1&version=2", bytes.NewBuffer([]byte(testRootServiceJson)))
|
|
|
|
// mock db client
|
|
client, _ := ent.Open("sqlite3", "file:ent?mode=memory&cache=shared&_fk=1")
|
|
client.Schema.Create(context.Background())
|
|
|
|
configServerConfig, _ := config.ParseConfigServerConfig("../etc/config.yaml")
|
|
configServer = &ConfigServer{
|
|
Config: configServerConfig,
|
|
Client: client,
|
|
}
|
|
|
|
response := createOrUpdateObRootServiceInfo(context.Background(), c)
|
|
require.Equal(t, http.StatusOK, response.Code)
|
|
}
|
|
|
|
func TestDeleteObRootServiceInfo(t *testing.T) {
|
|
// test gin
|
|
gin.SetMode(gin.TestMode)
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Request, _ = http.NewRequest("DELETE", "http://1.1.1.1:8080/services?Action=ObRootServiceInfo&ObCluster=c1&ObClusterId=1&version=2", nil)
|
|
|
|
// mock db client
|
|
client, _ := ent.Open("sqlite3", "file:ent?mode=memory&cache=shared&_fk=1")
|
|
client.Schema.Create(context.Background())
|
|
|
|
configServerConfig, _ := config.ParseConfigServerConfig("../etc/config.yaml")
|
|
configServer = &ConfigServer{
|
|
Config: configServerConfig,
|
|
Client: client,
|
|
}
|
|
|
|
client.ObCluster.
|
|
Create().
|
|
SetName("c1").
|
|
SetObClusterID(1).
|
|
SetType("PRIMARY").
|
|
SetRootserviceJSON(testRootServiceJson).
|
|
OnConflict().
|
|
SetRootserviceJSON(testRootServiceJson).
|
|
Exec(context.Background())
|
|
|
|
response := deleteObRootServiceInfo(context.Background(), c)
|
|
require.Equal(t, http.StatusOK, response.Code)
|
|
}
|
|
|
|
func TestDeleteObRootServiceInfoVersion1(t *testing.T) {
|
|
// test gin
|
|
gin.SetMode(gin.TestMode)
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Request, _ = http.NewRequest("DELETE", "http://1.1.1.1:8080/services?Action=ObRootServiceInfo&ObCluster=c1&ObClusterId=1", nil)
|
|
|
|
// mock db client
|
|
client, _ := ent.Open("sqlite3", "file:ent?mode=memory&cache=shared&_fk=1")
|
|
client.Schema.Create(context.Background())
|
|
|
|
configServerConfig, _ := config.ParseConfigServerConfig("../etc/config.yaml")
|
|
configServer = &ConfigServer{
|
|
Config: configServerConfig,
|
|
Client: client,
|
|
}
|
|
|
|
client.ObCluster.
|
|
Create().
|
|
SetName("c1").
|
|
SetObClusterID(1).
|
|
SetType("PRIMARY").
|
|
SetRootserviceJSON(testRootServiceJson).
|
|
OnConflict().
|
|
SetRootserviceJSON(testRootServiceJson).
|
|
Exec(context.Background())
|
|
|
|
response := deleteObRootServiceInfo(context.Background(), c)
|
|
require.Equal(t, http.StatusBadRequest, response.Code)
|
|
}
|
|
|
|
func TestDeleteObRootServiceInfoWithoutClusterId(t *testing.T) {
|
|
// test gin
|
|
gin.SetMode(gin.TestMode)
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Request, _ = http.NewRequest("DELETE", "http://1.1.1.1:8080/services?Action=ObRootServiceInfo&ObCluster=c1&version=2", nil)
|
|
|
|
// mock db client
|
|
client, _ := ent.Open("sqlite3", "file:ent?mode=memory&cache=shared&_fk=1")
|
|
client.Schema.Create(context.Background())
|
|
|
|
configServerConfig, _ := config.ParseConfigServerConfig("../etc/config.yaml")
|
|
configServer = &ConfigServer{
|
|
Config: configServerConfig,
|
|
Client: client,
|
|
}
|
|
|
|
client.ObCluster.
|
|
Create().
|
|
SetName("c1").
|
|
SetObClusterID(1).
|
|
SetType("PRIMARY").
|
|
SetRootserviceJSON(testRootServiceJson).
|
|
OnConflict().
|
|
SetRootserviceJSON(testRootServiceJson).
|
|
Exec(context.Background())
|
|
|
|
response := deleteObRootServiceInfo(context.Background(), c)
|
|
require.Equal(t, http.StatusBadRequest, response.Code)
|
|
}
|