Adds basic tests for dispatcher classes

.mockery.yaml

@@ -6,5 +6,8 @@ mock-build-tags: "!compile"
 packages:
   github.com/ksysoev/wasabi:
     interfaces:
-      Dispatcher:
       ConnectionRegistry:
+      Dispatcher:
+      Backend:
+      RequestHandler:
+      Connection:

dispatch/common.go

@@ -3,7 +3,7 @@ package dispatch
 import "github.com/ksysoev/wasabi"
 
 // RequestMiddlewere is interface for request middleweres
-type RequestMiddlewere func(next RequestHandler) RequestHandler
+type RequestMiddlewere func(next wasabi.RequestHandler) wasabi.RequestHandler
 
 // RequestHandlerFunc is a function that implements RequestHandler interface
 type RequestHandlerFunc func(conn wasabi.Connection, req wasabi.Request) error
@@ -12,9 +12,3 @@ type RequestHandlerFunc func(conn wasabi.Connection, req wasabi.Request) error
 func (f RequestHandlerFunc) Handle(conn wasabi.Connection, req wasabi.Request) error {
 	return f(conn, req)
 }
-
-// TODO: Shall we moove it wasabi package?
-// RequestHandler is interface for request handlers
-type RequestHandler interface {
-	Handle(conn wasabi.Connection, req wasabi.Request) error
-}

dispatch/dispatcher.go

@@ -34,7 +34,7 @@ func (d *PipeDispatcher) Use(middlewere RequestMiddlewere) {
 }
 
 // useMiddleware prepare request handler with middleweres chain
-func (d *PipeDispatcher) useMiddleware(endpoint RequestHandler) RequestHandler {
+func (d *PipeDispatcher) useMiddleware(endpoint wasabi.RequestHandler) wasabi.RequestHandler {
 	for i := len(d.middlewares) - 1; i >= 0; i-- {
 		endpoint = d.middlewares[i](endpoint)
 	}

interfaces.go

@@ -42,3 +42,8 @@ type Connection interface {
 	ID() string
 	HandleRequests()
 }
+
+// RequestHandler is interface for request handlers
+type RequestHandler interface {
+	Handle(conn Connection, req Request) error
+}

middleware/request/error_handler.go

@@ -7,8 +7,8 @@ import (
 
 type ErrorHandler func(conn wasabi.Connection, req wasabi.Request, err error) error
 
-func NewErrorHandlingMiddleware(onError ErrorHandler) func(next dispatch.RequestHandler) dispatch.RequestHandler {
-	return func(next dispatch.RequestHandler) dispatch.RequestHandler {
+func NewErrorHandlingMiddleware(onError ErrorHandler) func(next wasabi.RequestHandler) wasabi.RequestHandler {
+	return func(next wasabi.RequestHandler) wasabi.RequestHandler {
 		return dispatch.RequestHandlerFunc(func(conn wasabi.Connection, req wasabi.Request) error {
 			err := next.Handle(conn, req)
 

middleware/request/throttler.go

@@ -7,10 +7,10 @@ import (
 
 type token struct{}
 
-func NewTrottlerMiddleware(limit uint) func(next dispatch.RequestHandler) dispatch.RequestHandler {
+func NewTrottlerMiddleware(limit uint) func(next wasabi.RequestHandler) wasabi.RequestHandler {
 	sem := make(chan token, limit)
 
-	return func(next dispatch.RequestHandler) dispatch.RequestHandler {
+	return func(next wasabi.RequestHandler) wasabi.RequestHandler {
 		return dispatch.RequestHandlerFunc(func(conn wasabi.Connection, req wasabi.Request) error {
 			select {
 			case sem <- token{}: