Skip to content

Commit

Permalink
add test for dev flag
Browse files Browse the repository at this point in the history
  • Loading branch information
manav2401 committed Jan 25, 2022
1 parent 0662b70 commit 08f9931
Showing 1 changed file with 41 additions and 0 deletions.
41 changes: 41 additions & 0 deletions internal/cli/server/server_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
package server

import (
"testing"
"time"

"github.com/stretchr/testify/assert"
)

func TestServer_DeveloperMode(t *testing.T) {

// get the default config
config := DefaultConfig()

// enable developer mode
config.Developer.Enabled = true
config.Developer.Period = 2 // block time

// start the server
server, err1 := NewServer(config)
if err1 != nil {
t.Fatalf("failed to start server: %v", err1)
}

// record the initial block number
blockNumber := server.backend.BlockChain().CurrentBlock().Header().Number.Int64()

var i int64 = 0
for i = 0; i < 10; i++ {
// We expect the node to mine blocks every `config.Developer.Period` time period
time.Sleep(time.Duration(config.Developer.Period) * time.Second)
currBlock := server.backend.BlockChain().CurrentBlock().Header().Number.Int64()
expected := blockNumber + i + 1
if res := assert.Equal(t, currBlock, expected); res == false {
break
}
}

// stop the server
server.Stop()
}

0 comments on commit 08f9931

Please sign in to comment.