update gobreaker wrapper to allow multiservice breaker

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>

gobreaker_test.go

@@ -1,14 +1,14 @@
 package gobreaker
 
 import (
+	"context"
 	"testing"
 
 	"github.com/micro/go-micro/client"
+	"github.com/micro/go-micro/errors"
 	"github.com/micro/go-micro/registry/memory"
 	"github.com/micro/go-micro/selector"
 	"github.com/sony/gobreaker"
-
-	"context"
 )
 
 func TestBreaker(t *testing.T) {
@@ -20,8 +20,43 @@ func TestBreaker(t *testing.T) {
 		// set the selector
 		client.Selector(s),
 		// add the breaker wrapper
-		client.Wrap(NewClientWrapper(
-			gobreaker.NewCircuitBreaker(gobreaker.Settings{}),
+		client.Wrap(NewClientWrapper()),
+	)
+
+	req := c.NewRequest("test.service", "Test.Method", map[string]string{
+		"foo": "bar",
+	}, client.WithContentType("application/json"))
+
+	var rsp map[string]interface{}
+
+	// Force to point of trip
+	for i := 0; i < 6; i++ {
+		c.Call(context.TODO(), req, rsp)
+	}
+
+	err := c.Call(context.TODO(), req, rsp)
+	if err == nil {
+		t.Error("Expecting tripped breaker, got nil error")
+	}
+
+	merr := err.(*errors.Error)
+	if merr.Code != 502 {
+		t.Errorf("Expecting tripped breaker, got %v", err)
+	}
+}
+
+func TestCustomBreaker(t *testing.T) {
+	// setup
+	r := memory.NewRegistry()
+	s := selector.NewSelector(selector.Registry(r))
+
+	c := client.NewClient(
+		// set the selector
+		client.Selector(s),
+		// add the breaker wrapper
+		client.Wrap(NewCustomClientWrapper(
+			gobreaker.Settings{},
+			BreakService,
 		)),
 	)
 
@@ -41,7 +76,8 @@ func TestBreaker(t *testing.T) {
 		t.Error("Expecting tripped breaker, got nil error")
 	}
 
-	if err.Error() != "circuit breaker is open" {
+	merr := err.(*errors.Error)
+	if merr.Code != 502 {
 		t.Errorf("Expecting tripped breaker, got %v", err)
 	}
 }