update ethclient auction

eth/auction_test.go

@@ -4,36 +4,65 @@ import (
 	"os"
 	"testing"
 
+	"github.com/ethereum/go-ethereum/accounts/keystore"
 	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/crypto"
 	"github.com/ethereum/go-ethereum/ethclient"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
 
 const slotDeadlineConst = uint8(20)
-const openAuctionSlotsConst = 4320
-const closedAuctionSlotsConst = 2
-const outbiddingConst = 10
+const openAuctionSlotsConst = uint16(4320)
+const closedAuctionSlotsConst = uint16(2)
+const outbiddingConst = uint8(10)
 const currentSlotConst = 0
 
 var allocationRatioConst [3]uint8 = [3]uint8{40, 40, 20}
 
 var auctionClient *AuctionClient
 
-var donationAddressConstStr = os.Getenv("DONATION_ADDRESS")
-var bootCoordinatorConstStr = os.Getenv("BOOT_COORDINATOR_ADDRESS")
+/*var donationAddressStr = os.Getenv("DONATION_ADDRESS")
+var bootCoordinatorStr = os.Getenv("BOOT_COORDINATOR_ADDRESS")
 var integration = os.Getenv("INTEGRATION")
 var ehtClientDialURL = os.Getenv("ETHCLIENT_DIAL_URL")
-var auctionAddressStr = os.Getenv("AUCTION_ADDRESS")
+var auctionAddressStr = os.Getenv("AUCTION_ADDRESS")*/
+
+var donationAddressStr = "0x6c365935CA8710200C7595F0a72EB6023A7706Cd"
+var bootCoordinatorStr = "0xc783df8a850f42e7f7e57013759c285caa701eb6"
+var integration = os.Getenv("INTEGRATION")
+var ehtClientDialURL = "http://localhost:8545"
+var auctionAddressStr = "0x3619DbE27d7c1e7E91aA738697Ae7Bc5FC3eACA5"
+var ownerAddressStr = "0xc783df8a850f42e7F7e57013759C285caa701eB6"
+var governanceAddressStr = "0xead9c93b79ae7c1591b1fb5323bd777e86e150d4"
+var governancePrivateKey = "d49743deccbccc5dc7baa8e69e5be03298da8688a15dd202e20f15d5e0e9a9fb"
+var minBidStr = "10000000000000000000"
+var URL = "http://localhost:3000"
+var newURL = "http://localhost:3002"
+
+var pathKs = "test/ks"
+var password = "pass"
 
 func TestNewAction(t *testing.T) {
-	if integration != "" {
-		// Init eth client
-		ethClient, err := ethclient.Dial(ehtClientDialURL)
+	key, err := crypto.HexToECDSA(governancePrivateKey)
+	require.Nil(t, err)
+	ks := keystore.NewKeyStore(pathKs, keystore.StandardScryptN, keystore.StandardScryptP)
+	account, err := ks.ImportECDSA(key, password)
+	ks.Unlock(account, password)
+	// Init eth client
+	ethClient, err := ethclient.Dial(ehtClientDialURL)
+	require.Nil(t, err)
+	ethereumClient := NewEthereumClient(ethClient, &account, ks, nil)
+	auctionAddress := common.HexToAddress(auctionAddressStr)
+	auctionClient = NewAuctionClient(ethereumClient, auctionAddress)
+}
+
+func TestAuctionGetCurrentSlotNumber(t *testing.T) {
+	if auctionClient != nil {
+		currentSlot, err := auctionClient.AuctionGetCurrentSlotNumber()
 		require.Nil(t, err)
-		ethereumClient := NewEthereumClient(ethClient, nil, nil, nil)
-		auctionAddress := common.HexToAddress(auctionAddressStr)
-		auctionClient = NewAuctionClient(ethereumClient, auctionAddress)
+		currentSlotInt := int(currentSlot)
+		assert.Equal(t, currentSlotConst, currentSlotInt)
 	}
 }
 
@@ -45,12 +74,35 @@ func TestAuctionGetSlotDeadline(t *testing.T) {
 	}
 }
 
+func TestAuctionSetSlotDeadline(t *testing.T) {
+	newSlotDeadline := uint8(25)
+	if auctionClient != nil {
+		_, err := auctionClient.AuctionSetSlotDeadline(newSlotDeadline)
+		require.Nil(t, err)
+		slotDeadline, err := auctionClient.AuctionGetSlotDeadline()
+		assert.Equal(t, newSlotDeadline, slotDeadline)
+		_, err = auctionClient.AuctionSetSlotDeadline(slotDeadlineConst)
+		require.Nil(t, err)
+	}
+}
+
 func TestAuctionGetOpenAuctionSlots(t *testing.T) {
 	if auctionClient != nil {
 		openAuctionSlots, err := auctionClient.AuctionGetOpenAuctionSlots()
 		require.Nil(t, err)
-		openAuctionSlotsInt := int(openAuctionSlots)
-		assert.Equal(t, openAuctionSlotsConst, openAuctionSlotsInt)
+		assert.Equal(t, openAuctionSlotsConst, openAuctionSlots)
+	}
+}
+
+func TestAuctionSetOpenAuctionSlots(t *testing.T) {
+	newOpenAuctionSlots := uint16(4500)
+	if auctionClient != nil {
+		_, err := auctionClient.AuctionSetOpenAuctionSlots(newOpenAuctionSlots)
+		require.Nil(t, err)
+		openAuctionSlots, err := auctionClient.AuctionGetOpenAuctionSlots()
+		assert.Equal(t, newOpenAuctionSlots, openAuctionSlots)
+		_, err = auctionClient.AuctionSetOpenAuctionSlots(openAuctionSlotsConst)
+		require.Nil(t, err)
 	}
 }
 
@@ -58,8 +110,19 @@ func TestAuctionGetClosedAuctionSlots(t *testing.T) {
 	if auctionClient != nil {
 		closedAuctionSlots, err := auctionClient.AuctionGetClosedAuctionSlots()
 		require.Nil(t, err)
-		closedAuctionSlotsInt := int(closedAuctionSlots)
-		assert.Equal(t, closedAuctionSlotsConst, closedAuctionSlotsInt)
+		assert.Equal(t, closedAuctionSlotsConst, closedAuctionSlots)
+	}
+}
+
+func TestAuctionSetClosedAuctionSlots(t *testing.T) {
+	newClosedAuctionSlots := uint16(5)
+	if auctionClient != nil {
+		_, err := auctionClient.AuctionSetClosedAuctionSlots(newClosedAuctionSlots)
+		require.Nil(t, err)
+		closedAuctionSlots, err := auctionClient.AuctionGetClosedAuctionSlots()
+		assert.Equal(t, newClosedAuctionSlots, closedAuctionSlots)
+		_, err = auctionClient.AuctionSetClosedAuctionSlots(closedAuctionSlotsConst)
+		require.Nil(t, err)
 	}
 }
 
@@ -67,8 +130,19 @@ func TestAuctionGetOutbidding(t *testing.T) {
 	if auctionClient != nil {
 		outbidding, err := auctionClient.AuctionGetOutbidding()
 		require.Nil(t, err)
-		outbiddingInt := int(outbidding)
-		assert.Equal(t, outbiddingConst, outbiddingInt)
+		assert.Equal(t, outbiddingConst, outbidding)
+	}
+}
+
+func TestAuctionSetOutbidding(t *testing.T) {
+	newOutbidding := uint8(15)
+	if auctionClient != nil {
+		_, err := auctionClient.AuctionSetOutbidding(newOutbidding)
+		require.Nil(t, err)
+		outbidding, err := auctionClient.AuctionGetOutbidding()
+		assert.Equal(t, newOutbidding, outbidding)
+		_, err = auctionClient.AuctionSetOutbidding(outbiddingConst)
+		require.Nil(t, err)
 	}
 }
 
@@ -80,11 +154,23 @@ func TestAuctionGetAllocationRatio(t *testing.T) {
 	}
 }
 
+func TestAuctionSetAllocationRatio(t *testing.T) {
+	newAllocationRatio := [3]uint8{30, 30, 40}
+	if auctionClient != nil {
+		_, err := auctionClient.AuctionSetAllocationRatio(newAllocationRatio)
+		require.Nil(t, err)
+		allocationRatio, err := auctionClient.AuctionGetAllocationRatio()
+		assert.Equal(t, newAllocationRatio, allocationRatio)
+		_, err = auctionClient.AuctionSetAllocationRatio(allocationRatioConst)
+		require.Nil(t, err)
+	}
+}
+
 func TestAuctionGetDonationAddress(t *testing.T) {
 	if auctionClient != nil {
 		donationAddress, err := auctionClient.AuctionGetDonationAddress()
 		require.Nil(t, err)
-		donationAddressConst := common.HexToAddress(donationAddressConstStr)
+		donationAddressConst := common.HexToAddress(donationAddressStr)
 		assert.Equal(t, &donationAddressConst, donationAddress)
 	}
 }
@@ -93,16 +179,93 @@ func TestAuctionGetBootCoordinator(t *testing.T) {
 	if auctionClient != nil {
 		bootCoordinator, err := auctionClient.AuctionGetBootCoordinator()
 		require.Nil(t, err)
-		bootCoordinatorConst := common.HexToAddress(bootCoordinatorConstStr)
+		bootCoordinatorConst := common.HexToAddress(bootCoordinatorStr)
 		assert.Equal(t, &bootCoordinatorConst, bootCoordinator)
 	}
 }
 
-func TestAuctionGetCurrentSlotNumber(t *testing.T) {
+func TestAuctionSetDonationAddress(t *testing.T) {
+	newDonationAddress := common.HexToAddress(governanceAddressStr)
 	if auctionClient != nil {
-		currentSlot, err := auctionClient.AuctionGetCurrentSlotNumber()
+		_, err := auctionClient.AuctionSetDonationAddress(newDonationAddress)
+		require.Nil(t, err)
+		donationAddress, err := auctionClient.AuctionGetDonationAddress()
+		assert.Equal(t, &newDonationAddress, donationAddress)
+		donationAddressConst := common.HexToAddress(donationAddressStr)
+		_, err = auctionClient.AuctionSetDonationAddress(donationAddressConst)
+		require.Nil(t, err)
+	}
+}
+
+func TestAuctionSetBootCoordinator(t *testing.T) {
+	newBootCoordinator := common.HexToAddress(governanceAddressStr)
+	if auctionClient != nil {
+		_, err := auctionClient.AuctionSetBootCoordinator(newBootCoordinator)
+		require.Nil(t, err)
+		bootCoordinator, err := auctionClient.AuctionGetBootCoordinator()
+		assert.Equal(t, &newBootCoordinator, bootCoordinator)
+		bootCoordinatorConst := common.HexToAddress(bootCoordinatorStr)
+		_, err = auctionClient.AuctionSetBootCoordinator(bootCoordinatorConst)
+		require.Nil(t, err)
+	}
+}
+
+func TestAuctionGetMinBidEpoch(t *testing.T) {
+	epoch := uint8(3)
+	if auctionClient != nil {
+		minBid, err := auctionClient.AuctionGetMinBidEpoch(epoch)
+		require.Nil(t, err)
+		assert.Equal(t, minBid.String(), minBidStr)
+	}
+}
+
+/* func TestAuctionChangeEpochMinBid(t *testing.T) {
+	slotEpoch := int64(3)
+	epoch := uint8(3)
+	newInitialMinBid := new(big.Int)
+	newInitialMinBid.SetString("20000000000000000000", 10)
+	if auctionClient != nil {
+		_, err := auctionClient.AuctionChangeEpochMinBid(slotEpoch, newInitialMinBid)
+		require.Nil(t, err)
+		minBid, err := auctionClient.AuctionGetMinBidEpoch(epoch)
+		require.Nil(t, err)
+		assert.Equal(t, minBid, newInitialMinBid)
+		newMinBid := new(big.Int)
+		newMinBid.SetString("10000000000000000000", 10)
+		_, err = auctionClient.AuctionChangeEpochMinBid(slotEpoch, newMinBid)
+		require.Nil(t, err)
+	}
+} */
+func TestAuctionIsRegisteredCoordinator(t *testing.T) {
+	forgerAddress := common.HexToAddress(governanceAddressStr)
+	if auctionClient != nil {
+		registered, err := auctionClient.AuctionIsRegisteredCoordinator(forgerAddress)
+		require.Nil(t, err)
+		assert.Equal(t, registered, false)
+	}
+}
+
+func TestAuctionRegisterCoordinator(t *testing.T) {
+	forgerAddress := common.HexToAddress(governanceAddressStr)
+	if auctionClient != nil {
+		_, err := auctionClient.AuctionRegisterCoordinator(forgerAddress, URL)
+		require.Nil(t, err)
+	}
+}
+
+func TestAuctionIsRegisteredCoordinatorTrue(t *testing.T) {
+	forgerAddress := common.HexToAddress(governanceAddressStr)
+	if auctionClient != nil {
+		registered, err := auctionClient.AuctionIsRegisteredCoordinator(forgerAddress)
+		require.Nil(t, err)
+		assert.Equal(t, registered, true)
+	}
+}
+
+func AuctionUpdateCoordinatorInfo(t *testing.T) {
+	forgerAddress := common.HexToAddress(governanceAddressStr)
+	if auctionClient != nil {
+		_, err := auctionClient.AuctionUpdateCoordinatorInfo(forgerAddress, forgerAddress, newURL)
 		require.Nil(t, err)
-		currentSlotInt := int(currentSlot)
-		assert.Equal(t, currentSlotConst, currentSlotInt)
 	}
 }