39 lines
724 B
Go
39 lines
724 B
Go
|
package protocol
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/golang/mock/gomock"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Common resources
|
||
|
|
var (
|
||
|
|
mockCtrl *gomock.Controller
|
||
|
|
node *interfaces.MockNode
|
||
|
|
peer *net.MockPeer
|
||
|
|
services *interfaces.MockServices
|
||
|
|
p2p *interfaces.MockP2PService
|
||
|
|
)
|
||
|
|
|
||
|
|
// Setup function
|
||
|
|
func setup(t *testing.T) {
|
||
|
|
mockCtrl = gomock.NewController(t)
|
||
|
|
node = interfaces.NewMockNode(mockCtrl)
|
||
|
|
peer = net.NewMockPeer(mockCtrl)
|
||
|
|
services = interfaces.NewMockServices(mockCtrl)
|
||
|
|
p2p = interfaces.NewMockP2PService(mockCtrl)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Teardown function
|
||
|
|
func teardown() {
|
||
|
|
mockCtrl.Finish()
|
||
|
|
// Other cleanup tasks
|
||
|
|
}
|
||
|
|
|
||
|
|
// TestMain function for setup and teardown
|
||
|
|
func TestMain(m *testing.M) {
|
||
|
|
code := m.Run()
|
||
|
|
teardown()
|
||
|
|
os.Exit(code)
|
||
|
|
}
|