add test

larscom · Jul 20, 2024 · 1c1418b · 1c1418b
1 parent ee43f5c
commit 1c1418b
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 7 deletions.
diff --git a/internal/crypto/crypto.go b/internal/crypto/crypto.go
@@ -18,8 +18,8 @@ func CreateSignature(
 	parts := []string{fmt.Sprint(timestamp), httpMethod, "/v2", relativePath}
 	if len(body) > 0 {
 		parts = append(parts, string(body))
-
 	}
+
 	hash := hmac.New(sha256.New, []byte(apiSecret))
 	hash.Write([]byte(strings.Join(parts, "")))
 	return hex.EncodeToString(hash.Sum(nil))

diff --git a/internal/crypto/crypto_test.go b/internal/crypto/crypto_test.go
@@ -0,0 +1,19 @@
+package crypto
+
+import (
+	"github.com/larscom/bitvavo-go/internal/test"
+	"testing"
+)
+
+func TestCreateSignature(t *testing.T) {
+	body := "{\"market\":\"ETH-EUR\",\"amount\":1.5,\"price\":2500.5 }"
+	timestamp := int64(1721452468484)
+	sig := CreateSignature("GET", "/test", []byte(body), timestamp, "API_SECRET")
+	test.AssertEqual(t, "d922f806412a560232d5326d95c389893432325f0e89f303f8ed5c9c04cc242b", sig)
+}
+
+func TestCreateSignatureNoBody(t *testing.T) {
+	timestamp := int64(1721452468484)
+	sig := CreateSignature("GET", "/test", nil, timestamp, "API_SECRET")
+	test.AssertEqual(t, "dce7f0d49d559d6012733af234fa2bdef5a8492842726405e5c0b514f9bf1f55", sig)
+}
diff --git a/pkg/bitvavo/util_test.go → internal/test/util.go b/pkg/bitvavo/util_test.go → internal/test/util.go
@@ -1,8 +1,8 @@
-package bitvavo
+package test
 
 import "testing"
 
-func assert[T comparable](t *testing.T, expected T, actual T) {
+func AssertEqual[T comparable](t *testing.T, expected T, actual T) {
 	if expected != actual {
 		t.Errorf("\nexpected: %v\nactual: %v\n", expected, actual)
 	}

diff --git a/pkg/bitvavo/order_test.go b/pkg/bitvavo/order_test.go
@@ -2,6 +2,7 @@ package bitvavo
 
 import (
 	"fmt"
+	"github.com/larscom/bitvavo-go/internal/test"
 	"testing"
 
 	"github.com/goccy/go-json"
@@ -35,7 +36,7 @@ func TestMaxOrderNewMarshaller(t *testing.T) {
 
 	fmt.Printf("%q", actual)
 
-	assert(t, expected, actual)
+	test.AssertEqual(t, expected, actual)
 }
 
 func TestMinOrderNewMarshaller(t *testing.T) {
@@ -55,7 +56,7 @@ func TestMinOrderNewMarshaller(t *testing.T) {
 
 	fmt.Printf("%q", actual)
 
-	assert(t, expected, actual)
+	test.AssertEqual(t, expected, actual)
 }
 
 func TestMaxOrderUpdateMarshaller(t *testing.T) {
@@ -83,7 +84,7 @@ func TestMaxOrderUpdateMarshaller(t *testing.T) {
 
 	fmt.Printf("%q", actual)
 
-	assert(t, expected, actual)
+	test.AssertEqual(t, expected, actual)
 }
 
 func TestMinOrderUpdateMarshaller(t *testing.T) {
@@ -102,5 +103,5 @@ func TestMinOrderUpdateMarshaller(t *testing.T) {
 
 	fmt.Printf("%q", actual)
 
-	assert(t, expected, actual)
+	test.AssertEqual(t, expected, actual)
 }