add ob configserver (#943)
This commit is contained in:
chris-sun-star
286
tools/ob-configserver/server/observer_handler_test.go
Normal file
286
tools/ob-configserver/server/observer_handler_test.go
Normal file
@ -0,0 +1,286 @@
|
||||
/**
|
||||
* 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)
|
||||
}
|
||||
Reference in New Issue
Block a user