mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 03:51:29 +00:00
d17996f8b0
* Update V3 from V4 * Fix build v3 -> v4 * Update ssz * Update beacon_chain.pb.go * Fix formatter import * Update update-mockgen.sh comment to v4 * Fix conflicts. Pass build and tests * Fix test
110 lines
2.8 KiB
Go
110 lines
2.8 KiB
Go
package runtime
|
|
|
|
import (
|
|
"errors"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/v4/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/require"
|
|
)
|
|
|
|
type mockService struct {
|
|
status error
|
|
}
|
|
type secondMockService struct {
|
|
status error
|
|
}
|
|
|
|
func (_ *mockService) Start() {
|
|
}
|
|
|
|
func (_ *mockService) Stop() error {
|
|
return nil
|
|
}
|
|
|
|
func (m *mockService) Status() error {
|
|
return m.status
|
|
}
|
|
|
|
func (_ *secondMockService) Start() {
|
|
}
|
|
|
|
func (_ *secondMockService) Stop() error {
|
|
return nil
|
|
}
|
|
|
|
func (s *secondMockService) Status() error {
|
|
return s.status
|
|
}
|
|
|
|
func TestRegisterService_Twice(t *testing.T) {
|
|
registry := &ServiceRegistry{
|
|
services: make(map[reflect.Type]Service),
|
|
}
|
|
|
|
m := &mockService{}
|
|
require.NoError(t, registry.RegisterService(m), "Failed to register first service")
|
|
|
|
// Checks if first service was indeed registered.
|
|
require.Equal(t, 1, len(registry.serviceTypes))
|
|
assert.ErrorContains(t, "service already exists", registry.RegisterService(m))
|
|
}
|
|
|
|
func TestRegisterService_Different(t *testing.T) {
|
|
registry := &ServiceRegistry{
|
|
services: make(map[reflect.Type]Service),
|
|
}
|
|
|
|
m := &mockService{}
|
|
s := &secondMockService{}
|
|
require.NoError(t, registry.RegisterService(m), "Failed to register first service")
|
|
require.NoError(t, registry.RegisterService(s), "Failed to register second service")
|
|
|
|
require.Equal(t, 2, len(registry.serviceTypes))
|
|
|
|
_, exists := registry.services[reflect.TypeOf(m)]
|
|
assert.Equal(t, true, exists, "service of type %v not registered", reflect.TypeOf(m))
|
|
|
|
_, exists = registry.services[reflect.TypeOf(s)]
|
|
assert.Equal(t, true, exists, "service of type %v not registered", reflect.TypeOf(s))
|
|
}
|
|
|
|
func TestFetchService_OK(t *testing.T) {
|
|
registry := &ServiceRegistry{
|
|
services: make(map[reflect.Type]Service),
|
|
}
|
|
|
|
m := &mockService{}
|
|
require.NoError(t, registry.RegisterService(m), "Failed to register first service")
|
|
|
|
assert.ErrorContains(t, "input must be of pointer type, received value type instead", registry.FetchService(*m))
|
|
|
|
var s *secondMockService
|
|
assert.ErrorContains(t, "unknown service", registry.FetchService(&s))
|
|
|
|
var m2 *mockService
|
|
require.NoError(t, registry.FetchService(&m2), "Failed to fetch service")
|
|
require.Equal(t, m, m2)
|
|
}
|
|
|
|
func TestServiceStatus_OK(t *testing.T) {
|
|
registry := &ServiceRegistry{
|
|
services: make(map[reflect.Type]Service),
|
|
}
|
|
|
|
m := &mockService{}
|
|
require.NoError(t, registry.RegisterService(m), "Failed to register first service")
|
|
|
|
s := &secondMockService{}
|
|
require.NoError(t, registry.RegisterService(s), "Failed to register first service")
|
|
|
|
m.status = errors.New("something bad has happened")
|
|
s.status = errors.New("woah, horsee")
|
|
|
|
statuses := registry.Statuses()
|
|
|
|
assert.ErrorContains(t, "something bad has happened", statuses[reflect.TypeOf(m)])
|
|
assert.ErrorContains(t, "woah, horsee", statuses[reflect.TypeOf(s)])
|
|
}
|