1package api
  2
  3import (
  4	"testing"
  5	"time"
  6
  7	"github.com/hermeznetwork/hermez-node/common"
  8	"github.com/stretchr/testify/assert"
  9)
 10
 11const secondsPerBlock = 15
 12
 13func TestSetRollupVariables(t *testing.T) {
 14	rollupVars := &common.RollupVariables{}
 15	assert.Equal(t, *rollupVars, api.status.Rollup)
 16	api.SetRollupVariables(tc.rollupVars)
 17	assert.Equal(t, tc.rollupVars, api.status.Rollup)
 18}
 19
 20func TestSetWDelayerVariables(t *testing.T) {
 21	wdelayerVars := &common.WDelayerVariables{}
 22	assert.Equal(t, *wdelayerVars, api.status.WithdrawalDelayer)
 23	api.SetWDelayerVariables(tc.wdelayerVars)
 24	assert.Equal(t, tc.wdelayerVars, api.status.WithdrawalDelayer)
 25}
 26
 27func TestSetAuctionVariables(t *testing.T) {
 28	auctionVars := &common.AuctionVariables{}
 29	assert.Equal(t, *auctionVars, api.status.Auction)
 30	api.SetAuctionVariables(tc.auctionVars)
 31	assert.Equal(t, tc.auctionVars, api.status.Auction)
 32}
 33
 34func TestNextForgers(t *testing.T) {
 35	// It's assumed that bids for each slot will be received in increasing order
 36	bestBids := make(map[int64]testBid)
 37	for j := range tc.bids {
 38		bestBids[tc.bids[j].SlotNum] = tc.bids[j]
 39	}
 40	lastBlock := tc.blocks[len(tc.blocks)-1]
 41	for i := int64(0); i < tc.slots[len(tc.slots)-1].SlotNum; i++ {
 42		lastClosedSlot := i + int64(api.status.Auction.ClosedAuctionSlots)
 43		nextForgers, err := api.GetNextForgers(tc.blocks[len(tc.blocks)-1], i, lastClosedSlot)
 44		assert.NoError(t, err)
 45		for j := i; j <= lastClosedSlot; j++ {
 46			for q := range nextForgers {
 47				if nextForgers[q].Period.SlotNum == j {
 48					if nextForgers[q].Coordinator.ItemID != 0 {
 49						assert.Equal(t, bestBids[j].Bidder, nextForgers[q].Coordinator.Bidder)
 50					} else {
 51						assert.Equal(t, bootCoordinator.Bidder, nextForgers[q].Coordinator.Bidder)
 52					}
 53					firstBlockSlot, lastBlockSlot := api.getFirstLastBlock(j)
 54					fromTimestamp := lastBlock.Timestamp.Add(time.Second * time.Duration(secondsPerBlock*(firstBlockSlot-lastBlock.EthBlockNum)))
 55					toTimestamp := lastBlock.Timestamp.Add(time.Second * time.Duration(secondsPerBlock*(lastBlockSlot-lastBlock.EthBlockNum)))
 56					assert.Equal(t, fromTimestamp.Unix(), nextForgers[q].Period.FromTimestamp.Unix())
 57					assert.Equal(t, toTimestamp.Unix(), nextForgers[q].Period.ToTimestamp.Unix())
 58				}
 59			}
 60		}
 61	}
 62}
 63
 64func TestUpdateNetworkInfo(t *testing.T) {
 65	status := &Network{}
 66	assert.Equal(t, status.LastBlock, api.status.Network.LastBlock)
 67	assert.Equal(t, status.LastBatch.BatchNum, api.status.Network.LastBatch.BatchNum)
 68	assert.Equal(t, status.CurrentSlot, api.status.Network.CurrentSlot)
 69	assert.Equal(t, status.NextForgers, api.status.Network.NextForgers)
 70	lastBlock := tc.blocks[3]
 71	lastBatchNum := common.BatchNum(3)
 72	currentSlotNum := int64(1)
 73	err := api.UpdateNetworkInfo(lastBlock, lastBatchNum, currentSlotNum)
 74	assert.NoError(t, err)
 75	assert.Equal(t, lastBlock.EthBlockNum, api.status.Network.LastBlock)
 76	assert.Equal(t, lastBatchNum, api.status.Network.LastBatch.BatchNum)
 77	assert.Equal(t, currentSlotNum, api.status.Network.CurrentSlot)
 78	assert.Equal(t, int(api.status.Auction.ClosedAuctionSlots)+1, len(api.status.Network.NextForgers))
 79}
 80
 81func TestUpdateMetrics(t *testing.T) {
 82	// TODO: Improve checks when til is integrated
 83	// Update Metrics needs api.status.Network.LastBatch.BatchNum to be updated
 84	lastBlock := tc.blocks[3]
 85	lastBatchNum := common.BatchNum(3)
 86	currentSlotNum := int64(1)
 87	err := api.UpdateNetworkInfo(lastBlock, lastBatchNum, currentSlotNum)
 88	assert.NoError(t, err)
 89
 90	err = api.UpdateMetrics()
 91	assert.NoError(t, err)
 92	assert.Greater(t, api.status.Metrics.TransactionsPerBatch, float64(0))
 93	assert.Greater(t, api.status.Metrics.BatchFrequency, float64(0))
 94	assert.Greater(t, api.status.Metrics.TransactionsPerBatch, float64(0))
 95	assert.Greater(t, api.status.Metrics.TotalAccounts, int64(0))
 96	assert.Greater(t, api.status.Metrics.TotalBJJs, int64(0))
 97	assert.Greater(t, api.status.Metrics.AvgTransactionFee, float64(0))
 98}
 99
100func TestUpdateRecommendedFee(t *testing.T) {
101	err := api.UpdateRecommendedFee()
102	assert.NoError(t, err)
103	assert.Greater(t, api.status.RecommendedFee.ExistingAccount, float64(0))
104	assert.Equal(t, api.status.RecommendedFee.CreatesAccount, api.status.RecommendedFee.ExistingAccount*createAccountExtraFeePercentage)
105	assert.Equal(t, api.status.RecommendedFee.CreatesAccountAndRegister, api.status.RecommendedFee.ExistingAccount*createAccountInternalExtraFeePercentage)
106}