test_helper.go

package gdax

import (
	"errors"
	"fmt"
	"net/http"
	"os"
	"reflect"
	"time"

	ws "github.com/gorilla/websocket"
)

func NewTestClient() *Client {
	secret := os.Getenv("TEST_COINBASE_SECRET")
	key := os.Getenv("TEST_COINBASE_KEY")
	passphrase := os.Getenv("TEST_COINBASE_PASSPHRASE")

	client := NewClient(secret, key, passphrase)
	client.BaseURL = "https://api-public.sandbox.gdax.com"
	client.HttpClient = &http.Client{
		Timeout: 15 * time.Second,
	}
	client.RetryCount = 2

	return client
}

func NewTestWebsocketClient() (*ws.Conn, error) {
	var wsDialer ws.Dialer
	wsConn, _, err := wsDialer.Dial("wss://ws-feed-public.sandbox.gdax.com", nil)

	return wsConn, err
}

func StructHasZeroValues(i interface{}) bool {
	iv := reflect.ValueOf(i)

	//values := make([]interface{}, v.NumField())

	for i := 0; i < iv.NumField(); i++ {
		field := iv.Field(i)
		if reflect.Zero(field.Type()).Interface() == field.Interface() {
			return true
		}
	}

	return false
}

func CompareProperties(a, b interface{}, properties []string) (bool, error) {
	aValueOf := reflect.ValueOf(a)
	bValueOf := reflect.ValueOf(b)

	for _, property := range properties {
		aValue := reflect.Indirect(aValueOf).FieldByName(property).Interface()
		bValue := reflect.Indirect(bValueOf).FieldByName(property).Interface()

		if aValue != bValue {
			return false, errors.New(fmt.Sprintf("%s not equal: %s - %s", property, aValue, bValue))
		}
	}

	return true, nil
}

func Ensure(a interface{}) error {
	field := reflect.Indirect(reflect.ValueOf(a))

	switch field.Kind() {
	case reflect.Slice:
		if reflect.ValueOf(field.Interface()).Len() == 0 {
			return errors.New(fmt.Sprintf("Slice is zero"))
		}
	default:
		if reflect.Zero(field.Type()).Interface() == field.Interface() {
			return errors.New(fmt.Sprintf("Property is zero"))
		}
	}

	return nil
}

func EnsureProperties(a interface{}, properties []string) error {
	valueOf := reflect.ValueOf(a)

	for _, property := range properties {
		field := reflect.Indirect(valueOf).FieldByName(property)

		if err := Ensure(field.Interface()); err != nil {
			return errors.New(fmt.Sprintf("%s: %s", err.Error(), property))
		}
	}

	return nil
}