mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
Alex ShtinandGitHub
8e8f1b0f7c
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.
## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
122 lines
3.0 KiB
Go
122 lines
3.0 KiB
Go
package calculator
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"go.temporal.io/server/common/log"
|
|
"go.uber.org/mock/gomock"
|
|
)
|
|
|
|
type (
|
|
testCalculator struct {
|
|
quota float64
|
|
}
|
|
|
|
testNamespaceCalculator struct {
|
|
namespaceQuota map[string]float64
|
|
}
|
|
)
|
|
|
|
func TestQuotaLogger(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
controller := gomock.NewController(t)
|
|
|
|
mockLogger := log.NewMockLogger(controller)
|
|
|
|
quotaLogger := newQuotaLogger(mockLogger)
|
|
|
|
mockLogger.EXPECT().Info(gomock.Any(), gomock.Any(), gomock.Any()).Times(1)
|
|
quotaLogger.updateQuota(1.0)
|
|
|
|
quotaLogger.updateQuota(1.0)
|
|
|
|
mockLogger.EXPECT().Info(gomock.Any(), gomock.Any(), gomock.Any()).Times(1)
|
|
quotaLogger.updateQuota(2.0)
|
|
}
|
|
|
|
func TestLoggedCalculator(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
controller := gomock.NewController(t)
|
|
|
|
mockLogger := log.NewMockLogger(controller)
|
|
mockCalculator := &testCalculator{}
|
|
|
|
quota := 1.0
|
|
mockCalculator.updateQuota(quota)
|
|
|
|
loggedCalculator := NewLoggedCalculator(mockCalculator, mockLogger)
|
|
|
|
mockLogger.EXPECT().Info(gomock.Any(), gomock.Any(), gomock.Any()).Times(1)
|
|
actualQuota := loggedCalculator.GetQuota()
|
|
require.Equal(t, quota, actualQuota)
|
|
|
|
actualQuota = loggedCalculator.GetQuota()
|
|
require.Equal(t, quota, actualQuota)
|
|
|
|
quota = 2.0
|
|
mockCalculator.updateQuota(quota)
|
|
|
|
mockLogger.EXPECT().Info(gomock.Any(), gomock.Any(), gomock.Any()).Times(1)
|
|
actualQuota = loggedCalculator.GetQuota()
|
|
require.Equal(t, quota, actualQuota)
|
|
}
|
|
|
|
func TestLoggedNamespaceCalculator(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
controller := gomock.NewController(t)
|
|
|
|
mockLogger := log.NewMockLogger(controller)
|
|
mockCalculator := &testNamespaceCalculator{}
|
|
|
|
namespace1 := "test-namespace-1"
|
|
quota1 := 1.0
|
|
mockCalculator.updateQuota(namespace1, quota1)
|
|
|
|
namespace2 := "test-namespace-2"
|
|
quota2 := 2.0
|
|
mockCalculator.updateQuota(namespace2, quota2)
|
|
|
|
loggedCalculator := NewLoggedNamespaceCalculator(mockCalculator, mockLogger)
|
|
|
|
mockLogger.EXPECT().Info(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Times(1)
|
|
actualQuota := loggedCalculator.GetQuota(namespace1)
|
|
require.Equal(t, quota1, actualQuota)
|
|
|
|
mockLogger.EXPECT().Info(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Times(1)
|
|
actualQuota = loggedCalculator.GetQuota(namespace2)
|
|
require.Equal(t, quota2, actualQuota)
|
|
|
|
quota2 = 3.0
|
|
mockCalculator.updateQuota(namespace2, quota2)
|
|
|
|
mockLogger.EXPECT().Info(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).Times(1)
|
|
actualQuota = loggedCalculator.GetQuota(namespace2)
|
|
require.Equal(t, quota2, actualQuota)
|
|
|
|
actualQuota = loggedCalculator.GetQuota(namespace1)
|
|
require.Equal(t, quota1, actualQuota)
|
|
}
|
|
|
|
func (c *testCalculator) GetQuota() float64 {
|
|
return c.quota
|
|
}
|
|
|
|
func (c *testCalculator) updateQuota(newQuota float64) {
|
|
c.quota = newQuota
|
|
}
|
|
|
|
func (c *testNamespaceCalculator) GetQuota(namespace string) float64 {
|
|
return c.namespaceQuota[namespace]
|
|
}
|
|
|
|
func (c *testNamespaceCalculator) updateQuota(namespace string, newQuota float64) {
|
|
if c.namespaceQuota == nil {
|
|
c.namespaceQuota = make(map[string]float64)
|
|
}
|
|
c.namespaceQuota[namespace] = newQuota
|
|
}
|