Merge pull request #15 from it-chain/feature/avengers

implement network delay property

mock/event_service.go

@@ -19,24 +19,28 @@ package mock
 import (
 	"errors"
 	"reflect"
+	"time"
 )
 
 var ErrEventType = errors.New("Error type of event is not struct")
 
 type EventService struct {
 	ProcessId   string
 	PublishFunc func(processId string, topic string, event interface{}) error
+	delayTime   time.Duration
 }
 
 func NewEventService(processId string, publishFunc func(processId string, topic string, event interface{}) error) *EventService {
 	return &EventService{
 		ProcessId:   processId,
 		PublishFunc: publishFunc,
+		delayTime:   0,
 	}
 }
 
 func (s *EventService) Publish(topic string, event interface{}) error {
 
+	time.Sleep(s.delayTime)
 	if !eventIsStruct(event) {
 		return ErrEventType
 	}
@@ -48,7 +52,11 @@ func (s *EventService) Publish(topic string, event interface{}) error {
 	return nil
 }
 
-func (s *EventService) Close(){}
+func (s *EventService) SetDelayTime(t time.Duration) {
+	s.delayTime = t
+}
+
+func (s *EventService) Close() {}
 
 func eventIsStruct(event interface{}) bool {
 	return reflect.TypeOf(event).Kind() == reflect.Struct

mock/event_service_test.go

@@ -18,18 +18,22 @@ package mock_test
 
 import (
 	"testing"
+	"time"
+
 	"github.com/it-chain/avengers/mock"
 	"github.com/it-chain/engine/common/command"
 )
 
 func TestEventService_Publish(t *testing.T) {
-	eventService:=mock.NewEventService("1", func(processId string, topic string, event interface{}) error {
+	eventService := mock.NewEventService("1", func(processId string, topic string, event interface{}) error {
 		return nil
 	})
 
+	eventService.SetDelayTime(5 * time.Millisecond)
+
 	event := command.DeliverGrpc{
-		MessageId:"1",
+		MessageId: "1",
 	}
 
-	eventService.Publish("message.deliver",event)
+	eventService.Publish("message.deliver", event)
 }