735 lines
21 KiB
Go
735 lines
21 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package nomad
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/nomad/ci"
|
|
"github.com/hashicorp/nomad/helper/testlog"
|
|
"github.com/hashicorp/nomad/nomad/mock"
|
|
"github.com/hashicorp/nomad/nomad/structs"
|
|
"github.com/hashicorp/nomad/testutil"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func testBlockedEvals(t *testing.T) (*BlockedEvals, *EvalBroker) {
|
|
broker := testBroker(t, 0)
|
|
broker.SetEnabled(true)
|
|
blocked := NewBlockedEvals(broker, testlog.HCLogger(t))
|
|
blocked.SetEnabled(true)
|
|
return blocked, broker
|
|
}
|
|
|
|
func TestBlockedEvals_Block_Disabled(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, _ := testBlockedEvals(t)
|
|
blocked.SetEnabled(false)
|
|
|
|
// Create an escaped eval and add it to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e.EscapedComputedClass = true
|
|
blocked.Block(e)
|
|
|
|
// Verify block did nothing.
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(0, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 0)
|
|
}
|
|
|
|
func TestBlockedEvals_Block_SameJob(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, _ := testBlockedEvals(t)
|
|
|
|
// Create two blocked evals and add them to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e2 := mock.BlockedEval()
|
|
e2.JobID = e.JobID
|
|
blocked.Block(e)
|
|
blocked.Block(e2)
|
|
|
|
// Verify block didn't track duplicate.
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
}
|
|
|
|
func TestBlockedEvals_Block_Quota(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, _ := testBlockedEvals(t)
|
|
|
|
// Create a blocked eval on quota.
|
|
e := mock.BlockedEval()
|
|
e.QuotaLimitReached = "foo"
|
|
blocked.Block(e)
|
|
|
|
// Verify block did track eval.
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Equal(1, blockedStats.TotalQuotaLimit)
|
|
}
|
|
|
|
func TestBlockedEvals_Block_PriorUnblocks(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, _ := testBlockedEvals(t)
|
|
|
|
// Do unblocks prior to blocking.
|
|
blocked.Unblock("v1:123", 1000)
|
|
blocked.Unblock("v1:123", 1001)
|
|
|
|
// Create blocked eval with two classes ineligible.
|
|
e := mock.BlockedEval()
|
|
e.ClassEligibility = map[string]bool{"v1:123": false, "v1:456": false}
|
|
e.SnapshotIndex = 999
|
|
blocked.Block(e)
|
|
|
|
// Verify block did track eval.
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
}
|
|
|
|
func TestBlockedEvals_GetDuplicates(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, _ := testBlockedEvals(t)
|
|
|
|
// Create duplicate blocked evals and add them to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e.CreateIndex = 100
|
|
e2 := mock.BlockedEval()
|
|
e2.JobID = e.JobID
|
|
e2.CreateIndex = 101
|
|
e3 := mock.BlockedEval()
|
|
e3.JobID = e.JobID
|
|
e3.CreateIndex = 102
|
|
e4 := mock.BlockedEval()
|
|
e4.JobID = e.JobID
|
|
e4.CreateIndex = 100
|
|
blocked.Block(e)
|
|
blocked.Block(e2)
|
|
|
|
// Verify stats such that we are only tracking one.
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
|
|
// Get the duplicates.
|
|
out := blocked.GetDuplicates(0)
|
|
require.Len(out, 1)
|
|
require.Equal(e, out[0])
|
|
|
|
// Call block again after a small sleep.
|
|
go func() {
|
|
time.Sleep(500 * time.Millisecond)
|
|
blocked.Block(e3)
|
|
}()
|
|
|
|
// Get the duplicates.
|
|
out = blocked.GetDuplicates(1 * time.Second)
|
|
require.Len(out, 1)
|
|
require.Equal(e2, out[0])
|
|
|
|
// Verify stats such that we are only tracking one.
|
|
blockedStats = blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
|
|
// Add an older evaluation and assert it gets cancelled.
|
|
blocked.Block(e4)
|
|
out = blocked.GetDuplicates(0)
|
|
require.Len(out, 1)
|
|
require.Equal(e4, out[0])
|
|
|
|
// Verify stats such that we are only tracking one.
|
|
blockedStats = blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
}
|
|
|
|
func TestBlockedEvals_UnblockEscaped(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, broker := testBlockedEvals(t)
|
|
|
|
// Create an escaped eval and add it to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e.Status = structs.EvalStatusBlocked
|
|
e.EscapedComputedClass = true
|
|
blocked.Block(e)
|
|
|
|
// Verify block caused the eval to be tracked
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(1, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
|
|
blocked.Unblock("v1:123", 1000)
|
|
requireBlockedEvalsEnqueued(t, blocked, broker, 1)
|
|
}
|
|
|
|
func requireBlockedEvalsEnqueued(t *testing.T, blocked *BlockedEvals, broker *EvalBroker, enqueued int) {
|
|
testutil.WaitForResult(func() (bool, error) {
|
|
// Verify Unblock caused an enqueue
|
|
brokerStats := broker.Stats()
|
|
if brokerStats.TotalReady != enqueued {
|
|
return false, fmt.Errorf("missing enqueued evals: %#v", brokerStats)
|
|
}
|
|
|
|
// Prune old and empty metrics.
|
|
blocked.pruneStats(time.Now().UTC())
|
|
|
|
// Verify Unblock updates the stats
|
|
blockedStats := blocked.Stats()
|
|
ok := blockedStats.TotalBlocked == 0 &&
|
|
blockedStats.TotalEscaped == 0 &&
|
|
len(blockedStats.BlockedResources.ByJob) == 0
|
|
if !ok {
|
|
return false, fmt.Errorf("evals still blocked: %#v", blockedStats)
|
|
}
|
|
return true, nil
|
|
}, func(err error) {
|
|
t.Fatalf("err: %s", err)
|
|
})
|
|
}
|
|
|
|
func TestBlockedEvals_UnblockEligible(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, broker := testBlockedEvals(t)
|
|
|
|
// Create a blocked eval that is eligible on a specific node class and add
|
|
// it to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e.Status = structs.EvalStatusBlocked
|
|
e.ClassEligibility = map[string]bool{"v1:123": true}
|
|
blocked.Block(e)
|
|
|
|
// Verify block caused the eval to be tracked
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
|
|
blocked.Unblock("v1:123", 1000)
|
|
requireBlockedEvalsEnqueued(t, blocked, broker, 1)
|
|
}
|
|
|
|
func TestBlockedEvals_UnblockIneligible(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, broker := testBlockedEvals(t)
|
|
|
|
// Create a blocked eval that is ineligible on a specific node class and add
|
|
// it to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e.ClassEligibility = map[string]bool{"v1:123": false}
|
|
blocked.Block(e)
|
|
|
|
// Verify block caused the eval to be tracked
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
|
|
// Should do nothing
|
|
blocked.Unblock("v1:123", 1000)
|
|
|
|
testutil.WaitForResult(func() (bool, error) {
|
|
// Verify Unblock didn't cause an enqueue
|
|
brokerStats := broker.Stats()
|
|
if brokerStats.TotalReady != 0 {
|
|
return false, fmt.Errorf("eval unblocked: %#v", brokerStats)
|
|
}
|
|
|
|
// Prune old and empty metrics.
|
|
blocked.pruneStats(time.Now().UTC())
|
|
|
|
blockedStats := blocked.Stats()
|
|
ok := blockedStats.TotalBlocked == 1 &&
|
|
blockedStats.TotalEscaped == 0 &&
|
|
len(blockedStats.BlockedResources.ByJob) == 1
|
|
if !ok {
|
|
return false, fmt.Errorf("eval unblocked: %#v", blockedStats)
|
|
}
|
|
return true, nil
|
|
}, func(err error) {
|
|
t.Fatalf("err: %s", err)
|
|
})
|
|
}
|
|
|
|
func TestBlockedEvals_UnblockUnknown(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, broker := testBlockedEvals(t)
|
|
|
|
// Create a blocked eval that is ineligible on a specific node class and add
|
|
// it to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e.ClassEligibility = map[string]bool{"v1:123": true, "v1:456": false}
|
|
blocked.Block(e)
|
|
|
|
// Verify block caused the eval to be tracked.
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
|
|
// Should unblock because the eval hasn't seen this node class.
|
|
blocked.Unblock("v1:789", 1000)
|
|
requireBlockedEvalsEnqueued(t, blocked, broker, 1)
|
|
}
|
|
|
|
func TestBlockedEvals_UnblockEligible_Quota(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, broker := testBlockedEvals(t)
|
|
|
|
// Create a blocked eval that is eligible for a particular quota.
|
|
e := mock.BlockedEval()
|
|
e.QuotaLimitReached = "foo"
|
|
blocked.Block(e)
|
|
|
|
// Verify block caused the eval to be tracked.
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(1, blockedStats.TotalQuotaLimit)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
|
|
blocked.UnblockQuota("foo", 1000)
|
|
requireBlockedEvalsEnqueued(t, blocked, broker, 1)
|
|
}
|
|
|
|
func TestBlockedEvals_UnblockIneligible_Quota(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, broker := testBlockedEvals(t)
|
|
|
|
// Create a blocked eval that is eligible on a specific quota.
|
|
e := mock.BlockedEval()
|
|
e.QuotaLimitReached = "foo"
|
|
blocked.Block(e)
|
|
|
|
// Verify block caused the eval to be tracked.
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(1, blockedStats.TotalQuotaLimit)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
|
|
// Should do nothing.
|
|
blocked.UnblockQuota("bar", 1000)
|
|
|
|
testutil.WaitForResult(func() (bool, error) {
|
|
// Verify Unblock didn't cause an enqueue
|
|
brokerStats := broker.Stats()
|
|
if brokerStats.TotalReady != 0 {
|
|
return false, fmt.Errorf("eval unblocked: %#v", brokerStats)
|
|
}
|
|
|
|
// Prune old and empty metrics.
|
|
blocked.pruneStats(time.Now().UTC())
|
|
|
|
blockedStats := blocked.Stats()
|
|
ok := blockedStats.TotalBlocked == 1 &&
|
|
blockedStats.TotalEscaped == 0 &&
|
|
blockedStats.TotalQuotaLimit == 1 &&
|
|
len(blockedStats.BlockedResources.ByJob) == 1
|
|
if !ok {
|
|
return false, fmt.Errorf("eval unblocked: %#v", blockedStats)
|
|
}
|
|
return true, nil
|
|
}, func(err error) {
|
|
t.Fatalf("err: %s", err)
|
|
})
|
|
}
|
|
|
|
func TestBlockedEvals_Reblock(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, broker := testBlockedEvals(t)
|
|
|
|
// Create an evaluation, Enqueue/Dequeue it to get a token
|
|
e := mock.BlockedEval()
|
|
e.SnapshotIndex = 500
|
|
e.ClassEligibility = map[string]bool{"v1:123": true, "v1:456": false}
|
|
broker.Enqueue(e)
|
|
|
|
_, token, err := broker.Dequeue([]string{e.Type}, time.Second)
|
|
require.NoError(err)
|
|
|
|
// Reblock the evaluation
|
|
blocked.Reblock(e, token)
|
|
|
|
// Verify block caused the eval to be tracked
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
|
|
// Should unblock because the eval
|
|
blocked.Unblock("v1:123", 1000)
|
|
|
|
brokerStats := broker.Stats()
|
|
require.Equal(0, brokerStats.TotalReady)
|
|
require.Equal(1, brokerStats.TotalUnacked)
|
|
|
|
// Ack the evaluation which should cause the reblocked eval to transition
|
|
// to ready
|
|
err = broker.Ack(e.ID, token)
|
|
require.NoError(err)
|
|
|
|
requireBlockedEvalsEnqueued(t, blocked, broker, 1)
|
|
}
|
|
|
|
// Test the block case in which the eval should be immediately unblocked since
|
|
// it is escaped and old
|
|
func TestBlockedEvals_Block_ImmediateUnblock_Escaped(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, broker := testBlockedEvals(t)
|
|
|
|
// Do an unblock prior to blocking
|
|
blocked.Unblock("v1:123", 1000)
|
|
|
|
// Create a blocked eval that is eligible on a specific node class and add
|
|
// it to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e.EscapedComputedClass = true
|
|
e.SnapshotIndex = 900
|
|
blocked.Block(e)
|
|
|
|
// Verify block caused the eval to be immediately unblocked
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(0, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 0)
|
|
|
|
requireBlockedEvalsEnqueued(t, blocked, broker, 1)
|
|
}
|
|
|
|
// Test the block case in which the eval should be immediately unblocked since
|
|
// there is an unblock on an unseen class that occurred while it was in the
|
|
// scheduler
|
|
func TestBlockedEvals_Block_ImmediateUnblock_UnseenClass_After(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, broker := testBlockedEvals(t)
|
|
|
|
// Do an unblock prior to blocking
|
|
blocked.Unblock("v1:123", 1000)
|
|
|
|
// Create a blocked eval that is eligible on a specific node class and add
|
|
// it to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e.EscapedComputedClass = false
|
|
e.SnapshotIndex = 900
|
|
blocked.Block(e)
|
|
|
|
// Verify block caused the eval to be immediately unblocked
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(0, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 0)
|
|
|
|
requireBlockedEvalsEnqueued(t, blocked, broker, 1)
|
|
}
|
|
|
|
// Test the block case in which the eval should not immediately unblock since
|
|
// there is an unblock on an unseen class that occurred before it was in the
|
|
// scheduler
|
|
func TestBlockedEvals_Block_ImmediateUnblock_UnseenClass_Before(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, _ := testBlockedEvals(t)
|
|
|
|
// Do an unblock prior to blocking
|
|
blocked.Unblock("v1:123", 500)
|
|
|
|
// Create a blocked eval that is eligible on a specific node class and add
|
|
// it to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e.EscapedComputedClass = false
|
|
e.SnapshotIndex = 900
|
|
blocked.Block(e)
|
|
|
|
// Verify block caused the eval to be immediately unblocked
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
}
|
|
|
|
// Test the block case in which the eval should be immediately unblocked since
|
|
// it a class it is eligible for has been unblocked
|
|
func TestBlockedEvals_Block_ImmediateUnblock_SeenClass(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, broker := testBlockedEvals(t)
|
|
|
|
// Do an unblock prior to blocking
|
|
blocked.Unblock("v1:123", 1000)
|
|
|
|
// Create a blocked eval that is eligible on a specific node class and add
|
|
// it to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e.ClassEligibility = map[string]bool{"v1:123": true, "v1:456": false}
|
|
e.SnapshotIndex = 900
|
|
blocked.Block(e)
|
|
|
|
// Verify block caused the eval to be immediately unblocked
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(0, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 0)
|
|
|
|
requireBlockedEvalsEnqueued(t, blocked, broker, 1)
|
|
}
|
|
|
|
// Test the block case in which the eval should be immediately unblocked since
|
|
// it a quota has changed that it is using
|
|
func TestBlockedEvals_Block_ImmediateUnblock_Quota(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, broker := testBlockedEvals(t)
|
|
|
|
// Do an unblock prior to blocking
|
|
blocked.UnblockQuota("my-quota", 1000)
|
|
|
|
// Create a blocked eval that is eligible on a specific node class and add
|
|
// it to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e.QuotaLimitReached = "my-quota"
|
|
e.SnapshotIndex = 900
|
|
blocked.Block(e)
|
|
|
|
// Verify block caused the eval to be immediately unblocked
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(0, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Equal(0, blockedStats.TotalQuotaLimit)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 0)
|
|
|
|
requireBlockedEvalsEnqueued(t, blocked, broker, 1)
|
|
}
|
|
|
|
func TestBlockedEvals_UnblockFailed(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, broker := testBlockedEvals(t)
|
|
|
|
// Create blocked evals that are due to failures
|
|
e := mock.BlockedEval()
|
|
e.TriggeredBy = structs.EvalTriggerMaxPlans
|
|
e.EscapedComputedClass = true
|
|
blocked.Block(e)
|
|
|
|
e2 := mock.BlockedEval()
|
|
e2.Status = structs.EvalStatusBlocked
|
|
e2.TriggeredBy = structs.EvalTriggerMaxPlans
|
|
e2.ClassEligibility = map[string]bool{"v1:123": true, "v1:456": false}
|
|
blocked.Block(e2)
|
|
|
|
e3 := mock.BlockedEval()
|
|
e3.TriggeredBy = structs.EvalTriggerMaxPlans
|
|
e3.QuotaLimitReached = "foo"
|
|
blocked.Block(e3)
|
|
|
|
// Trigger an unblock fail
|
|
blocked.UnblockFailed()
|
|
|
|
// Prune old and empty metrics.
|
|
blocked.pruneStats(time.Now().UTC())
|
|
|
|
// Verify UnblockFailed caused the eval to be immediately unblocked
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(0, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Equal(0, blockedStats.TotalQuotaLimit)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 0)
|
|
|
|
requireBlockedEvalsEnqueued(t, blocked, broker, 3)
|
|
|
|
// Reblock an eval for the same job and check that it gets tracked.
|
|
blocked.Block(e)
|
|
blockedStats = blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(1, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
}
|
|
|
|
func TestBlockedEvals_Untrack(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, _ := testBlockedEvals(t)
|
|
|
|
// Create blocked eval and add to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e.ClassEligibility = map[string]bool{"v1:123": false, "v1:456": false}
|
|
e.SnapshotIndex = 1000
|
|
blocked.Block(e)
|
|
|
|
// Verify block did track
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
|
|
// Untrack and verify
|
|
blocked.Untrack(e.JobID, e.Namespace)
|
|
blocked.pruneStats(time.Now().UTC())
|
|
|
|
blockedStats = blocked.Stats()
|
|
require.Equal(0, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 0)
|
|
}
|
|
|
|
func TestBlockedEvals_Untrack_Quota(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, _ := testBlockedEvals(t)
|
|
|
|
// Create a blocked eval and add it to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e.QuotaLimitReached = "foo"
|
|
e.SnapshotIndex = 1000
|
|
blocked.Block(e)
|
|
|
|
// Verify block did track
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
|
|
// Untrack and verify
|
|
blocked.Untrack(e.JobID, e.Namespace)
|
|
blocked.pruneStats(time.Now().UTC())
|
|
|
|
blockedStats = blocked.Stats()
|
|
require.Equal(0, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 0)
|
|
}
|
|
|
|
func TestBlockedEvals_UnblockNode(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, broker := testBlockedEvals(t)
|
|
|
|
require.NotNil(t, broker)
|
|
|
|
// Create a blocked evals and add it to the blocked tracker.
|
|
e := mock.BlockedEval()
|
|
e.Type = structs.JobTypeSystem
|
|
e.NodeID = "foo"
|
|
e.SnapshotIndex = 999
|
|
blocked.Block(e)
|
|
|
|
// Verify block did track
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
|
|
blocked.UnblockNode("foo", 1000)
|
|
requireBlockedEvalsEnqueued(t, blocked, broker, 1)
|
|
|
|
blocked.pruneStats(time.Now().UTC())
|
|
blockedStats = blocked.Stats()
|
|
require.Empty(blocked.system.byNode)
|
|
require.Equal(0, blockedStats.TotalBlocked)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 0)
|
|
}
|
|
|
|
func TestBlockedEvals_SystemUntrack(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, _ := testBlockedEvals(t)
|
|
|
|
// Create a blocked evals and add it to the blocked tracker.
|
|
e := mock.Eval()
|
|
e.Type = structs.JobTypeSystem
|
|
e.NodeID = "foo"
|
|
blocked.Block(e)
|
|
|
|
// Verify block did track
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Equal(0, blockedStats.TotalQuotaLimit)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
|
|
// Untrack and verify
|
|
blocked.Untrack(e.JobID, e.Namespace)
|
|
blocked.pruneStats(time.Now().UTC())
|
|
blockedStats = blocked.Stats()
|
|
require.Equal(0, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Equal(0, blockedStats.TotalQuotaLimit)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 0)
|
|
}
|
|
|
|
func TestBlockedEvals_SystemDisableFlush(t *testing.T) {
|
|
ci.Parallel(t)
|
|
require := require.New(t)
|
|
|
|
blocked, _ := testBlockedEvals(t)
|
|
|
|
// Create a blocked evals and add it to the blocked tracker.
|
|
e := mock.Eval()
|
|
e.Type = structs.JobTypeSystem
|
|
e.NodeID = "foo"
|
|
blocked.Block(e)
|
|
|
|
// Verify block did track
|
|
blockedStats := blocked.Stats()
|
|
require.Equal(1, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Equal(0, blockedStats.TotalQuotaLimit)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 1)
|
|
|
|
// Disable empties
|
|
blocked.SetEnabled(false)
|
|
blockedStats = blocked.Stats()
|
|
require.Equal(0, blockedStats.TotalBlocked)
|
|
require.Equal(0, blockedStats.TotalEscaped)
|
|
require.Equal(0, blockedStats.TotalQuotaLimit)
|
|
require.Len(blockedStats.BlockedResources.ByJob, 0)
|
|
require.Empty(blocked.system.evals)
|
|
require.Empty(blocked.system.byJob)
|
|
require.Empty(blocked.system.byNode)
|
|
}
|