feat: optimized api impl & change idl
This commit is contained in:
lijunwen.gigoo
@ -494,7 +494,7 @@ func (t *PluginApplicationService) GetProductCallInfo(ctx context.Context, req *
|
||||
}
|
||||
|
||||
// Call GetSaasUserInfo
|
||||
_, err = t.userSVC.GetSaasUserInfo(ctx, *userID)
|
||||
_, err = t.userSVC.GetSaasUserInfo(ctx)
|
||||
if err != nil {
|
||||
logs.CtxErrorf(ctx, "GetSaasUserInfo failed: %v", err)
|
||||
return &productAPI.GetProductCallInfoResponse{
|
||||
@ -504,7 +504,7 @@ func (t *PluginApplicationService) GetProductCallInfo(ctx context.Context, req *
|
||||
}
|
||||
|
||||
// Call GetUserBenefit
|
||||
_, err = t.userSVC.GetUserBenefit(ctx, *userID)
|
||||
benefit, err := t.userSVC.GetUserBenefit(ctx)
|
||||
if err != nil {
|
||||
logs.CtxErrorf(ctx, "GetUserBenefit failed: %v", err)
|
||||
return &productAPI.GetProductCallInfoResponse{
|
||||
@ -515,13 +515,15 @@ func (t *PluginApplicationService) GetProductCallInfo(ctx context.Context, req *
|
||||
|
||||
// Build response data
|
||||
data := &productAPI.GetProductCallInfoData{
|
||||
McpJSON: "", // TODO: Set appropriate MCP JSON based on requirements
|
||||
UserLevel: productAPI.UserLevel_Free, // TODO: Map from userBenefit to appropriate UserLevel
|
||||
UserLevel: productAPI.UserLevel_Free,
|
||||
}
|
||||
|
||||
// TODO: Set CallCountLimit and CallRateLimit based on userBenefit
|
||||
// data.CallCountLimit = &productAPI.ProductCallCountLimit{...}
|
||||
// data.CallRateLimit = &productAPI.ProductCallRateLimit{...}
|
||||
|
||||
data.CallCountLimit = &productAPI.ProductCallCountLimit{
|
||||
IsUnlimited: benefit.IsUnlimited,
|
||||
UsedCount: benefit.UsedCount,
|
||||
TotalCount: benefit.TotalCount,
|
||||
}
|
||||
|
||||
return &productAPI.GetProductCallInfoResponse{
|
||||
Code: 0,
|
||||
|
||||
262
backend/application/plugin/plugin_test.go
Normal file
262
backend/application/plugin/plugin_test.go
Normal file
@ -0,0 +1,262 @@
|
||||
/*
|
||||
* Copyright 2025 coze-dev Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package plugin
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"go.uber.org/mock/gomock"
|
||||
|
||||
productAPI "github.com/coze-dev/coze-studio/backend/api/model/marketplace/product_public_api"
|
||||
pluginCommon "github.com/coze-dev/coze-studio/backend/api/model/plugin_develop/common"
|
||||
"github.com/coze-dev/coze-studio/backend/crossdomain/contract/plugin/consts"
|
||||
"github.com/coze-dev/coze-studio/backend/crossdomain/contract/plugin/model"
|
||||
"github.com/coze-dev/coze-studio/backend/domain/plugin/dto"
|
||||
"github.com/coze-dev/coze-studio/backend/domain/plugin/entity"
|
||||
mockPlugin "github.com/coze-dev/coze-studio/backend/internal/mock/domain/plugin"
|
||||
"github.com/coze-dev/coze-studio/backend/pkg/lang/ptr"
|
||||
)
|
||||
|
||||
func TestPluginApplicationService_GetCozeSaasPluginList(t *testing.T) {
|
||||
ctrl := gomock.NewController(t)
|
||||
defer ctrl.Finish()
|
||||
|
||||
mockDomainSVC := mockPlugin.NewMockPluginService(ctrl)
|
||||
|
||||
service := &PluginApplicationService{
|
||||
DomainSVC: mockDomainSVC,
|
||||
}
|
||||
|
||||
ctx := context.Background()
|
||||
req := &productAPI.GetProductListRequest{}
|
||||
|
||||
t.Run("Success - Normal case with plugins", func(t *testing.T) {
|
||||
// Prepare test data
|
||||
testPlugins := []*entity.PluginInfo{
|
||||
createTestPluginInfo(1, "Test Plugin 1", "Description 1"),
|
||||
createTestPluginInfo(2, "Test Plugin 2", "Description 2"),
|
||||
}
|
||||
|
||||
domainResp := &dto.ListPluginProductsResponse{
|
||||
Plugins: testPlugins,
|
||||
Total: 2,
|
||||
}
|
||||
|
||||
// Setup mock expectations
|
||||
mockDomainSVC.EXPECT().
|
||||
ListSaasPluginProducts(ctx, gomock.Any()).
|
||||
Return(domainResp, nil).
|
||||
Times(1)
|
||||
|
||||
// Execute the method
|
||||
resp, err := service.GetCozeSaasPluginList(ctx, req)
|
||||
|
||||
// Assertions
|
||||
assert.NoError(t, err)
|
||||
assert.NotNil(t, resp)
|
||||
assert.Equal(t, int32(0), resp.Code)
|
||||
assert.Equal(t, "success", resp.Message)
|
||||
assert.NotNil(t, resp.Data)
|
||||
assert.Equal(t, int32(2), resp.Data.Total)
|
||||
assert.False(t, resp.Data.HasMore)
|
||||
assert.Len(t, resp.Data.Products, 2)
|
||||
|
||||
// Verify product info conversion
|
||||
product1 := resp.Data.Products[0]
|
||||
assert.Equal(t, int64(1), product1.MetaInfo.ID)
|
||||
assert.Equal(t, int64(1), product1.MetaInfo.EntityID)
|
||||
assert.Equal(t, "Test Plugin 1", product1.MetaInfo.Name)
|
||||
assert.Equal(t, "Description 1", product1.MetaInfo.Description)
|
||||
assert.Equal(t, "https://example.com/icon.png", product1.MetaInfo.IconURL)
|
||||
assert.Equal(t, int64(1640995200), product1.MetaInfo.ListedAt)
|
||||
assert.NotNil(t, product1.PluginExtra)
|
||||
assert.False(t, product1.PluginExtra.IsOfficial) // IsOfficial() returns RefProductID != nil, which is nil in our test data
|
||||
})
|
||||
|
||||
t.Run("Error - ListSaasPluginProducts returns error", func(t *testing.T) {
|
||||
// Setup mock to return error
|
||||
expectedError := errors.New("failed to fetch SaaS plugins")
|
||||
mockDomainSVC.EXPECT().
|
||||
ListSaasPluginProducts(ctx, gomock.Any()).
|
||||
Return(nil, expectedError).
|
||||
Times(1)
|
||||
|
||||
// Execute the method
|
||||
resp, err := service.GetCozeSaasPluginList(ctx, req)
|
||||
|
||||
// Assertions
|
||||
assert.NoError(t, err) // The method doesn't return error, it handles it internally
|
||||
assert.NotNil(t, resp)
|
||||
assert.Equal(t, int32(-1), resp.Code)
|
||||
assert.Equal(t, "Failed to get SaaS plugin list", resp.Message)
|
||||
assert.Nil(t, resp.Data)
|
||||
})
|
||||
|
||||
t.Run("Success - Empty plugin list", func(t *testing.T) {
|
||||
// Prepare empty response
|
||||
domainResp := &dto.ListPluginProductsResponse{
|
||||
Plugins: []*entity.PluginInfo{},
|
||||
Total: 0,
|
||||
}
|
||||
|
||||
// Setup mock expectations
|
||||
mockDomainSVC.EXPECT().
|
||||
ListSaasPluginProducts(ctx, gomock.Any()).
|
||||
Return(domainResp, nil).
|
||||
Times(1)
|
||||
|
||||
// Execute the method
|
||||
resp, err := service.GetCozeSaasPluginList(ctx, req)
|
||||
|
||||
// Assertions
|
||||
assert.NoError(t, err)
|
||||
assert.NotNil(t, resp)
|
||||
assert.Equal(t, int32(0), resp.Code)
|
||||
assert.Equal(t, "success", resp.Message)
|
||||
assert.NotNil(t, resp.Data)
|
||||
assert.Equal(t, int32(0), resp.Data.Total)
|
||||
assert.False(t, resp.Data.HasMore)
|
||||
assert.Len(t, resp.Data.Products, 0)
|
||||
assert.Empty(t, resp.Data.Products)
|
||||
})
|
||||
|
||||
t.Run("Success - Multiple plugins", func(t *testing.T) {
|
||||
// Prepare test data with multiple plugins
|
||||
testPlugins := []*entity.PluginInfo{
|
||||
createTestPluginInfo(100, "Weather Plugin", "Get weather information"),
|
||||
createTestPluginInfo(200, "Translation Plugin", "Translate text between languages"),
|
||||
createTestPluginInfo(300, "Calculator Plugin", "Perform mathematical calculations"),
|
||||
createTestPluginInfo(400, "News Plugin", "Get latest news updates"),
|
||||
createTestPluginInfo(500, "Email Plugin", "Send and manage emails"),
|
||||
}
|
||||
|
||||
domainResp := &dto.ListPluginProductsResponse{
|
||||
Plugins: testPlugins,
|
||||
Total: 5,
|
||||
}
|
||||
|
||||
// Setup mock expectations
|
||||
mockDomainSVC.EXPECT().
|
||||
ListSaasPluginProducts(ctx, gomock.Any()).
|
||||
Return(domainResp, nil).
|
||||
Times(1)
|
||||
|
||||
// Execute the method
|
||||
resp, err := service.GetCozeSaasPluginList(ctx, req)
|
||||
|
||||
// Assertions
|
||||
assert.NoError(t, err)
|
||||
assert.NotNil(t, resp)
|
||||
assert.Equal(t, int32(0), resp.Code)
|
||||
assert.Equal(t, "success", resp.Message)
|
||||
assert.NotNil(t, resp.Data)
|
||||
assert.Equal(t, int32(5), resp.Data.Total)
|
||||
assert.False(t, resp.Data.HasMore)
|
||||
assert.Len(t, resp.Data.Products, 5)
|
||||
|
||||
// Verify all plugins are converted correctly
|
||||
expectedNames := []string{"Weather Plugin", "Translation Plugin", "Calculator Plugin", "News Plugin", "Email Plugin"}
|
||||
expectedIDs := []int64{100, 200, 300, 400, 500}
|
||||
|
||||
for i, product := range resp.Data.Products {
|
||||
assert.Equal(t, expectedIDs[i], product.MetaInfo.ID)
|
||||
assert.Equal(t, expectedIDs[i], product.MetaInfo.EntityID)
|
||||
assert.Equal(t, expectedNames[i], product.MetaInfo.Name)
|
||||
assert.Equal(t, "https://example.com/icon.png", product.MetaInfo.IconURL)
|
||||
assert.Equal(t, int64(1640995200), product.MetaInfo.ListedAt)
|
||||
assert.NotNil(t, product.PluginExtra)
|
||||
assert.False(t, product.PluginExtra.IsOfficial) // IsOfficial() returns RefProductID != nil, which is nil in our test data
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// createTestPluginInfo creates a test PluginInfo entity for testing
|
||||
func createTestPluginInfo(id int64, name, desc string) *entity.PluginInfo {
|
||||
manifest := &model.PluginManifest{
|
||||
SchemaVersion: "v1",
|
||||
NameForModel: name,
|
||||
NameForHuman: name,
|
||||
DescriptionForModel: desc,
|
||||
DescriptionForHuman: desc,
|
||||
LogoURL: "https://example.com/icon.png",
|
||||
Auth: &model.AuthV2{
|
||||
Type: consts.AuthzTypeOfNone,
|
||||
},
|
||||
API: model.APIDesc{
|
||||
Type: "openapi",
|
||||
},
|
||||
}
|
||||
|
||||
pluginInfo := &model.PluginInfo{
|
||||
ID: id,
|
||||
PluginType: pluginCommon.PluginType_PLUGIN,
|
||||
SpaceID: 0,
|
||||
DeveloperID: 0,
|
||||
APPID: nil,
|
||||
IconURI: ptr.Of("https://example.com/icon.png"),
|
||||
ServerURL: ptr.Of(""),
|
||||
CreatedAt: 1640995200, // 2022-01-01 00:00:00
|
||||
UpdatedAt: 1640995200,
|
||||
Manifest: manifest,
|
||||
}
|
||||
|
||||
return entity.NewPluginInfo(pluginInfo)
|
||||
}
|
||||
|
||||
// createTestPluginInfoWithCustomIcon creates a test PluginInfo entity with custom icon for testing
|
||||
func createTestPluginInfoWithCustomIcon(id int64, name, desc, iconURL string) *entity.PluginInfo {
|
||||
manifest := &model.PluginManifest{
|
||||
SchemaVersion: "v1",
|
||||
NameForModel: name,
|
||||
NameForHuman: name,
|
||||
DescriptionForModel: desc,
|
||||
DescriptionForHuman: desc,
|
||||
LogoURL: iconURL,
|
||||
Auth: &model.AuthV2{
|
||||
Type: consts.AuthzTypeOfNone,
|
||||
},
|
||||
API: model.APIDesc{
|
||||
Type: "openapi",
|
||||
},
|
||||
}
|
||||
|
||||
pluginInfo := &model.PluginInfo{
|
||||
ID: id,
|
||||
PluginType: pluginCommon.PluginType_PLUGIN,
|
||||
SpaceID: 0,
|
||||
DeveloperID: 0,
|
||||
APPID: nil,
|
||||
IconURI: ptr.Of(iconURL),
|
||||
ServerURL: ptr.Of(""),
|
||||
CreatedAt: 1640995200,
|
||||
UpdatedAt: 1640995200,
|
||||
Manifest: manifest,
|
||||
}
|
||||
|
||||
return entity.NewPluginInfo(pluginInfo)
|
||||
}
|
||||
|
||||
// createTestDomainResponse creates a test domain response for testing
|
||||
func createTestDomainResponse(plugins []*entity.PluginInfo) *dto.ListPluginProductsResponse {
|
||||
return &dto.ListPluginProductsResponse{
|
||||
Plugins: plugins,
|
||||
Total: int64(len(plugins)),
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user