add test

node/contract.go

@@ -116,6 +116,8 @@ func (node *Node) AddLotteryContract() {
 	priKey, err := crypto.HexToECDSA(contract_constants.DemoAccounts[0].Private)
 	if err != nil {
 		utils.GetLogInstance().Error("Error when creating private key for demo contract")
+		// Exit here to recognize the coding working.
+		// Basically we will remove this logic when launching so it's fine for now.
 		os.Exit(1)
 	}
 

node/contract_test.go

@@ -0,0 +1,32 @@
+package node
+
+import (
+	"testing"
+
+	"github.com/harmony-one/harmony/consensus"
+	"github.com/harmony-one/harmony/internal/utils"
+	"github.com/harmony-one/harmony/p2p"
+	"github.com/harmony-one/harmony/p2p/p2pimpl"
+)
+
+func prepareNode(t *testing.T) *Node {
+	_, pubKey := utils.GenKey("1", "2")
+	leader := p2p.Peer{IP: "127.0.0.1", Port: "8882", ConsensusPubKey: pubKey}
+	validator := p2p.Peer{IP: "127.0.0.1", Port: "8885"}
+	priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
+	host, err := p2pimpl.NewHost(&leader, priKey)
+	if err != nil {
+		t.Fatalf("newhost failure: %v", err)
+	}
+	consensus := consensus.New(host, "0", []p2p.Peer{leader, validator}, leader)
+	return New(host, consensus, nil)
+
+}
+
+func TestAddLotteryContract(t *testing.T) {
+	node := prepareNode(t)
+	node.AddLotteryContract()
+	if len(node.DemoContractAddress) == 0 {
+		t.Error("Can not create demo contract")
+	}
+}