utils_test.go 2.72 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
// Copyright 2015 The go-ethereum Authors
// This file is part of the go-ethereum library.
//
// The go-ethereum library is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The go-ethereum library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.

// Contains a batch of utility type declarations used by the tests. As the node
// operates on unique types, a lot of them are needed to check various features.

package node

import (
	"github.com/ethereum/go-ethereum/p2p"
24
	"github.com/ethereum/go-ethereum/rpc"
25 26
)

27 28
// NoopLifecycle is a trivial implementation of the Service interface.
type NoopLifecycle struct{}
29

30 31
func (s *NoopLifecycle) Start() error { return nil }
func (s *NoopLifecycle) Stop() error  { return nil }
32

33 34 35 36
func NewNoop() *Noop {
	noop := new(Noop)
	return noop
}
37

38
// Set of services all wrapping the base NoopLifecycle resulting in the same method
39
// signatures but different outer types.
40
type Noop struct{ NoopLifecycle }
41

42
// InstrumentedService is an implementation of Lifecycle for which all interface
43 44
// methods can be instrumented both return value as well as event hook wise.
type InstrumentedService struct {
45 46
	start error
	stop  error
47

48 49
	startHook func()
	stopHook  func()
50

51
	protocols []p2p.Protocol
52 53
}

54
func (s *InstrumentedService) Start() error {
55
	if s.startHook != nil {
56
		s.startHook()
57 58 59 60 61 62 63 64 65 66 67
	}
	return s.start
}

func (s *InstrumentedService) Stop() error {
	if s.stopHook != nil {
		s.stopHook()
	}
	return s.stop
}

68
type FullService struct{}
69

70 71
func NewFullService(stack *Node) (*FullService, error) {
	fs := new(FullService)
72

73 74 75 76
	stack.RegisterProtocols(fs.Protocols())
	stack.RegisterAPIs(fs.APIs())
	stack.RegisterLifecycle(fs)
	return fs, nil
77 78
}

79
func (f *FullService) Start() error { return nil }
80

81
func (f *FullService) Stop() error { return nil }
82

83 84 85 86 87 88 89 90 91 92 93
func (f *FullService) Protocols() []p2p.Protocol {
	return []p2p.Protocol{
		p2p.Protocol{
			Name:    "test1",
			Version: uint(1),
		},
		p2p.Protocol{
			Name:    "test2",
			Version: uint(2),
		},
	}
94 95
}

96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111
func (f *FullService) APIs() []rpc.API {
	return []rpc.API{
		{
			Namespace: "admin",
			Version:   "1.0",
		},
		{
			Namespace: "debug",
			Version:   "1.0",
			Public:    true,
		},
		{
			Namespace: "net",
			Version:   "1.0",
			Public:    true,
		},
112 113
	}
}