3
0

upd(rbac): refactor debug level to info/debug

This commit is contained in:
Tit Petric
2018-07-23 13:59:56 +02:00
parent 01ac832e80
commit 4d018e16cd
2 changed files with 18 additions and 11 deletions

View File

@@ -20,8 +20,8 @@ type (
Transport *http.Transport
Client *http.Client
isDebug bool
config configuration
debugLevel string
config configuration
}
ClientInterface interface {
@@ -51,8 +51,6 @@ func New() (*Client, error) {
Timeout: timeout,
}).Dial,
TLSHandshakeTimeout: timeout,
// @todo: === remove this line ===
//TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
}
client := &http.Client{
@@ -66,13 +64,12 @@ func New() (*Client, error) {
return &Client{
Transport: transport,
Client: client,
isDebug: false,
config: config,
}, nil
}
func (c *Client) Debug(debug bool) *Client {
c.isDebug = debug
func (c *Client) Debug(debugLevel string) *Client {
c.debugLevel = debugLevel
return c
}
@@ -95,7 +92,7 @@ func (c *Client) Delete(url string) (*http.Response, error) {
func (c *Client) Request(method string, url string, body interface{}) (*http.Response, error) {
link := strings.TrimRight(c.config.baseURL, "/") + "/" + strings.TrimLeft(url, "/")
if c.isDebug {
if c.debugLevel == "info" {
fmt.Println("RBAC >>>", method, link)
}
@@ -120,7 +117,7 @@ func (c *Client) Request(method string, url string, body interface{}) (*http.Res
// req.Header.Add("X-TENANT-ID", c.config.tenant)
req.Header["X-TENANT-ID"] = []string{c.config.tenant}
if c.isDebug {
if c.debugLevel == "debug" {
fmt.Println("RBAC >>> (request)")
b, err := httputil.DumpRequestOut(req, true)
if err != nil {
@@ -134,7 +131,7 @@ func (c *Client) Request(method string, url string, body interface{}) (*http.Res
}
resp, err := c.Client.Do(req)
if c.isDebug {
if c.debugLevel == "debug" {
fmt.Println("RBAC <<< (response)")
if err != nil {
fmt.Println("RBAC <<< Error:", err)
@@ -151,9 +148,11 @@ func (c *Client) Request(method string, url string, body interface{}) (*http.Res
}
fmt.Println("-----------------")
}
if c.debugLevel == "info" {
fmt.Println("RBAC <<< Response with", resp.StatusCode)
}
if err != nil {
return nil, err
}
return resp, nil
}

View File

@@ -3,6 +3,7 @@ package rbac_test
import (
"github.com/crusttech/crust/rbac"
"github.com/namsral/flag"
"testing"
)
var loaded bool
@@ -15,3 +16,10 @@ func getClient() (*rbac.Client, error) {
}
return rbac.New()
}
func assert(t *testing.T, ok bool, format string, args ...interface{}) bool {
if !ok {
t.Errorf(format, args...)
}
return ok
}