mirror of
https://github.com/hanzoai/team.git
synced 2026-09-30 01:57:35 +00:00
zeekay
8feb24ea1f
The server module named github.com/hanzoai/team/server/v8 — a path that resolves to the retired workspace's repository, not this one — while server/public still published github.com/mattermost/mattermost/server/public. Both are github.com/hanzoteam/server/... now, which is where this code actually lives. That second half is what made the binary unbootable. Upstream's public path stayed claimed by upstream, mattermost-plugin-ai imported it, and the build linked public/model twice; public/plugin's gob.Register then ran twice under one name and panicked at init, before main. Reverting the rename traded the brand for a working binary. Rebranding the plugin alongside it needs no such trade: the agents fork is github.com/hanzoteam/agents/v2 against our public module, so no module named mattermost/mattermost is in the graph at all. Measured: `go list -deps ./cmd/mattermost` names exactly one public/model, and it is ours.
147 lines
4.7 KiB
Go
147 lines
4.7 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package awsmeter
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/service/marketplacemetering"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/hanzoteam/server/server/public/model"
|
|
"github.com/hanzoteam/server/server/public/plugin/plugintest/mock"
|
|
"github.com/hanzoteam/server/server/v8/channels/store/storetest/mocks"
|
|
)
|
|
|
|
type mockMarketplaceMeteringClient struct {
|
|
MeteringService
|
|
}
|
|
|
|
func (m *mockMarketplaceMeteringClient) MeterUsage(ctx context.Context, input *marketplacemetering.MeterUsageInput, optFns ...func(*marketplacemetering.Options)) (*marketplacemetering.MeterUsageOutput, error) {
|
|
return &marketplacemetering.MeterUsageOutput{
|
|
MeteringRecordId: new("1"),
|
|
}, nil
|
|
}
|
|
|
|
type mockMarketplaceMeteringClientWithError struct {
|
|
MeteringService
|
|
}
|
|
|
|
func (m *mockMarketplaceMeteringClientWithError) MeterUsage(ctx context.Context, input *marketplacemetering.MeterUsageInput, optFns ...func(*marketplacemetering.Options)) (*marketplacemetering.MeterUsageOutput, error) {
|
|
return nil, errors.New("error")
|
|
}
|
|
|
|
func TestAwsMeterUsage(t *testing.T) {
|
|
startTime := time.Now()
|
|
endTime := time.Now()
|
|
dimensions := []string{model.AwsMeteringDimensionUsageHrs}
|
|
|
|
userStoreMock := mocks.UserStore{}
|
|
userStoreMock.On("AnalyticsActiveCountForPeriod", model.GetMillisForTime(startTime), model.GetMillisForTime(endTime), mock.AnythingOfType("model.UserCountOptions")).Return(int32(2), nil)
|
|
|
|
storeMock := mocks.Store{}
|
|
storeMock.On("User").Return(&userStoreMock)
|
|
|
|
reports := make([]*AWSMeterReport, 1)
|
|
reports[0] = &AWSMeterReport{
|
|
Dimension: model.AwsMeteringDimensionUsageHrs,
|
|
Value: 2,
|
|
Timestamp: startTime,
|
|
}
|
|
|
|
// Define a mock struct to be used in your unit tests of myFunc.
|
|
svc := &AWSMeterService{
|
|
AwsDryRun: false,
|
|
AwsProductCode: "12345",
|
|
AwsMeteringSvc: &mockMarketplaceMeteringClient{},
|
|
}
|
|
|
|
config := &model.Config{}
|
|
config.SetDefaults()
|
|
|
|
awsmeter := &AwsMeter{
|
|
store: &storeMock,
|
|
service: svc,
|
|
config: config,
|
|
}
|
|
|
|
t.Run("Send report for one usage category", func(t *testing.T) {
|
|
resultReports := awsmeter.GetUserCategoryUsage(dimensions, startTime, endTime)
|
|
require.NotNil(t, resultReports)
|
|
assert.Equal(t, 1, len(resultReports))
|
|
assert.Equal(t, reports[0].Dimension, resultReports[0].Dimension)
|
|
assert.Equal(t, reports[0].Value, resultReports[0].Value)
|
|
assert.Equal(t, reports[0].Timestamp, resultReports[0].Timestamp)
|
|
|
|
err := awsmeter.ReportUserCategoryUsage(context.Background(), resultReports)
|
|
require.NoError(t, err)
|
|
})
|
|
|
|
t.Run("Error in AWS service call", func(t *testing.T) {
|
|
awsmeter.service.AwsMeteringSvc = &mockMarketplaceMeteringClientWithError{}
|
|
resultReports := awsmeter.GetUserCategoryUsage(dimensions, startTime, endTime)
|
|
require.NotNil(t, resultReports)
|
|
assert.Equal(t, 1, len(resultReports))
|
|
err := awsmeter.ReportUserCategoryUsage(context.Background(), resultReports)
|
|
require.Error(t, err)
|
|
})
|
|
|
|
t.Run("Invalid dimension", func(t *testing.T) {
|
|
awsmeter.service.AwsMeteringSvc = &mockMarketplaceMeteringClient{}
|
|
dimensions = []string{"invalid dimension"}
|
|
resultReports := awsmeter.GetUserCategoryUsage(dimensions, startTime, endTime)
|
|
require.NotNil(t, resultReports)
|
|
assert.Equal(t, 0, len(resultReports))
|
|
err := awsmeter.ReportUserCategoryUsage(context.Background(), resultReports)
|
|
require.NoError(t, err)
|
|
})
|
|
}
|
|
|
|
func TestAwsMeterUsageWithDBError(t *testing.T) {
|
|
startTime := time.Now()
|
|
endTime := time.Now()
|
|
dimensions := []string{model.AwsMeteringDimensionUsageHrs}
|
|
|
|
userStoreMock := mocks.UserStore{}
|
|
userStoreMock.On("AnalyticsActiveCountForPeriod", model.GetMillisForTime(startTime), model.GetMillisForTime(endTime), mock.AnythingOfType("model.UserCountOptions")).Return(int32(0), errors.New("error"))
|
|
|
|
storeMock := mocks.Store{}
|
|
storeMock.On("User").Return(&userStoreMock)
|
|
|
|
reports := make([]*AWSMeterReport, 1)
|
|
reports[0] = &AWSMeterReport{
|
|
Dimension: model.AwsMeteringDimensionUsageHrs,
|
|
Value: 2,
|
|
Timestamp: startTime,
|
|
}
|
|
|
|
// Define a mock struct to be used in your unit tests of myFunc.
|
|
svc := &AWSMeterService{
|
|
AwsDryRun: false,
|
|
AwsProductCode: "12345",
|
|
AwsMeteringSvc: &mockMarketplaceMeteringClient{},
|
|
}
|
|
|
|
config := &model.Config{}
|
|
config.SetDefaults()
|
|
|
|
awsmeter := &AwsMeter{
|
|
store: &storeMock,
|
|
service: svc,
|
|
config: config,
|
|
}
|
|
|
|
t.Run("Error in DB query", func(t *testing.T) {
|
|
resultReports := awsmeter.GetUserCategoryUsage(dimensions, startTime, endTime)
|
|
require.NotNil(t, resultReports)
|
|
assert.Equal(t, 0, len(resultReports))
|
|
err := awsmeter.ReportUserCategoryUsage(context.Background(), resultReports)
|
|
require.NoError(t, err)
|
|
})
|
|
}
|