d183a241e9
* chore: wrap HTTP server in a dskit module Much of the logic from this comes from the POC branch, so: - credit for this work goes to everyone else - mistakes are my own This is needed to support microservice deployment modes. * added an arbitrarily-chosen 30second timeout
34 lines
989 B
Go
34 lines
989 B
Go
package registry
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/grafana/dskit/services"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/modules"
|
|
)
|
|
|
|
func TestNewRegistry(t *testing.T) {
|
|
var registeredInvisibleModules []string
|
|
var registeredModules []string
|
|
|
|
moduleManager := &modules.MockModuleManager{
|
|
RegisterModuleFunc: func(name string, initFn func() (services.Service, error)) {
|
|
registeredModules = append(registeredModules, name)
|
|
},
|
|
RegisterInvisibleModuleFunc: func(name string, initFn func() (services.Service, error)) {
|
|
registeredInvisibleModules = append(registeredInvisibleModules, name)
|
|
},
|
|
}
|
|
|
|
mockSvcName := "test-registry"
|
|
mockSvc := modules.NewMockNamedService(mockSvcName)
|
|
|
|
r := newRegistry(log.New("modules.registry"), moduleManager, mockSvc)
|
|
require.NotNil(t, r)
|
|
require.Equal(t, []string{mockSvcName}, registeredInvisibleModules)
|
|
require.Equal(t, []string{modules.All}, registeredModules)
|
|
}
|