mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-15 12:27:59 +00:00
156 lines
4.9 KiB
Go
156 lines
4.9 KiB
Go
|
//go:build integration
|
||
|
|
||
|
package handlers_test
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"encoding/json"
|
||
|
"fmt"
|
||
|
"strconv"
|
||
|
"testing"
|
||
|
"time"
|
||
|
|
||
|
"github.com/stretchr/testify/require"
|
||
|
"google.golang.org/protobuf/types/known/durationpb"
|
||
|
"google.golang.org/protobuf/types/known/timestamppb"
|
||
|
|
||
|
"github.com/zitadel/zitadel/internal/repository/quota"
|
||
|
"github.com/zitadel/zitadel/pkg/grpc/admin"
|
||
|
quota_pb "github.com/zitadel/zitadel/pkg/grpc/quota"
|
||
|
"github.com/zitadel/zitadel/pkg/grpc/system"
|
||
|
)
|
||
|
|
||
|
func TestServer_QuotaNotification_Limit(t *testing.T) {
|
||
|
_, instanceID, iamOwnerCtx := Tester.UseIsolatedInstance(CTX, SystemCTX)
|
||
|
amount := 10
|
||
|
percent := 50
|
||
|
percentAmount := amount * percent / 100
|
||
|
|
||
|
_, err := Tester.Client.System.AddQuota(SystemCTX, &system.AddQuotaRequest{
|
||
|
InstanceId: instanceID,
|
||
|
Unit: quota_pb.Unit_UNIT_REQUESTS_ALL_AUTHENTICATED,
|
||
|
From: timestamppb.Now(),
|
||
|
ResetInterval: durationpb.New(time.Minute * 5),
|
||
|
Amount: uint64(amount),
|
||
|
Limit: true,
|
||
|
Notifications: []*quota_pb.Notification{
|
||
|
{
|
||
|
Percent: uint32(percent),
|
||
|
Repeat: true,
|
||
|
CallUrl: "http://localhost:8082",
|
||
|
},
|
||
|
{
|
||
|
Percent: 100,
|
||
|
Repeat: true,
|
||
|
CallUrl: "http://localhost:8082",
|
||
|
},
|
||
|
},
|
||
|
})
|
||
|
require.NoError(t, err)
|
||
|
|
||
|
for i := 0; i < percentAmount; i++ {
|
||
|
_, err := Tester.Client.Admin.GetDefaultOrg(iamOwnerCtx, &admin.GetDefaultOrgRequest{})
|
||
|
if err != nil {
|
||
|
require.NoError(t, fmt.Errorf("error in %d call of %d: %f", i, percentAmount, err))
|
||
|
}
|
||
|
}
|
||
|
awaitNotification(t, time.Now(), Tester.QuotaNotificationChan, quota.RequestsAllAuthenticated, percent)
|
||
|
|
||
|
for i := 0; i < (amount - percentAmount); i++ {
|
||
|
_, err := Tester.Client.Admin.GetDefaultOrg(iamOwnerCtx, &admin.GetDefaultOrgRequest{})
|
||
|
require.NoError(t, err)
|
||
|
}
|
||
|
awaitNotification(t, time.Now(), Tester.QuotaNotificationChan, quota.RequestsAllAuthenticated, 100)
|
||
|
|
||
|
_, limitErr := Tester.Client.Admin.GetDefaultOrg(iamOwnerCtx, &admin.GetDefaultOrgRequest{})
|
||
|
require.Error(t, limitErr)
|
||
|
}
|
||
|
|
||
|
func TestServer_QuotaNotification_NoLimit(t *testing.T) {
|
||
|
_, instanceID, iamOwnerCtx := Tester.UseIsolatedInstance(CTX, SystemCTX)
|
||
|
amount := 10
|
||
|
percent := 50
|
||
|
percentAmount := amount * percent / 100
|
||
|
|
||
|
_, err := Tester.Client.System.AddQuota(SystemCTX, &system.AddQuotaRequest{
|
||
|
InstanceId: instanceID,
|
||
|
Unit: quota_pb.Unit_UNIT_REQUESTS_ALL_AUTHENTICATED,
|
||
|
From: timestamppb.Now(),
|
||
|
ResetInterval: durationpb.New(time.Minute * 5),
|
||
|
Amount: uint64(amount),
|
||
|
Limit: false,
|
||
|
Notifications: []*quota_pb.Notification{
|
||
|
{
|
||
|
Percent: uint32(percent),
|
||
|
Repeat: false,
|
||
|
CallUrl: "http://localhost:8082",
|
||
|
},
|
||
|
{
|
||
|
Percent: 100,
|
||
|
Repeat: true,
|
||
|
CallUrl: "http://localhost:8082",
|
||
|
},
|
||
|
},
|
||
|
})
|
||
|
require.NoError(t, err)
|
||
|
|
||
|
for i := 0; i < percentAmount; i++ {
|
||
|
_, err := Tester.Client.Admin.GetDefaultOrg(iamOwnerCtx, &admin.GetDefaultOrgRequest{})
|
||
|
if err != nil {
|
||
|
require.NoError(t, fmt.Errorf("error in %d call of %d: %f", i, percentAmount, err))
|
||
|
}
|
||
|
}
|
||
|
awaitNotification(t, time.Now(), Tester.QuotaNotificationChan, quota.RequestsAllAuthenticated, percent)
|
||
|
|
||
|
for i := 0; i < (amount - percentAmount); i++ {
|
||
|
_, err := Tester.Client.Admin.GetDefaultOrg(iamOwnerCtx, &admin.GetDefaultOrgRequest{})
|
||
|
if err != nil {
|
||
|
require.NoError(t, fmt.Errorf("error in %d call of %d: %f", percentAmount+i, amount, err))
|
||
|
}
|
||
|
}
|
||
|
awaitNotification(t, time.Now(), Tester.QuotaNotificationChan, quota.RequestsAllAuthenticated, 100)
|
||
|
|
||
|
for i := 0; i < amount; i++ {
|
||
|
_, err := Tester.Client.Admin.GetDefaultOrg(iamOwnerCtx, &admin.GetDefaultOrgRequest{})
|
||
|
if err != nil {
|
||
|
require.NoError(t, fmt.Errorf("error in %d call of %d over limit: %f", i, amount, err))
|
||
|
}
|
||
|
}
|
||
|
awaitNotification(t, time.Now(), Tester.QuotaNotificationChan, quota.RequestsAllAuthenticated, 200)
|
||
|
|
||
|
_, limitErr := Tester.Client.Admin.GetDefaultOrg(iamOwnerCtx, &admin.GetDefaultOrgRequest{})
|
||
|
require.NoError(t, limitErr)
|
||
|
}
|
||
|
|
||
|
func awaitNotification(t *testing.T, start time.Time, bodies chan []byte, unit quota.Unit, percent int) {
|
||
|
for {
|
||
|
select {
|
||
|
case body := <-bodies:
|
||
|
plain := new(bytes.Buffer)
|
||
|
if err := json.Indent(plain, body, "", " "); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
t.Log("received notificationDueEvent", plain.String())
|
||
|
event := struct {
|
||
|
Unit quota.Unit `json:"unit"`
|
||
|
ID string `json:"id"`
|
||
|
CallURL string `json:"callURL"`
|
||
|
PeriodStart time.Time `json:"periodStart"`
|
||
|
Threshold uint16 `json:"threshold"`
|
||
|
Usage uint64 `json:"usage"`
|
||
|
}{}
|
||
|
if err := json.Unmarshal(body, &event); err != nil {
|
||
|
t.Error(err)
|
||
|
}
|
||
|
if event.ID == "" {
|
||
|
continue
|
||
|
}
|
||
|
if event.Unit == unit && event.Threshold == uint16(percent) {
|
||
|
return
|
||
|
}
|
||
|
case <-time.After(20 * time.Second):
|
||
|
t.Fatalf("start %s stop %s timed out waiting for unit %s and percent %d", start.Format(time.RFC3339), time.Now().Format(time.RFC3339), strconv.Itoa(int(unit)), percent)
|
||
|
}
|
||
|
}
|
||
|
}
|