test/Rainbond
1
0
Fork 0
mirror of https://gitee.com/rainbond/Rainbond.git synced 2024-12-02 19:57:42 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity
a303e86dff
Rainbond/api/handler/application_config_group_test.go

311 lines
12 KiB
Go
Raw Normal View History

application config group handler test
2020-09-22 18:18:31 +08:00
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"
modified test method
2020-09-23 11:08:01 +08:00
"github.com/pkg/errors"
application config group handler test
2020-09-22 18:18:31 +08:00
)
modified test method
2020-09-23 11:08:01 +08:00
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) {
modified unit test
2020-09-23 15:57:19 +08:00
serviceResult := []*dbmodel.TenantServices{
{ServiceID: "sid1", ServiceAlias: "sid1_name"},
modified test method
2020-09-23 11:08:01 +08:00
}
config := &dbmodel.ApplicationConfigGroup{
AppID: "appID1",
ConfigGroupName: "configName1",
DeployType: "env",
}
modified service name and some method
2020-09-23 14:48:53 +08:00
tenantServiceDao := daomock.NewMockTenantServiceDao(ctrl)
modified unit test
2020-09-23 15:57:19 +08:00
tenantServiceDao.EXPECT().GetServicesByServiceIDs(gomock.Any()).Return(serviceResult, nil)
modified service name and some method
2020-09-23 14:48:53 +08:00
manager.EXPECT().TenantServiceDao().Return(tenantServiceDao)
modified some method name
2020-09-23 15:30:31 +08:00
serviceConfigGroupDao := daomock.NewMockAppConfigGroupServiceDao(ctrl)
modified test method
2020-09-23 11:08:01 +08:00
serviceConfigGroupDao.EXPECT().AddModel(gomock.Any()).Return(nil).AnyTimes()
modified some method name
2020-09-23 15:30:31 +08:00
manager.EXPECT().AppConfigGroupServiceDao().Return(serviceConfigGroupDao).AnyTimes()
application config group handler test
2020-09-22 18:18:31 +08:00
modified some method name
2020-09-23 15:30:31 +08:00
configItemDao := daomock.NewMockAppConfigGroupItemDao(ctrl)
modified test method
2020-09-23 11:08:01 +08:00
configItemDao.EXPECT().AddModel(gomock.Any()).Return(nil).AnyTimes()
modified some method name
2020-09-23 15:30:31 +08:00
manager.EXPECT().AppConfigGroupItemDao().Return(configItemDao).AnyTimes()
application config group handler test
2020-09-22 18:18:31 +08:00
modified some method name
2020-09-23 15:30:31 +08:00
applicationConfigDao := daomock.NewMockAppConfigGroupDao(ctrl)
modified test method
2020-09-23 11:08:01 +08:00
applicationConfigDao.EXPECT().AddModel(gomock.Any()).Return(nil)
update config group
2020-09-23 21:41:13 +08:00
applicationConfigDao.EXPECT().GetConfigGroupByID(gomock.Any(), gomock.Any()).Return(config, nil)
modified some method name
2020-09-23 15:30:31 +08:00
manager.EXPECT().AppConfigGroupDao().Return(applicationConfigDao).AnyTimes()
modified test method
2020-09-23 11:08:01 +08:00
},
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) {
modified unit test
2020-09-23 15:57:19 +08:00
serviceResult := []*dbmodel.TenantServices{
{ServiceID: "sid1", ServiceAlias: "sid1_name"},
modified service name and some method
2020-09-23 14:48:53 +08:00
}
tenantServiceDao := daomock.NewMockTenantServiceDao(ctrl)
modified unit test
2020-09-23 15:57:19 +08:00
tenantServiceDao.EXPECT().GetServicesByServiceIDs(gomock.Any()).Return(serviceResult, nil)
modified service name and some method
2020-09-23 14:48:53 +08:00
manager.EXPECT().TenantServiceDao().Return(tenantServiceDao)
modified some method name
2020-09-23 15:30:31 +08:00
serviceConfigGroupDao := daomock.NewMockAppConfigGroupServiceDao(ctrl)
modified test method
2020-09-23 11:08:01 +08:00
serviceConfigGroupDao.EXPECT().AddModel(gomock.Any()).Return(errors.New("add service config failed")).AnyTimes()
modified some method name
2020-09-23 15:30:31 +08:00
manager.EXPECT().AppConfigGroupServiceDao().Return(serviceConfigGroupDao).AnyTimes()
modified test method
2020-09-23 11:08:01 +08:00
},
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) {
modified unit test
2020-09-23 15:57:19 +08:00
serviceResult := []*dbmodel.TenantServices{
{ServiceID: "sid1", ServiceAlias: "sid1_name"},
modified service name and some method
2020-09-23 14:48:53 +08:00
}
tenantServiceDao := daomock.NewMockTenantServiceDao(ctrl)
modified unit test
2020-09-23 15:57:19 +08:00
tenantServiceDao.EXPECT().GetServicesByServiceIDs(gomock.Any()).Return(serviceResult, nil)
modified service name and some method
2020-09-23 14:48:53 +08:00
manager.EXPECT().TenantServiceDao().Return(tenantServiceDao)
modified some method name
2020-09-23 15:30:31 +08:00
serviceConfigGroupDao := daomock.NewMockAppConfigGroupServiceDao(ctrl)
modified test method
2020-09-23 11:08:01 +08:00
serviceConfigGroupDao.EXPECT().AddModel(gomock.Any()).Return(nil).AnyTimes()
modified some method name
2020-09-23 15:30:31 +08:00
manager.EXPECT().AppConfigGroupServiceDao().Return(serviceConfigGroupDao).AnyTimes()
application config group handler test
2020-09-22 18:18:31 +08:00
modified some method name
2020-09-23 15:30:31 +08:00
configItemDao := daomock.NewMockAppConfigGroupItemDao(ctrl)
modified test method
2020-09-23 11:08:01 +08:00
configItemDao.EXPECT().AddModel(gomock.Any()).Return(errors.New("add config item failed")).AnyTimes()
modified some method name
2020-09-23 15:30:31 +08:00
manager.EXPECT().AppConfigGroupItemDao().Return(configItemDao).AnyTimes()
modified test method
2020-09-23 11:08:01 +08:00
},
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) {
modified unit test
2020-09-23 15:57:19 +08:00
serviceResult := []*dbmodel.TenantServices{
{ServiceID: "sid1", ServiceAlias: "sid1_name"},
modified service name and some method
2020-09-23 14:48:53 +08:00
}
tenantServiceDao := daomock.NewMockTenantServiceDao(ctrl)
modified unit test
2020-09-23 15:57:19 +08:00
tenantServiceDao.EXPECT().GetServicesByServiceIDs(gomock.Any()).Return(serviceResult, nil)
modified service name and some method
2020-09-23 14:48:53 +08:00
manager.EXPECT().TenantServiceDao().Return(tenantServiceDao)
modified some method name
2020-09-23 15:30:31 +08:00
serviceConfigGroupDao := daomock.NewMockAppConfigGroupServiceDao(ctrl)
modified test method
2020-09-23 11:08:01 +08:00
serviceConfigGroupDao.EXPECT().AddModel(gomock.Any()).Return(nil).AnyTimes()
modified some method name
2020-09-23 15:30:31 +08:00
manager.EXPECT().AppConfigGroupServiceDao().Return(serviceConfigGroupDao).AnyTimes()
application config group handler test
2020-09-22 18:18:31 +08:00
modified some method name
2020-09-23 15:30:31 +08:00
configItemDao := daomock.NewMockAppConfigGroupItemDao(ctrl)
modified test method
2020-09-23 11:08:01 +08:00
configItemDao.EXPECT().AddModel(gomock.Any()).Return(nil).AnyTimes()
modified some method name
2020-09-23 15:30:31 +08:00
manager.EXPECT().AppConfigGroupItemDao().Return(configItemDao).AnyTimes()
application config group handler test
2020-09-22 18:18:31 +08:00
modified some method name
2020-09-23 15:30:31 +08:00
applicationConfigDao := daomock.NewMockAppConfigGroupDao(ctrl)
modified test method
2020-09-23 11:08:01 +08:00
applicationConfigDao.EXPECT().AddModel(gomock.Any()).Return(errors.New("add application config group failed"))
modified some method name
2020-09-23 15:30:31 +08:00
manager.EXPECT().AppConfigGroupDao().Return(applicationConfigDao).AnyTimes()
modified test method
2020-09-23 11:08:01 +08:00
},
wanterr: true,
},
}
for i := range tests {
tc := tests[i]
t.Run(tc.name, func(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
application config group handler test
2020-09-22 18:18:31 +08:00
modified test method
2020-09-23 11:08:01 +08:00
manager := db.NewMockManager(ctrl)
db.SetTestManager(manager)
tc.mockFunc(manager, ctrl)
application config group handler test
2020-09-22 18:18:31 +08:00
new api for getting app status
2020-09-28 16:43:27 +08:00
appAction := ApplicationAction{}
modified test method
2020-09-23 11:08:01 +08:00
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)
}
})
application config group handler test
2020-09-22 18:18:31 +08:00
}
modified test method
2020-09-23 11:08:01 +08:00
application config group handler test
2020-09-22 18:18:31 +08:00
}
added list config group unit test
2020-09-24 14:30:34 +08:00
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) {
modified app_config_group_service model name
2020-11-13 09:42:17 +08:00
configGroupsServiceResult := []*dbmodel.ConfigGroupService{
added list config group unit test
2020-09-24 14:30:34 +08:00
{ServiceID: "sid1", ServiceAlias: "sid1_name"},
}
configGroupsResult := []*dbmodel.ApplicationConfigGroup{
{AppID: "appID1", ConfigGroupName: "configName1", DeployType: "env"},
}
make config group work
2020-10-08 16:10:03 +08:00
configGroupItemResult := []*dbmodel.ConfigGroupItem{
added list config group unit test
2020-09-24 14:30:34 +08:00
{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) {
modified app_config_group_service model name
2020-11-13 09:42:17 +08:00
configGroupsServiceResult := []*dbmodel.ConfigGroupService{
added list config group unit test
2020-09-24 14:30:34 +08:00
{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)
Optimize TCP policy update issues.
2021-02-05 15:54:55 +08:00
appAction := NewApplicationHandler(nil, nil)
added list config group unit test
2020-09-24 14:30:34 +08:00
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")
}
}
}
})
}
}
Reference in New Issue Copy Permalink