account tests

Author: gregorgololicic

go.mod

@@ -10,5 +10,6 @@ require (
 	github.com/onflow/flow-go-sdk v0.13.1
 	github.com/spf13/cobra v1.1.1
 	github.com/spf13/viper v1.7.0
+	github.com/stretchr/testify v1.5.1
 	google.golang.org/grpc v1.31.1
 )

go.sum

@@ -342,6 +342,7 @@ github.com/status-im/keycard-go v0.0.0-20190316090335-8537d3370df4/go.mod h1:RZL
 github.com/steakknife/bloomfilter v0.0.0-20180922174646-6819c0d2a570/go.mod h1:8OR4w3TdeIHIh1g6EMY5p0gVNOovcWC+1vpc7naMuAw=
 github.com/steakknife/hamming v0.0.0-20180906055917-c99c65617cd3/go.mod h1:hpGUWaI9xL8pRQCTXQgocU38Qw1g0Us7n5PxxTwTCYU=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/objx v0.1.1 h1:2vfRuCMp5sSVIDSqO8oNnWJq7mPa6KVP3iPIwFBuy8A=
 github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
 github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=

test/account_test.go

@@ -0,0 +1,54 @@
+package test
+
+import (
+	"testing"
+
+	"github.com/onflow/flow-go-sdk/crypto"
+	"github.com/stretchr/testify/assert"
+
+	"github.com/onflow/flow-go-sdk"
+	"github.com/sideninja/flow-cli/models"
+)
+
+var sigAlgo = crypto.StringToSignatureAlgorithm("ECDSA_P256")
+var pubKey, _ = crypto.DecodePublicKeyHex(sigAlgo, "0a4c9f7bf0c8adf6ebdf4859c11d8e2867e0eaa4af6880e18a0730a0b44a494e976cefa0caf8efb7ec2da469c3f320dab4a2ca72fb340621776f4a1403ae39ed")
+
+var flowAccount = flow.Account{
+	Address: flow.HexToAddress("01cf0e2f2f715450"),
+	Balance: 10,
+	Code:    nil,
+	Keys: []*flow.AccountKey{
+		&flow.AccountKey{
+			Index:          1,
+			PublicKey:      pubKey,
+			SigAlgo:        sigAlgo,
+			HashAlgo:       crypto.HashAlgorithm(1),
+			Weight:         1000,
+			SequenceNumber: 1,
+			Revoked:        false,
+		},
+	},
+}
+
+var account = &models.Account{&flowAccount}
+
+func TestAccountEncapsulation(t *testing.T) {
+	assert.Equal(t, account.Address.String(), "01cf0e2f2f715450")
+	assert.Equal(t, int(account.Balance), 10)
+}
+
+func TestAccountStringSerialization(t *testing.T) {
+
+	assert.Equal(t, account.String("address"), "01cf0e2f2f715450")
+	assert.Equal(t, account.String("balance"), "10")
+	assert.Equal(t, account.String("keys"), "\nKey 0\tPublic Key\t\t 0a4c9f7bf0c8adf6ebdf4859c11d8e2867e0eaa4af6880e18a0730a0b44a494e976cefa0caf8efb7ec2da469c3f320dab4a2ca72fb340621776f4a1403ae39ed\n\tWeight\t\t\t 1000\n\tSignature Algorithm\t ECDSA_P256\n\tHash Algorithm\t\t SHA2_256\n\n")
+
+	assert.Equal(t, account.String(""), "\nAddress\t 01cf0e2f2f715450\nBalance\t 10\nKeys\t 1\n\nKey 0\tPublic Key\t\t 0a4c9f7bf0c8adf6ebdf4859c11d8e2867e0eaa4af6880e18a0730a0b44a494e976cefa0caf8efb7ec2da469c3f320dab4a2ca72fb340621776f4a1403ae39ed\n\tWeight\t\t\t 1000\n\tSignature Algorithm\t ECDSA_P256\n\tHash Algorithm\t\t SHA2_256\n\n\nCode\t\t \n")
+}
+
+func TestAccountJSONSerialization(t *testing.T) {
+	assert.Equal(t, account.JSON(""), "{\"Address\":\"01cf0e2f2f715450\",\"Balance\":10,\"Code\":null,\"Keys\":[{\"Index\":1,\"PublicKey\":{},\"SigAlgo\":2,\"HashAlgo\":1,\"Weight\":1000,\"SequenceNumber\":1,\"Revoked\":false}],\"Contracts\":null}")
+	assert.Equal(t, account.JSON("balance"), "{\"Balance\":10}")
+	assert.Equal(t, account.JSON("address"), "{\"Address\":\"01cf0e2f2f715450\"}")
+	assert.Equal(t, account.JSON("keys"), "[{\"Index\":1,\"PublicKey\":{},\"SigAlgo\":2,\"HashAlgo\":1,\"Weight\":1000,\"SequenceNumber\":1,\"Revoked\":false}]")
+}