mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-11-30 02:38:17 +08:00
311 lines
12 KiB
Go
311 lines
12 KiB
Go
package handler
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/go-playground/assert/v2"
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/goodrain/rainbond/api/model"
|
|
"github.com/goodrain/rainbond/db"
|
|
daomock "github.com/goodrain/rainbond/db/dao"
|
|
dbmodel "github.com/goodrain/rainbond/db/model"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func TestAddAppConfigGroup(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
appID string
|
|
request *model.ApplicationConfigGroup
|
|
mockFunc func(manager *db.MockManager, ctrl *gomock.Controller)
|
|
wanterr bool
|
|
}{
|
|
{
|
|
name: "add config group success",
|
|
appID: "appID1",
|
|
request: &model.ApplicationConfigGroup{
|
|
ConfigGroupName: "configName1",
|
|
DeployType: "env",
|
|
ServiceIDs: []string{"sid1"},
|
|
ConfigItems: []model.ConfigItem{
|
|
{ItemKey: "key1", ItemValue: "value1"},
|
|
{ItemKey: "key2", ItemValue: "value2"},
|
|
},
|
|
},
|
|
mockFunc: func(manager *db.MockManager, ctrl *gomock.Controller) {
|
|
serviceResult := []*dbmodel.TenantServices{
|
|
{ServiceID: "sid1", ServiceAlias: "sid1_name"},
|
|
}
|
|
config := &dbmodel.ApplicationConfigGroup{
|
|
AppID: "appID1",
|
|
ConfigGroupName: "configName1",
|
|
DeployType: "env",
|
|
}
|
|
tenantServiceDao := daomock.NewMockTenantServiceDao(ctrl)
|
|
tenantServiceDao.EXPECT().GetServicesByServiceIDs(gomock.Any()).Return(serviceResult, nil)
|
|
manager.EXPECT().TenantServiceDao().Return(tenantServiceDao)
|
|
|
|
serviceConfigGroupDao := daomock.NewMockAppConfigGroupServiceDao(ctrl)
|
|
serviceConfigGroupDao.EXPECT().AddModel(gomock.Any()).Return(nil).AnyTimes()
|
|
manager.EXPECT().AppConfigGroupServiceDao().Return(serviceConfigGroupDao).AnyTimes()
|
|
|
|
configItemDao := daomock.NewMockAppConfigGroupItemDao(ctrl)
|
|
configItemDao.EXPECT().AddModel(gomock.Any()).Return(nil).AnyTimes()
|
|
manager.EXPECT().AppConfigGroupItemDao().Return(configItemDao).AnyTimes()
|
|
|
|
applicationConfigDao := daomock.NewMockAppConfigGroupDao(ctrl)
|
|
applicationConfigDao.EXPECT().AddModel(gomock.Any()).Return(nil)
|
|
applicationConfigDao.EXPECT().GetConfigGroupByID(gomock.Any(), gomock.Any()).Return(config, nil)
|
|
manager.EXPECT().AppConfigGroupDao().Return(applicationConfigDao).AnyTimes()
|
|
},
|
|
wanterr: false,
|
|
},
|
|
{
|
|
name: "add config group service failed",
|
|
appID: "appID1",
|
|
request: &model.ApplicationConfigGroup{
|
|
ConfigGroupName: "configName1",
|
|
DeployType: "env",
|
|
ServiceIDs: []string{"sid1"},
|
|
ConfigItems: []model.ConfigItem{
|
|
{ItemKey: "key1", ItemValue: "value1"},
|
|
{ItemKey: "key2", ItemValue: "value2"},
|
|
},
|
|
},
|
|
mockFunc: func(manager *db.MockManager, ctrl *gomock.Controller) {
|
|
serviceResult := []*dbmodel.TenantServices{
|
|
{ServiceID: "sid1", ServiceAlias: "sid1_name"},
|
|
}
|
|
tenantServiceDao := daomock.NewMockTenantServiceDao(ctrl)
|
|
tenantServiceDao.EXPECT().GetServicesByServiceIDs(gomock.Any()).Return(serviceResult, nil)
|
|
manager.EXPECT().TenantServiceDao().Return(tenantServiceDao)
|
|
|
|
serviceConfigGroupDao := daomock.NewMockAppConfigGroupServiceDao(ctrl)
|
|
serviceConfigGroupDao.EXPECT().AddModel(gomock.Any()).Return(errors.New("add service config failed")).AnyTimes()
|
|
manager.EXPECT().AppConfigGroupServiceDao().Return(serviceConfigGroupDao).AnyTimes()
|
|
},
|
|
wanterr: true,
|
|
},
|
|
{
|
|
name: "add config item failed",
|
|
appID: "appID1",
|
|
request: &model.ApplicationConfigGroup{
|
|
ConfigGroupName: "configName1",
|
|
DeployType: "env",
|
|
ServiceIDs: []string{"sid1"},
|
|
ConfigItems: []model.ConfigItem{
|
|
{ItemKey: "key1", ItemValue: "value1"},
|
|
{ItemKey: "key2", ItemValue: "value2"},
|
|
},
|
|
},
|
|
mockFunc: func(manager *db.MockManager, ctrl *gomock.Controller) {
|
|
serviceResult := []*dbmodel.TenantServices{
|
|
{ServiceID: "sid1", ServiceAlias: "sid1_name"},
|
|
}
|
|
tenantServiceDao := daomock.NewMockTenantServiceDao(ctrl)
|
|
tenantServiceDao.EXPECT().GetServicesByServiceIDs(gomock.Any()).Return(serviceResult, nil)
|
|
manager.EXPECT().TenantServiceDao().Return(tenantServiceDao)
|
|
|
|
serviceConfigGroupDao := daomock.NewMockAppConfigGroupServiceDao(ctrl)
|
|
serviceConfigGroupDao.EXPECT().AddModel(gomock.Any()).Return(nil).AnyTimes()
|
|
manager.EXPECT().AppConfigGroupServiceDao().Return(serviceConfigGroupDao).AnyTimes()
|
|
|
|
configItemDao := daomock.NewMockAppConfigGroupItemDao(ctrl)
|
|
configItemDao.EXPECT().AddModel(gomock.Any()).Return(errors.New("add config item failed")).AnyTimes()
|
|
manager.EXPECT().AppConfigGroupItemDao().Return(configItemDao).AnyTimes()
|
|
},
|
|
wanterr: true,
|
|
},
|
|
{
|
|
name: "add application config group failed",
|
|
appID: "appID1",
|
|
request: &model.ApplicationConfigGroup{
|
|
ConfigGroupName: "configName1",
|
|
DeployType: "env",
|
|
ServiceIDs: []string{"sid1"},
|
|
ConfigItems: []model.ConfigItem{
|
|
{ItemKey: "key1", ItemValue: "value1"},
|
|
{ItemKey: "key2", ItemValue: "value2"},
|
|
},
|
|
},
|
|
mockFunc: func(manager *db.MockManager, ctrl *gomock.Controller) {
|
|
serviceResult := []*dbmodel.TenantServices{
|
|
{ServiceID: "sid1", ServiceAlias: "sid1_name"},
|
|
}
|
|
tenantServiceDao := daomock.NewMockTenantServiceDao(ctrl)
|
|
tenantServiceDao.EXPECT().GetServicesByServiceIDs(gomock.Any()).Return(serviceResult, nil)
|
|
manager.EXPECT().TenantServiceDao().Return(tenantServiceDao)
|
|
|
|
serviceConfigGroupDao := daomock.NewMockAppConfigGroupServiceDao(ctrl)
|
|
serviceConfigGroupDao.EXPECT().AddModel(gomock.Any()).Return(nil).AnyTimes()
|
|
manager.EXPECT().AppConfigGroupServiceDao().Return(serviceConfigGroupDao).AnyTimes()
|
|
|
|
configItemDao := daomock.NewMockAppConfigGroupItemDao(ctrl)
|
|
configItemDao.EXPECT().AddModel(gomock.Any()).Return(nil).AnyTimes()
|
|
manager.EXPECT().AppConfigGroupItemDao().Return(configItemDao).AnyTimes()
|
|
|
|
applicationConfigDao := daomock.NewMockAppConfigGroupDao(ctrl)
|
|
applicationConfigDao.EXPECT().AddModel(gomock.Any()).Return(errors.New("add application config group failed"))
|
|
manager.EXPECT().AppConfigGroupDao().Return(applicationConfigDao).AnyTimes()
|
|
},
|
|
wanterr: true,
|
|
},
|
|
}
|
|
for i := range tests {
|
|
tc := tests[i]
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
|
|
manager := db.NewMockManager(ctrl)
|
|
db.SetTestManager(manager)
|
|
tc.mockFunc(manager, ctrl)
|
|
|
|
appAction := ApplicationAction{}
|
|
resp, err := appAction.AddConfigGroup(tc.appID, tc.request)
|
|
if (err != nil) != tc.wanterr {
|
|
t.Errorf("Unexpected error = %v, wantErr %v", err, tc.wanterr)
|
|
return
|
|
}
|
|
if resp != nil {
|
|
assert.Equal(t, resp.AppID, tc.appID)
|
|
assert.Equal(t, resp.DeployType, tc.request.DeployType)
|
|
}
|
|
})
|
|
}
|
|
|
|
}
|
|
|
|
func TestListConfigGroups(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
appID string
|
|
request *model.ApplicationConfigGroup
|
|
mockFunc func(manager *db.MockManager, ctrl *gomock.Controller)
|
|
wanterr bool
|
|
}{
|
|
{
|
|
name: "list config group success",
|
|
appID: "appID1",
|
|
request: &model.ApplicationConfigGroup{
|
|
ConfigGroupName: "configName1",
|
|
DeployType: "env",
|
|
ServiceIDs: []string{"sid1"},
|
|
ConfigItems: []model.ConfigItem{
|
|
{ItemKey: "key1", ItemValue: "value1"},
|
|
{ItemKey: "key2", ItemValue: "value2"},
|
|
},
|
|
},
|
|
mockFunc: func(manager *db.MockManager, ctrl *gomock.Controller) {
|
|
configGroupsServiceResult := []*dbmodel.ConfigGroupService{
|
|
{ServiceID: "sid1", ServiceAlias: "sid1_name"},
|
|
}
|
|
configGroupsResult := []*dbmodel.ApplicationConfigGroup{
|
|
{AppID: "appID1", ConfigGroupName: "configName1", DeployType: "env"},
|
|
}
|
|
configGroupItemResult := []*dbmodel.ConfigGroupItem{
|
|
{ItemKey: "itemKey1", ItemValue: "itemValue1"},
|
|
}
|
|
applicationConfigDao := daomock.NewMockAppConfigGroupDao(ctrl)
|
|
applicationConfigDao.EXPECT().GetConfigGroupsByAppID(gomock.Any(), gomock.Any(), gomock.Any()).Return(configGroupsResult, int64(1), nil)
|
|
manager.EXPECT().AppConfigGroupDao().Return(applicationConfigDao).AnyTimes()
|
|
|
|
serviceConfigGroupDao := daomock.NewMockAppConfigGroupServiceDao(ctrl)
|
|
serviceConfigGroupDao.EXPECT().GetConfigGroupServicesByID(gomock.Any(), gomock.Any()).Return(configGroupsServiceResult, nil).AnyTimes()
|
|
manager.EXPECT().AppConfigGroupServiceDao().Return(serviceConfigGroupDao).AnyTimes()
|
|
|
|
configItemDao := daomock.NewMockAppConfigGroupItemDao(ctrl)
|
|
configItemDao.EXPECT().GetConfigGroupItemsByID(gomock.Any(), gomock.Any()).Return(configGroupItemResult, nil).AnyTimes()
|
|
manager.EXPECT().AppConfigGroupItemDao().Return(configItemDao).AnyTimes()
|
|
},
|
|
wanterr: false,
|
|
},
|
|
{
|
|
name: "list config group failed because get config group error",
|
|
appID: "appID1",
|
|
mockFunc: func(manager *db.MockManager, ctrl *gomock.Controller) {
|
|
applicationConfigDao := daomock.NewMockAppConfigGroupDao(ctrl)
|
|
applicationConfigDao.EXPECT().GetConfigGroupsByAppID(gomock.Any(), gomock.Any(), gomock.Any()).Return(nil, int64(0), errors.New("get config group error"))
|
|
manager.EXPECT().AppConfigGroupDao().Return(applicationConfigDao).AnyTimes()
|
|
},
|
|
wanterr: true,
|
|
},
|
|
{
|
|
name: "list config group failed because get config group service error",
|
|
appID: "appID1",
|
|
mockFunc: func(manager *db.MockManager, ctrl *gomock.Controller) {
|
|
configGroupsResult := []*dbmodel.ApplicationConfigGroup{
|
|
{AppID: "appID1", ConfigGroupName: "configName1", DeployType: "env"},
|
|
}
|
|
applicationConfigDao := daomock.NewMockAppConfigGroupDao(ctrl)
|
|
applicationConfigDao.EXPECT().GetConfigGroupsByAppID(gomock.Any(), gomock.Any(), gomock.Any()).Return(configGroupsResult, int64(1), nil)
|
|
manager.EXPECT().AppConfigGroupDao().Return(applicationConfigDao).AnyTimes()
|
|
|
|
serviceConfigGroupDao := daomock.NewMockAppConfigGroupServiceDao(ctrl)
|
|
serviceConfigGroupDao.EXPECT().GetConfigGroupServicesByID(gomock.Any(), gomock.Any()).Return(nil, errors.New("get config group service error")).AnyTimes()
|
|
manager.EXPECT().AppConfigGroupServiceDao().Return(serviceConfigGroupDao).AnyTimes()
|
|
},
|
|
wanterr: true,
|
|
},
|
|
{
|
|
name: "list config group failed because get config group item error",
|
|
appID: "appID1",
|
|
mockFunc: func(manager *db.MockManager, ctrl *gomock.Controller) {
|
|
configGroupsServiceResult := []*dbmodel.ConfigGroupService{
|
|
{ServiceID: "sid1", ServiceAlias: "sid1_name"},
|
|
}
|
|
configGroupsResult := []*dbmodel.ApplicationConfigGroup{
|
|
{AppID: "appID1", ConfigGroupName: "configName1", DeployType: "env"},
|
|
}
|
|
applicationConfigDao := daomock.NewMockAppConfigGroupDao(ctrl)
|
|
applicationConfigDao.EXPECT().GetConfigGroupsByAppID(gomock.Any(), gomock.Any(), gomock.Any()).Return(configGroupsResult, int64(1), nil)
|
|
manager.EXPECT().AppConfigGroupDao().Return(applicationConfigDao).AnyTimes()
|
|
|
|
serviceConfigGroupDao := daomock.NewMockAppConfigGroupServiceDao(ctrl)
|
|
serviceConfigGroupDao.EXPECT().GetConfigGroupServicesByID(gomock.Any(), gomock.Any()).Return(configGroupsServiceResult, nil).AnyTimes()
|
|
manager.EXPECT().AppConfigGroupServiceDao().Return(serviceConfigGroupDao).AnyTimes()
|
|
|
|
configItemDao := daomock.NewMockAppConfigGroupItemDao(ctrl)
|
|
configItemDao.EXPECT().GetConfigGroupItemsByID(gomock.Any(), gomock.Any()).Return(nil, errors.New("get config group item error")).AnyTimes()
|
|
manager.EXPECT().AppConfigGroupItemDao().Return(configItemDao).AnyTimes()
|
|
},
|
|
wanterr: true,
|
|
},
|
|
}
|
|
for i := range tests {
|
|
tc := tests[i]
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
|
|
manager := db.NewMockManager(ctrl)
|
|
db.SetTestManager(manager)
|
|
tc.mockFunc(manager, ctrl)
|
|
|
|
appAction := NewApplicationHandler(nil, nil, nil, nil)
|
|
resp, err := appAction.ListConfigGroups(tc.appID, 1, 10)
|
|
if (err != nil) != tc.wanterr {
|
|
t.Errorf("Unexpected error = %v, wantErr %v", err, tc.wanterr)
|
|
return
|
|
}
|
|
if resp != nil {
|
|
for _, r := range resp.ConfigGroup {
|
|
assert.Equal(t, r.AppID, tc.appID)
|
|
assert.Equal(t, r.ConfigGroupName, tc.request.ConfigGroupName)
|
|
assert.Equal(t, r.DeployType, tc.request.DeployType)
|
|
got := false
|
|
for _, s := range r.Services {
|
|
if s.ServiceID == tc.request.ServiceIDs[0] {
|
|
got = true
|
|
}
|
|
}
|
|
if got != true {
|
|
t.Errorf("get config group service error,serviceID not exists")
|
|
}
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
}
|