[Utility] Add servicer module mock

shared/modules/fisherman_module.go

@@ -0,0 +1,5 @@
+package modules
+
+type FishermanModule interface {
+	Module
+}

shared/modules/servicer_module.go

@@ -0,0 +1,16 @@
+package modules
+
+//go:generate mockgen -destination=./mocks/servicer_module_mock.go github.com/pokt-network/pocket/shared/modules ServicerModule
+
+import (
+	coreTypes "github.com/pokt-network/pocket/shared/core/types"
+)
+
+const (
+	ServicerModuleName = "servicer"
+)
+
+type ServicerModule interface {
+	Module
+	HandleRelay(*coreTypes.Relay) (*coreTypes.RelayResponse, error)
+}

shared/modules/utility_module.go

@@ -58,19 +58,6 @@ type UtilityModule interface {
 	GetValidatorModule() (ValidatorModule, error)
 }
 
-type FishermanModule interface {
-	Module
-}
-
-type ServicerModule interface {
-	Module
-	HandleRelay(*coreTypes.Relay) (*coreTypes.RelayResponse, error)
-}
-
-type ValidatorModule interface {
-	Module
-}
-
 // TECHDEBT: Remove this interface from `shared/modules` and use the `Actor` protobuf type instead
 // There will need to be some documentation or indicator that the Actor struct returned may not be
 // fully hydrated. Alternatively, we could eat the performance cost and just hydrate the entire struct

shared/modules/validator_module.go

@@ -0,0 +1,5 @@
+package modules
+
+type ValidatorModule interface {
+	Module
+}

utility/module.go

@@ -142,10 +142,10 @@ func (u *utilityModule) GetActorModules() map[string]modules.Module {
 }
 
 func (u *utilityModule) GetServicerModule() (modules.ServicerModule, error) {
-	if u.actorModules[servicer.ServicerModuleName] == nil {
+	if u.actorModules[modules.ServicerModuleName] == nil {
 		return nil, errors.New("servicer module not enabled")
 	}
-	if m, ok := u.actorModules[servicer.ServicerModuleName].(modules.ServicerModule); ok {
+	if m, ok := u.actorModules[modules.ServicerModuleName].(modules.ServicerModule); ok {
 		return m, nil
 	}
 	return nil, errors.New("failed to cast servicer module")

utility/servicer/module.go

@@ -35,10 +35,6 @@ var (
 	_ modules.ServicerModule = &servicer{}
 )
 
-const (
-	ServicerModuleName = "servicer"
-)
-
 // sessionTokens is used to cache the starting number of tokens available
 // during a specific session: it is used as the value for a map with keys being applications' public keys
 // TODO: What if we have a servicer managing more than one session from the same app at once? We may/may not need to resolve this in the future.
@@ -121,7 +117,7 @@ func (s *servicer) Stop() error {
 }
 
 func (s *servicer) GetModuleName() string {
-	return ServicerModuleName
+	return modules.ServicerModuleName
 }
 
 // HandleRelay processes a relay after performing validation.