Skip to content

Commit

Permalink
reset engine_test
Browse files Browse the repository at this point in the history
  • Loading branch information
MStreet3 committed Mar 4, 2025
1 parent f30ee72 commit ad6dcae
Showing 1 changed file with 0 additions and 56 deletions.
56 changes: 0 additions & 56 deletions core/services/workflows/engine_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -570,62 +570,6 @@ func TestEngine_RateLimit(t *testing.T) {
}
})

t.Run("per workflow ID rate limit", func(t *testing.T) {
ctx := testutils.Context(t)
reg := coreCap.NewRegistry(logger.TestLogger(t))

trigger, _ := mockTrigger(t)
require.NoError(t, reg.Add(ctx, trigger))
require.NoError(t, reg.Add(ctx, mockConsensus("")))
target1 := mockTarget("")
require.NoError(t, reg.Add(ctx, target1))

target2 := newMockCapability(
capabilities.MustNewCapabilityInfo(
"[email protected]",
capabilities.CapabilityTypeTarget,
"a write capability targeting ethereum sepolia testnet",
),
func(req capabilities.CapabilityRequest) (capabilities.CapabilityResponse, error) {
m := req.Inputs.Underlying["report"].(*values.Map)
return capabilities.CapabilityResponse{
Value: m,
}, nil
},
)
require.NoError(t, reg.Add(ctx, target2))

setRateLimiter := func(c *Config) {
rl, err := ratelimiter.NewRateLimiter(ratelimiter.Config{
GlobalRPS: 1000.0,
GlobalBurst: 1000,
PerSenderRPS: 1.0,
PerSenderBurst: 1,
})
require.NoError(t, err)
c.RateLimiter = rl
}

eng, testHooks := newTestEngineWithYAMLSpec(
t,
reg,
hardcodedWorkflow,
setRateLimiter,
)

// Call RateLimiter once with the test workflow ID, so next execution gets rate limited
senderAllow, globalAllow := eng.ratelimiter.Allow(testWorkflowID)
require.True(t, senderAllow)
require.True(t, globalAllow)
servicetest.Run(t, eng)

select {
case <-testHooks.rateLimited:
case <-ctx.Done():
t.FailNow()
}
})

t.Run("global rate limit", func(t *testing.T) {
ctx := testutils.Context(t)
reg := coreCap.NewRegistry(logger.TestLogger(t))
Expand Down

0 comments on commit ad6dcae

Please sign in to comment.