mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-22 19:40:37 +00:00
5569a68452
* Value assigned to a variable is never read before being overwritten * The result of append is not used anywhere * Suspicious assignment of range-loop vars detected * Unused method receiver detected * Revert "Auxiliary commit to revert individual files from 54edcb445484a2e5d79612e19af8e949b8861253" This reverts commit bbd1e1beabf7b0c5cfc4f514dcc820062ad6c063. * Method modifies receiver * Fix test * Duplicate imports detected * Incorrectly formatted error string * Types of function parameters can be combined * One more "Unused method receiver detected" * Unused parameter detected in function
110 lines
2.8 KiB
Go
110 lines
2.8 KiB
Go
package runtime
|
|
|
|
import (
|
|
"errors"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/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)])
|
|
}
|