mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-31 00:11:36 +01:00
badbc51e63
`config.Container` implements a multi-doc container which implements both `Container` interface (encoding, validation, etc.), and `Conifg` interface (accessing parts of the config). Refactor `generate` and `bundle` packages to support multi-doc, and provide backwards compatibility. Implement a first (mostly example) machine config document for SideroLink API URL. Many places don't properly support multi-doc yet (e.g. config patches). Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
103 lines
2.5 KiB
Go
103 lines
2.5 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/.
|
|
|
|
//go:build integration_api
|
|
|
|
package api
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/siderolabs/talos/internal/integration/base"
|
|
"github.com/siderolabs/talos/pkg/machinery/client"
|
|
machinetype "github.com/siderolabs/talos/pkg/machinery/config/machine"
|
|
)
|
|
|
|
// EventsSuite verifies Events API.
|
|
type EventsSuite struct {
|
|
base.APISuite
|
|
|
|
ctx context.Context //nolint:containedctx
|
|
ctxCancel context.CancelFunc
|
|
|
|
nodeCtx context.Context //nolint:containedctx
|
|
}
|
|
|
|
// SuiteName ...
|
|
func (suite *EventsSuite) SuiteName() string {
|
|
return "api.EventsSuite"
|
|
}
|
|
|
|
// SetupTest ...
|
|
func (suite *EventsSuite) SetupTest() {
|
|
// make sure API calls have timeout
|
|
suite.ctx, suite.ctxCancel = context.WithTimeout(context.Background(), 30*time.Second)
|
|
|
|
suite.nodeCtx = client.WithNodes(suite.ctx, suite.RandomDiscoveredNodeInternalIP(machinetype.TypeWorker))
|
|
}
|
|
|
|
// TearDownTest ...
|
|
func (suite *EventsSuite) TearDownTest() {
|
|
if suite.ctxCancel != nil {
|
|
suite.ctxCancel()
|
|
}
|
|
}
|
|
|
|
// TestEventsWatch verifies events watch API.
|
|
func (suite *EventsSuite) TestEventsWatch() {
|
|
receiveEvents := func(opts ...client.EventsOptionFunc) []client.Event {
|
|
var result []client.Event
|
|
|
|
watchCtx, watchCtxCancel := context.WithCancel(suite.nodeCtx)
|
|
defer watchCtxCancel()
|
|
|
|
suite.Assert().NoError(
|
|
suite.Client.EventsWatch(
|
|
watchCtx, func(ch <-chan client.Event) {
|
|
defer watchCtxCancel()
|
|
|
|
timer := time.NewTimer(500 * time.Millisecond)
|
|
defer timer.Stop()
|
|
|
|
for {
|
|
select {
|
|
case event, ok := <-ch:
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
result = append(result, event)
|
|
case <-timer.C:
|
|
return
|
|
}
|
|
}
|
|
}, opts...,
|
|
),
|
|
)
|
|
|
|
return result
|
|
}
|
|
|
|
allEvents := receiveEvents(client.WithTailEvents(-1))
|
|
suite.Require().Greater(len(allEvents), 20)
|
|
|
|
suite.Assert().Len(receiveEvents(), 0)
|
|
suite.Assert().Len(receiveEvents(client.WithTailEvents(5)), 5)
|
|
suite.Assert().Len(receiveEvents(client.WithTailEvents(20)), 20)
|
|
|
|
// pick some ID of 15th event in the past; API should return at least 14 events
|
|
// (as check excludes that event with picked ID)
|
|
id := allEvents[len(allEvents)-15].ID
|
|
eventsSinceID := receiveEvents(client.WithTailID(id))
|
|
suite.Require().GreaterOrEqual(
|
|
len(eventsSinceID),
|
|
14,
|
|
) // there might some new events since allEvents, but at least 15 should be received
|
|
}
|
|
|
|
func init() {
|
|
allSuites = append(allSuites, new(EventsSuite))
|
|
}
|