/internal/streamingcoord/server/resource/test_utility.go

//go:build test
// +build test

package resource

import (
	"github.com/milvus-io/milvus/internal/streamingnode/client/manager"
	"github.com/milvus-io/milvus/pkg/v3/mlog"
)

// OptStreamingManagerClient provides streaming manager client to the resource.
func OptStreamingManagerClient(c manager.ManagerClient) optResourceInit {
	return func(r *resourceImpl) {
		r.streamingNodeManagerClient = c
	}
}

// InitForTest initializes the singleton of resources for test.
func InitForTest(opts ...optResourceInit) {
	newR := &resourceImpl{
		logger: mlog.With(),
	}
	for _, opt := range opts {
		opt(newR)
	}
	r.Store(newR)
}