mirror of
https://github.com/siderolabs/discovery-service.git
synced 2025-03-14 09:55:08 +00:00
c55142668f
Update removing multiple old middlewares, rework the way data is passed through the context, logging fields, etc. Fix minimum keepalive interval enforcement. Signed-off-by: Andrey Smirnov <andrey.smirnov@siderolabs.com>
118 lines
2.8 KiB
Go
118 lines
2.8 KiB
Go
// Copyright (c) 2024 Sidero Labs, Inc.
|
|
//
|
|
// Use of this software is governed by the Business Source License
|
|
// included in the LICENSE file.
|
|
|
|
package state_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/siderolabs/discovery-service/internal/state"
|
|
"github.com/siderolabs/discovery-service/pkg/limits"
|
|
)
|
|
|
|
func TestAffiliateMutations(t *testing.T) {
|
|
now := time.Now()
|
|
|
|
affiliate := state.NewAffiliate("id1")
|
|
|
|
affiliate.ClearChanged()
|
|
assert.False(t, affiliate.IsChanged())
|
|
|
|
affiliate.Update([]byte("data"), now.Add(time.Minute))
|
|
|
|
assert.Equal(t, &state.AffiliateExport{
|
|
ID: "id1",
|
|
Data: []byte("data"),
|
|
Endpoints: [][]byte{},
|
|
}, affiliate.Export())
|
|
|
|
assert.True(t, affiliate.IsChanged())
|
|
|
|
affiliate.ClearChanged()
|
|
|
|
affiliate.Update([]byte("data1"), now.Add(time.Minute))
|
|
|
|
assert.Equal(t, &state.AffiliateExport{
|
|
ID: "id1",
|
|
Data: []byte("data1"),
|
|
Endpoints: [][]byte{},
|
|
}, affiliate.Export())
|
|
|
|
assert.True(t, affiliate.IsChanged())
|
|
|
|
affiliate.ClearChanged()
|
|
|
|
assert.NoError(t, affiliate.MergeEndpoints([][]byte{
|
|
[]byte("e1"),
|
|
[]byte("e2"),
|
|
}, now.Add(time.Minute)))
|
|
|
|
assert.Equal(t, &state.AffiliateExport{
|
|
ID: "id1",
|
|
Data: []byte("data1"),
|
|
Endpoints: [][]byte{[]byte("e1"), []byte("e2")},
|
|
}, affiliate.Export())
|
|
|
|
assert.True(t, affiliate.IsChanged())
|
|
affiliate.ClearChanged()
|
|
|
|
assert.NoError(t, affiliate.MergeEndpoints([][]byte{
|
|
[]byte("e1"),
|
|
}, now.Add(time.Minute)))
|
|
|
|
assert.False(t, affiliate.IsChanged())
|
|
|
|
assert.NoError(t, affiliate.MergeEndpoints([][]byte{
|
|
[]byte("e1"),
|
|
[]byte("e3"),
|
|
}, now.Add(3*time.Minute)))
|
|
|
|
assert.Equal(t, &state.AffiliateExport{
|
|
ID: "id1",
|
|
Data: []byte("data1"),
|
|
Endpoints: [][]byte{[]byte("e1"), []byte("e2"), []byte("e3")},
|
|
}, affiliate.Export())
|
|
|
|
assert.True(t, affiliate.IsChanged())
|
|
|
|
remove, changed := affiliate.GarbageCollect(now)
|
|
assert.False(t, remove)
|
|
assert.False(t, changed)
|
|
|
|
remove, changed = affiliate.GarbageCollect(now.Add(2 * time.Minute))
|
|
assert.False(t, remove)
|
|
assert.True(t, changed)
|
|
|
|
assert.Equal(t, &state.AffiliateExport{
|
|
ID: "id1",
|
|
Data: []byte("data1"),
|
|
Endpoints: [][]byte{[]byte("e1"), []byte("e3")},
|
|
}, affiliate.Export())
|
|
|
|
remove, changed = affiliate.GarbageCollect(now.Add(4 * time.Minute))
|
|
assert.True(t, remove)
|
|
assert.True(t, changed)
|
|
}
|
|
|
|
func TestAffiliateTooManyEndpoints(t *testing.T) {
|
|
now := time.Now()
|
|
|
|
affiliate := state.NewAffiliate("id1")
|
|
|
|
for i := 0; i < limits.AffiliateEndpointsMax; i++ {
|
|
assert.NoError(t, affiliate.MergeEndpoints([][]byte{[]byte(fmt.Sprintf("endpoint%d", i))}, now))
|
|
}
|
|
|
|
err := affiliate.MergeEndpoints([][]byte{[]byte("endpoint")}, now)
|
|
require.Error(t, err)
|
|
|
|
assert.ErrorIs(t, err, state.ErrTooManyEndpoints)
|
|
}
|