mirror of
https://github.com/siderolabs/discovery-service.git
synced 2025-03-14 09:55:08 +00:00
619546696a
Fixes #5 Batch watch responses in a single batch so that client can quickly know that initial snapshot got delivered. Bump go.mod deps. Implement more limits. Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
117 lines
2.8 KiB
Go
117 lines
2.8 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package state_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/talos-systems/discovery-service/internal/state"
|
|
"github.com/talos-systems/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)
|
|
}
|