gopkg.in/dedis/onet.v2@v2.0.0-20181115163211-c8f3724038a7/simul/manage/close_all_test.go (about)

     1  package manage
     2  
     3  import (
     4  	"testing"
     5  	"time"
     6  
     7  	"gopkg.in/dedis/kyber.v2/suites"
     8  	"gopkg.in/dedis/onet.v2"
     9  	"gopkg.in/dedis/onet.v2/log"
    10  )
    11  
    12  var tSuite = suites.MustFind("Ed25519")
    13  
    14  func TestMain(m *testing.M) {
    15  	log.MainTest(m)
    16  }
    17  
    18  // Tests a 2-node system
    19  func TestCloseAll(t *testing.T) {
    20  	local := onet.NewLocalTest(tSuite)
    21  	nbrNodes := 2
    22  	_, _, tree := local.GenTree(nbrNodes, true)
    23  	defer local.CloseAll()
    24  
    25  	pi, err := local.CreateProtocol("CloseAll", tree)
    26  	if err != nil {
    27  		t.Fatal("Couldn't start protocol:", err)
    28  	}
    29  	done := make(chan bool)
    30  	go func() {
    31  		pi.Start()
    32  		done <- true
    33  	}()
    34  	select {
    35  	case <-done:
    36  	case <-time.After(10 * time.Second):
    37  		t.Fatal("Didn't finish in 10 seconds")
    38  	}
    39  }