From 5c9426302691a8d3c3f78e60878c289517425e34 Mon Sep 17 00:00:00 2001
From: Rongjian Lan <rongjian.lan@gmail.com>
Date: Thu, 3 Jan 2019 12:33:59 -0800
Subject: [PATCH] Add tests for client service

---
 api/client/service/server_test.go | 84 +++++++++++++++++++++++++++++++
 1 file changed, 84 insertions(+)
 create mode 100644 api/client/service/server_test.go

diff --git a/api/client/service/server_test.go b/api/client/service/server_test.go
new file mode 100644
index 000000000..2ddd5c005
--- /dev/null
+++ b/api/client/service/server_test.go
@@ -0,0 +1,84 @@
+package client
+
+import (
+	"bytes"
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/crypto"
+	"github.com/harmony-one/harmony/api/client/service/proto"
+	"github.com/harmony-one/harmony/core/state"
+	"math/big"
+	"testing"
+
+	"github.com/ethereum/go-ethereum/params"
+	"github.com/harmony-one/harmony/consensus"
+	"github.com/harmony-one/harmony/core"
+	"github.com/harmony-one/harmony/core/vm"
+	"github.com/harmony-one/harmony/internal/db"
+)
+
+var (
+	// Test accounts
+	testBankKey, _  = crypto.GenerateKey()
+	testBankAddress = crypto.PubkeyToAddress(testBankKey.PublicKey)
+	testBankFunds   = big.NewInt(8000000000000000000)
+
+	chainConfig = params.TestChainConfig
+)
+
+func TestGetFreeToken(test *testing.T) {
+	hash := common.Hash{}
+	hash.SetBytes([]byte("hello"))
+	server := NewServer(func() (*state.StateDB, error) {
+		return nil, nil
+	}, func(common.Address) common.Hash {
+		return hash
+	})
+
+	testBankKey, _ := crypto.GenerateKey()
+	testBankAddress := crypto.PubkeyToAddress(testBankKey.PublicKey)
+	response, err := server.GetFreeToken(nil, &client.GetFreeTokenRequest{Address: testBankAddress.Bytes()})
+
+	if err != nil {
+		test.Errorf("Failed to get free token")
+	}
+	if bytes.Compare(response.TxId, hash.Bytes()) != 0 {
+		test.Errorf("Wrong transaction id is returned")
+	}
+}
+
+func TestFetchAccountState(test *testing.T) {
+	var (
+		database = db.NewMemDatabase()
+		gspec    = core.Genesis{
+			Config:  chainConfig,
+			Alloc:   core.GenesisAlloc{testBankAddress: {Balance: testBankFunds}},
+			ShardID: 10,
+		}
+	)
+
+	genesis := gspec.MustCommit(database)
+	_ = genesis
+	chain, _ := core.NewBlockChain(database, nil, gspec.Config, consensus.NewFaker(), vm.Config{}, nil)
+
+	hash := common.Hash{}
+	hash.SetBytes([]byte("hello"))
+	server := NewServer(func() (*state.StateDB, error) {
+		return chain.State()
+	}, func(common.Address) common.Hash {
+		return hash
+	})
+
+	response, err := server.FetchAccountState(nil, &client.FetchAccountStateRequest{Address: testBankAddress.Bytes()})
+
+	if err != nil {
+		test.Errorf("Failed to get free token")
+	}
+
+	if bytes.Compare(response.Balance, testBankFunds.Bytes()) != 0 {
+		test.Errorf("Wrong balance is returned")
+	}
+
+	if response.Nonce != 0 {
+		test.Errorf("Wrong nonce is returned")
+	}
+}