http_test.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879
  1. // Copyright 2017 The go-ethereum Authors
  2. // This file is part of the go-ethereum library.
  3. //
  4. // The go-ethereum library is free software: you can redistribute it and/or modify
  5. // it under the terms of the GNU Lesser General Public License as published by
  6. // the Free Software Foundation, either version 3 of the License, or
  7. // (at your option) any later version.
  8. //
  9. // The go-ethereum library is distributed in the hope that it will be useful,
  10. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. // GNU Lesser General Public License for more details.
  13. //
  14. // You should have received a copy of the GNU Lesser General Public License
  15. // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
  16. package simulations
  17. import (
  18. "context"
  19. "flag"
  20. "fmt"
  21. "math/rand"
  22. "net/http/httptest"
  23. "reflect"
  24. "sync"
  25. "sync/atomic"
  26. "testing"
  27. "time"
  28. "github.com/ethereum/go-ethereum/event"
  29. "github.com/ethereum/go-ethereum/log"
  30. "github.com/ethereum/go-ethereum/node"
  31. "github.com/ethereum/go-ethereum/p2p"
  32. "github.com/ethereum/go-ethereum/p2p/enode"
  33. "github.com/ethereum/go-ethereum/p2p/simulations/adapters"
  34. "github.com/ethereum/go-ethereum/rpc"
  35. "github.com/mattn/go-colorable"
  36. )
  37. var (
  38. loglevel = flag.Int("loglevel", 2, "verbosity of logs")
  39. )
  40. func init() {
  41. flag.Parse()
  42. log.PrintOrigins(true)
  43. log.Root().SetHandler(log.LvlFilterHandler(log.Lvl(*loglevel), log.StreamHandler(colorable.NewColorableStderr(), log.TerminalFormat(true))))
  44. }
  45. // testService implements the node.Service interface and provides protocols
  46. // and APIs which are useful for testing nodes in a simulation network
  47. type testService struct {
  48. id enode.ID
  49. // peerCount is incremented once a peer handshake has been performed
  50. peerCount int64
  51. peers map[enode.ID]*testPeer
  52. peersMtx sync.Mutex
  53. // state stores []byte which is used to test creating and loading
  54. // snapshots
  55. state atomic.Value
  56. }
  57. func newTestService(ctx *adapters.ServiceContext) (node.Service, error) {
  58. svc := &testService{
  59. id: ctx.Config.ID,
  60. peers: make(map[enode.ID]*testPeer),
  61. }
  62. svc.state.Store(ctx.Snapshot)
  63. return svc, nil
  64. }
  65. type testPeer struct {
  66. testReady chan struct{}
  67. dumReady chan struct{}
  68. }
  69. func (t *testService) peer(id enode.ID) *testPeer {
  70. t.peersMtx.Lock()
  71. defer t.peersMtx.Unlock()
  72. if peer, ok := t.peers[id]; ok {
  73. return peer
  74. }
  75. peer := &testPeer{
  76. testReady: make(chan struct{}),
  77. dumReady: make(chan struct{}),
  78. }
  79. t.peers[id] = peer
  80. return peer
  81. }
  82. func (t *testService) Protocols() []p2p.Protocol {
  83. return []p2p.Protocol{
  84. {
  85. Name: "test",
  86. Version: 1,
  87. Length: 3,
  88. Run: t.RunTest,
  89. },
  90. {
  91. Name: "dum",
  92. Version: 1,
  93. Length: 1,
  94. Run: t.RunDum,
  95. },
  96. {
  97. Name: "prb",
  98. Version: 1,
  99. Length: 1,
  100. Run: t.RunPrb,
  101. },
  102. }
  103. }
  104. func (t *testService) APIs() []rpc.API {
  105. return []rpc.API{{
  106. Namespace: "test",
  107. Version: "1.0",
  108. Service: &TestAPI{
  109. state: &t.state,
  110. peerCount: &t.peerCount,
  111. },
  112. }}
  113. }
  114. func (t *testService) Start(server *p2p.Server) error {
  115. return nil
  116. }
  117. func (t *testService) Stop() error {
  118. return nil
  119. }
  120. // handshake performs a peer handshake by sending and expecting an empty
  121. // message with the given code
  122. func (t *testService) handshake(rw p2p.MsgReadWriter, code uint64) error {
  123. errc := make(chan error, 2)
  124. go func() { errc <- p2p.Send(rw, code, struct{}{}) }()
  125. go func() { errc <- p2p.ExpectMsg(rw, code, struct{}{}) }()
  126. for i := 0; i < 2; i++ {
  127. if err := <-errc; err != nil {
  128. return err
  129. }
  130. }
  131. return nil
  132. }
  133. func (t *testService) RunTest(p *p2p.Peer, rw p2p.MsgReadWriter) error {
  134. peer := t.peer(p.ID())
  135. // perform three handshakes with three different message codes,
  136. // used to test message sending and filtering
  137. if err := t.handshake(rw, 2); err != nil {
  138. return err
  139. }
  140. if err := t.handshake(rw, 1); err != nil {
  141. return err
  142. }
  143. if err := t.handshake(rw, 0); err != nil {
  144. return err
  145. }
  146. // close the testReady channel so that other protocols can run
  147. close(peer.testReady)
  148. // track the peer
  149. atomic.AddInt64(&t.peerCount, 1)
  150. defer atomic.AddInt64(&t.peerCount, -1)
  151. // block until the peer is dropped
  152. for {
  153. _, err := rw.ReadMsg()
  154. if err != nil {
  155. return err
  156. }
  157. }
  158. }
  159. func (t *testService) RunDum(p *p2p.Peer, rw p2p.MsgReadWriter) error {
  160. peer := t.peer(p.ID())
  161. // wait for the test protocol to perform its handshake
  162. <-peer.testReady
  163. // perform a handshake
  164. if err := t.handshake(rw, 0); err != nil {
  165. return err
  166. }
  167. // close the dumReady channel so that other protocols can run
  168. close(peer.dumReady)
  169. // block until the peer is dropped
  170. for {
  171. _, err := rw.ReadMsg()
  172. if err != nil {
  173. return err
  174. }
  175. }
  176. }
  177. func (t *testService) RunPrb(p *p2p.Peer, rw p2p.MsgReadWriter) error {
  178. peer := t.peer(p.ID())
  179. // wait for the dum protocol to perform its handshake
  180. <-peer.dumReady
  181. // perform a handshake
  182. if err := t.handshake(rw, 0); err != nil {
  183. return err
  184. }
  185. // block until the peer is dropped
  186. for {
  187. _, err := rw.ReadMsg()
  188. if err != nil {
  189. return err
  190. }
  191. }
  192. }
  193. func (t *testService) Snapshot() ([]byte, error) {
  194. return t.state.Load().([]byte), nil
  195. }
  196. // TestAPI provides a test API to:
  197. // * get the peer count
  198. // * get and set an arbitrary state byte slice
  199. // * get and increment a counter
  200. // * subscribe to counter increment events
  201. type TestAPI struct {
  202. state *atomic.Value
  203. peerCount *int64
  204. counter int64
  205. feed event.Feed
  206. }
  207. func (t *TestAPI) PeerCount() int64 {
  208. return atomic.LoadInt64(t.peerCount)
  209. }
  210. func (t *TestAPI) Get() int64 {
  211. return atomic.LoadInt64(&t.counter)
  212. }
  213. func (t *TestAPI) Add(delta int64) {
  214. atomic.AddInt64(&t.counter, delta)
  215. t.feed.Send(delta)
  216. }
  217. func (t *TestAPI) GetState() []byte {
  218. return t.state.Load().([]byte)
  219. }
  220. func (t *TestAPI) SetState(state []byte) {
  221. t.state.Store(state)
  222. }
  223. func (t *TestAPI) Events(ctx context.Context) (*rpc.Subscription, error) {
  224. notifier, supported := rpc.NotifierFromContext(ctx)
  225. if !supported {
  226. return nil, rpc.ErrNotificationsUnsupported
  227. }
  228. rpcSub := notifier.CreateSubscription()
  229. go func() {
  230. events := make(chan int64)
  231. sub := t.feed.Subscribe(events)
  232. defer sub.Unsubscribe()
  233. for {
  234. select {
  235. case event := <-events:
  236. notifier.Notify(rpcSub.ID, event)
  237. case <-sub.Err():
  238. return
  239. case <-rpcSub.Err():
  240. return
  241. case <-notifier.Closed():
  242. return
  243. }
  244. }
  245. }()
  246. return rpcSub, nil
  247. }
  248. var testServices = adapters.Services{
  249. "test": newTestService,
  250. }
  251. func testHTTPServer(t *testing.T) (*Network, *httptest.Server) {
  252. t.Helper()
  253. adapter := adapters.NewSimAdapter(testServices)
  254. network := NewNetwork(adapter, &NetworkConfig{
  255. DefaultService: "test",
  256. })
  257. return network, httptest.NewServer(NewServer(network))
  258. }
  259. // TestHTTPNetwork tests interacting with a simulation network using the HTTP
  260. // API
  261. func TestHTTPNetwork(t *testing.T) {
  262. // start the server
  263. network, s := testHTTPServer(t)
  264. defer s.Close()
  265. // subscribe to events so we can check them later
  266. client := NewClient(s.URL)
  267. events := make(chan *Event, 100)
  268. var opts SubscribeOpts
  269. sub, err := client.SubscribeNetwork(events, opts)
  270. if err != nil {
  271. t.Fatalf("error subscribing to network events: %s", err)
  272. }
  273. defer sub.Unsubscribe()
  274. // check we can retrieve details about the network
  275. gotNetwork, err := client.GetNetwork()
  276. if err != nil {
  277. t.Fatalf("error getting network: %s", err)
  278. }
  279. if gotNetwork.ID != network.ID {
  280. t.Fatalf("expected network to have ID %q, got %q", network.ID, gotNetwork.ID)
  281. }
  282. // start a simulation network
  283. nodeIDs := startTestNetwork(t, client)
  284. // check we got all the events
  285. x := &expectEvents{t, events, sub}
  286. x.expect(
  287. x.nodeEvent(nodeIDs[0], false),
  288. x.nodeEvent(nodeIDs[1], false),
  289. x.nodeEvent(nodeIDs[0], true),
  290. x.nodeEvent(nodeIDs[1], true),
  291. x.connEvent(nodeIDs[0], nodeIDs[1], false),
  292. x.connEvent(nodeIDs[0], nodeIDs[1], true),
  293. )
  294. // reconnect the stream and check we get the current nodes and conns
  295. events = make(chan *Event, 100)
  296. opts.Current = true
  297. sub, err = client.SubscribeNetwork(events, opts)
  298. if err != nil {
  299. t.Fatalf("error subscribing to network events: %s", err)
  300. }
  301. defer sub.Unsubscribe()
  302. x = &expectEvents{t, events, sub}
  303. x.expect(
  304. x.nodeEvent(nodeIDs[0], true),
  305. x.nodeEvent(nodeIDs[1], true),
  306. x.connEvent(nodeIDs[0], nodeIDs[1], true),
  307. )
  308. }
  309. func startTestNetwork(t *testing.T, client *Client) []string {
  310. // create two nodes
  311. nodeCount := 2
  312. nodeIDs := make([]string, nodeCount)
  313. for i := 0; i < nodeCount; i++ {
  314. config := adapters.RandomNodeConfig()
  315. node, err := client.CreateNode(config)
  316. if err != nil {
  317. t.Fatalf("error creating node: %s", err)
  318. }
  319. nodeIDs[i] = node.ID
  320. }
  321. // check both nodes exist
  322. nodes, err := client.GetNodes()
  323. if err != nil {
  324. t.Fatalf("error getting nodes: %s", err)
  325. }
  326. if len(nodes) != nodeCount {
  327. t.Fatalf("expected %d nodes, got %d", nodeCount, len(nodes))
  328. }
  329. for i, nodeID := range nodeIDs {
  330. if nodes[i].ID != nodeID {
  331. t.Fatalf("expected node %d to have ID %q, got %q", i, nodeID, nodes[i].ID)
  332. }
  333. node, err := client.GetNode(nodeID)
  334. if err != nil {
  335. t.Fatalf("error getting node %d: %s", i, err)
  336. }
  337. if node.ID != nodeID {
  338. t.Fatalf("expected node %d to have ID %q, got %q", i, nodeID, node.ID)
  339. }
  340. }
  341. // start both nodes
  342. for _, nodeID := range nodeIDs {
  343. if err := client.StartNode(nodeID); err != nil {
  344. t.Fatalf("error starting node %q: %s", nodeID, err)
  345. }
  346. }
  347. // connect the nodes
  348. for i := 0; i < nodeCount-1; i++ {
  349. peerId := i + 1
  350. if i == nodeCount-1 {
  351. peerId = 0
  352. }
  353. if err := client.ConnectNode(nodeIDs[i], nodeIDs[peerId]); err != nil {
  354. t.Fatalf("error connecting nodes: %s", err)
  355. }
  356. }
  357. return nodeIDs
  358. }
  359. type expectEvents struct {
  360. *testing.T
  361. events chan *Event
  362. sub event.Subscription
  363. }
  364. func (t *expectEvents) nodeEvent(id string, up bool) *Event {
  365. node := Node{
  366. Config: &adapters.NodeConfig{
  367. ID: enode.HexID(id),
  368. },
  369. up: up,
  370. }
  371. return &Event{
  372. Type: EventTypeNode,
  373. Node: &node,
  374. }
  375. }
  376. func (t *expectEvents) connEvent(one, other string, up bool) *Event {
  377. return &Event{
  378. Type: EventTypeConn,
  379. Conn: &Conn{
  380. One: enode.HexID(one),
  381. Other: enode.HexID(other),
  382. Up: up,
  383. },
  384. }
  385. }
  386. func (t *expectEvents) expectMsgs(expected map[MsgFilter]int) {
  387. actual := make(map[MsgFilter]int)
  388. timeout := time.After(10 * time.Second)
  389. loop:
  390. for {
  391. select {
  392. case event := <-t.events:
  393. t.Logf("received %s event: %s", event.Type, event)
  394. if event.Type != EventTypeMsg || event.Msg.Received {
  395. continue loop
  396. }
  397. if event.Msg == nil {
  398. t.Fatal("expected event.Msg to be set")
  399. }
  400. filter := MsgFilter{
  401. Proto: event.Msg.Protocol,
  402. Code: int64(event.Msg.Code),
  403. }
  404. actual[filter]++
  405. if actual[filter] > expected[filter] {
  406. t.Fatalf("received too many msgs for filter: %v", filter)
  407. }
  408. if reflect.DeepEqual(actual, expected) {
  409. return
  410. }
  411. case err := <-t.sub.Err():
  412. t.Fatalf("network stream closed unexpectedly: %s", err)
  413. case <-timeout:
  414. t.Fatal("timed out waiting for expected events")
  415. }
  416. }
  417. }
  418. func (t *expectEvents) expect(events ...*Event) {
  419. t.Helper()
  420. timeout := time.After(10 * time.Second)
  421. i := 0
  422. for {
  423. select {
  424. case event := <-t.events:
  425. t.Logf("received %s event: %s", event.Type, event)
  426. expected := events[i]
  427. if event.Type != expected.Type {
  428. t.Fatalf("expected event %d to have type %q, got %q", i, expected.Type, event.Type)
  429. }
  430. switch expected.Type {
  431. case EventTypeNode:
  432. if event.Node == nil {
  433. t.Fatal("expected event.Node to be set")
  434. }
  435. if event.Node.ID() != expected.Node.ID() {
  436. t.Fatalf("expected node event %d to have id %q, got %q", i, expected.Node.ID().TerminalString(), event.Node.ID().TerminalString())
  437. }
  438. if event.Node.Up() != expected.Node.Up() {
  439. t.Fatalf("expected node event %d to have up=%t, got up=%t", i, expected.Node.Up(), event.Node.Up())
  440. }
  441. case EventTypeConn:
  442. if event.Conn == nil {
  443. t.Fatal("expected event.Conn to be set")
  444. }
  445. if event.Conn.One != expected.Conn.One {
  446. t.Fatalf("expected conn event %d to have one=%q, got one=%q", i, expected.Conn.One.TerminalString(), event.Conn.One.TerminalString())
  447. }
  448. if event.Conn.Other != expected.Conn.Other {
  449. t.Fatalf("expected conn event %d to have other=%q, got other=%q", i, expected.Conn.Other.TerminalString(), event.Conn.Other.TerminalString())
  450. }
  451. if event.Conn.Up != expected.Conn.Up {
  452. t.Fatalf("expected conn event %d to have up=%t, got up=%t", i, expected.Conn.Up, event.Conn.Up)
  453. }
  454. }
  455. i++
  456. if i == len(events) {
  457. return
  458. }
  459. case err := <-t.sub.Err():
  460. t.Fatalf("network stream closed unexpectedly: %s", err)
  461. case <-timeout:
  462. t.Fatal("timed out waiting for expected events")
  463. }
  464. }
  465. }
  466. // TestHTTPNodeRPC tests calling RPC methods on nodes via the HTTP API
  467. func TestHTTPNodeRPC(t *testing.T) {
  468. // start the server
  469. _, s := testHTTPServer(t)
  470. defer s.Close()
  471. // start a node in the network
  472. client := NewClient(s.URL)
  473. config := adapters.RandomNodeConfig()
  474. node, err := client.CreateNode(config)
  475. if err != nil {
  476. t.Fatalf("error creating node: %s", err)
  477. }
  478. if err := client.StartNode(node.ID); err != nil {
  479. t.Fatalf("error starting node: %s", err)
  480. }
  481. // create two RPC clients
  482. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  483. defer cancel()
  484. rpcClient1, err := client.RPCClient(ctx, node.ID)
  485. if err != nil {
  486. t.Fatalf("error getting node RPC client: %s", err)
  487. }
  488. rpcClient2, err := client.RPCClient(ctx, node.ID)
  489. if err != nil {
  490. t.Fatalf("error getting node RPC client: %s", err)
  491. }
  492. // subscribe to events using client 1
  493. events := make(chan int64, 1)
  494. sub, err := rpcClient1.Subscribe(ctx, "test", events, "events")
  495. if err != nil {
  496. t.Fatalf("error subscribing to events: %s", err)
  497. }
  498. defer sub.Unsubscribe()
  499. // call some RPC methods using client 2
  500. if err := rpcClient2.CallContext(ctx, nil, "test_add", 10); err != nil {
  501. t.Fatalf("error calling RPC method: %s", err)
  502. }
  503. var result int64
  504. if err := rpcClient2.CallContext(ctx, &result, "test_get"); err != nil {
  505. t.Fatalf("error calling RPC method: %s", err)
  506. }
  507. if result != 10 {
  508. t.Fatalf("expected result to be 10, got %d", result)
  509. }
  510. // check we got an event from client 1
  511. select {
  512. case event := <-events:
  513. if event != 10 {
  514. t.Fatalf("expected event to be 10, got %d", event)
  515. }
  516. case <-ctx.Done():
  517. t.Fatal(ctx.Err())
  518. }
  519. }
  520. // TestHTTPSnapshot tests creating and loading network snapshots
  521. func TestHTTPSnapshot(t *testing.T) {
  522. // start the server
  523. network, s := testHTTPServer(t)
  524. defer s.Close()
  525. var eventsDone = make(chan struct{})
  526. count := 1
  527. eventsDoneChan := make(chan *Event)
  528. eventSub := network.Events().Subscribe(eventsDoneChan)
  529. go func() {
  530. defer eventSub.Unsubscribe()
  531. for event := range eventsDoneChan {
  532. if event.Type == EventTypeConn && !event.Control {
  533. count--
  534. if count == 0 {
  535. eventsDone <- struct{}{}
  536. return
  537. }
  538. }
  539. }
  540. }()
  541. // create a two-node network
  542. client := NewClient(s.URL)
  543. nodeCount := 2
  544. nodes := make([]*p2p.NodeInfo, nodeCount)
  545. for i := 0; i < nodeCount; i++ {
  546. config := adapters.RandomNodeConfig()
  547. node, err := client.CreateNode(config)
  548. if err != nil {
  549. t.Fatalf("error creating node: %s", err)
  550. }
  551. if err := client.StartNode(node.ID); err != nil {
  552. t.Fatalf("error starting node: %s", err)
  553. }
  554. nodes[i] = node
  555. }
  556. if err := client.ConnectNode(nodes[0].ID, nodes[1].ID); err != nil {
  557. t.Fatalf("error connecting nodes: %s", err)
  558. }
  559. // store some state in the test services
  560. states := make([]string, nodeCount)
  561. for i, node := range nodes {
  562. rpc, err := client.RPCClient(context.Background(), node.ID)
  563. if err != nil {
  564. t.Fatalf("error getting RPC client: %s", err)
  565. }
  566. defer rpc.Close()
  567. state := fmt.Sprintf("%x", rand.Int())
  568. if err := rpc.Call(nil, "test_setState", []byte(state)); err != nil {
  569. t.Fatalf("error setting service state: %s", err)
  570. }
  571. states[i] = state
  572. }
  573. <-eventsDone
  574. // create a snapshot
  575. snap, err := client.CreateSnapshot()
  576. if err != nil {
  577. t.Fatalf("error creating snapshot: %s", err)
  578. }
  579. for i, state := range states {
  580. gotState := snap.Nodes[i].Snapshots["test"]
  581. if string(gotState) != state {
  582. t.Fatalf("expected snapshot state %q, got %q", state, gotState)
  583. }
  584. }
  585. // create another network
  586. network2, s := testHTTPServer(t)
  587. defer s.Close()
  588. client = NewClient(s.URL)
  589. count = 1
  590. eventSub = network2.Events().Subscribe(eventsDoneChan)
  591. go func() {
  592. defer eventSub.Unsubscribe()
  593. for event := range eventsDoneChan {
  594. if event.Type == EventTypeConn && !event.Control {
  595. count--
  596. if count == 0 {
  597. eventsDone <- struct{}{}
  598. return
  599. }
  600. }
  601. }
  602. }()
  603. // subscribe to events so we can check them later
  604. events := make(chan *Event, 100)
  605. var opts SubscribeOpts
  606. sub, err := client.SubscribeNetwork(events, opts)
  607. if err != nil {
  608. t.Fatalf("error subscribing to network events: %s", err)
  609. }
  610. defer sub.Unsubscribe()
  611. // load the snapshot
  612. if err := client.LoadSnapshot(snap); err != nil {
  613. t.Fatalf("error loading snapshot: %s", err)
  614. }
  615. <-eventsDone
  616. // check the nodes and connection exists
  617. net, err := client.GetNetwork()
  618. if err != nil {
  619. t.Fatalf("error getting network: %s", err)
  620. }
  621. if len(net.Nodes) != nodeCount {
  622. t.Fatalf("expected network to have %d nodes, got %d", nodeCount, len(net.Nodes))
  623. }
  624. for i, node := range nodes {
  625. id := net.Nodes[i].ID().String()
  626. if id != node.ID {
  627. t.Fatalf("expected node %d to have ID %s, got %s", i, node.ID, id)
  628. }
  629. }
  630. if len(net.Conns) != 1 {
  631. t.Fatalf("expected network to have 1 connection, got %d", len(net.Conns))
  632. }
  633. conn := net.Conns[0]
  634. if conn.One.String() != nodes[0].ID {
  635. t.Fatalf("expected connection to have one=%q, got one=%q", nodes[0].ID, conn.One)
  636. }
  637. if conn.Other.String() != nodes[1].ID {
  638. t.Fatalf("expected connection to have other=%q, got other=%q", nodes[1].ID, conn.Other)
  639. }
  640. if !conn.Up {
  641. t.Fatal("should be up")
  642. }
  643. // check the node states were restored
  644. for i, node := range nodes {
  645. rpc, err := client.RPCClient(context.Background(), node.ID)
  646. if err != nil {
  647. t.Fatalf("error getting RPC client: %s", err)
  648. }
  649. defer rpc.Close()
  650. var state []byte
  651. if err := rpc.Call(&state, "test_getState"); err != nil {
  652. t.Fatalf("error getting service state: %s", err)
  653. }
  654. if string(state) != states[i] {
  655. t.Fatalf("expected snapshot state %q, got %q", states[i], state)
  656. }
  657. }
  658. // check we got all the events
  659. x := &expectEvents{t, events, sub}
  660. x.expect(
  661. x.nodeEvent(nodes[0].ID, false),
  662. x.nodeEvent(nodes[0].ID, true),
  663. x.nodeEvent(nodes[1].ID, false),
  664. x.nodeEvent(nodes[1].ID, true),
  665. x.connEvent(nodes[0].ID, nodes[1].ID, false),
  666. x.connEvent(nodes[0].ID, nodes[1].ID, true),
  667. )
  668. }
  669. // TestMsgFilterPassMultiple tests streaming message events using a filter
  670. // with multiple protocols
  671. func TestMsgFilterPassMultiple(t *testing.T) {
  672. // start the server
  673. _, s := testHTTPServer(t)
  674. defer s.Close()
  675. // subscribe to events with a message filter
  676. client := NewClient(s.URL)
  677. events := make(chan *Event, 10)
  678. opts := SubscribeOpts{
  679. Filter: "prb:0-test:0",
  680. }
  681. sub, err := client.SubscribeNetwork(events, opts)
  682. if err != nil {
  683. t.Fatalf("error subscribing to network events: %s", err)
  684. }
  685. defer sub.Unsubscribe()
  686. // start a simulation network
  687. startTestNetwork(t, client)
  688. // check we got the expected events
  689. x := &expectEvents{t, events, sub}
  690. x.expectMsgs(map[MsgFilter]int{
  691. {"test", 0}: 2,
  692. {"prb", 0}: 2,
  693. })
  694. }
  695. // TestMsgFilterPassWildcard tests streaming message events using a filter
  696. // with a code wildcard
  697. func TestMsgFilterPassWildcard(t *testing.T) {
  698. // start the server
  699. _, s := testHTTPServer(t)
  700. defer s.Close()
  701. // subscribe to events with a message filter
  702. client := NewClient(s.URL)
  703. events := make(chan *Event, 10)
  704. opts := SubscribeOpts{
  705. Filter: "prb:0,2-test:*",
  706. }
  707. sub, err := client.SubscribeNetwork(events, opts)
  708. if err != nil {
  709. t.Fatalf("error subscribing to network events: %s", err)
  710. }
  711. defer sub.Unsubscribe()
  712. // start a simulation network
  713. startTestNetwork(t, client)
  714. // check we got the expected events
  715. x := &expectEvents{t, events, sub}
  716. x.expectMsgs(map[MsgFilter]int{
  717. {"test", 2}: 2,
  718. {"test", 1}: 2,
  719. {"test", 0}: 2,
  720. {"prb", 0}: 2,
  721. })
  722. }
  723. // TestMsgFilterPassSingle tests streaming message events using a filter
  724. // with a single protocol and code
  725. func TestMsgFilterPassSingle(t *testing.T) {
  726. // start the server
  727. _, s := testHTTPServer(t)
  728. defer s.Close()
  729. // subscribe to events with a message filter
  730. client := NewClient(s.URL)
  731. events := make(chan *Event, 10)
  732. opts := SubscribeOpts{
  733. Filter: "dum:0",
  734. }
  735. sub, err := client.SubscribeNetwork(events, opts)
  736. if err != nil {
  737. t.Fatalf("error subscribing to network events: %s", err)
  738. }
  739. defer sub.Unsubscribe()
  740. // start a simulation network
  741. startTestNetwork(t, client)
  742. // check we got the expected events
  743. x := &expectEvents{t, events, sub}
  744. x.expectMsgs(map[MsgFilter]int{
  745. {"dum", 0}: 2,
  746. })
  747. }
  748. // TestMsgFilterPassSingle tests streaming message events using an invalid
  749. // filter
  750. func TestMsgFilterFailBadParams(t *testing.T) {
  751. // start the server
  752. _, s := testHTTPServer(t)
  753. defer s.Close()
  754. client := NewClient(s.URL)
  755. events := make(chan *Event, 10)
  756. opts := SubscribeOpts{
  757. Filter: "foo:",
  758. }
  759. _, err := client.SubscribeNetwork(events, opts)
  760. if err == nil {
  761. t.Fatalf("expected event subscription to fail but succeeded!")
  762. }
  763. opts.Filter = "bzz:aa"
  764. _, err = client.SubscribeNetwork(events, opts)
  765. if err == nil {
  766. t.Fatalf("expected event subscription to fail but succeeded!")
  767. }
  768. opts.Filter = "invalid"
  769. _, err = client.SubscribeNetwork(events, opts)
  770. if err == nil {
  771. t.Fatalf("expected event subscription to fail but succeeded!")
  772. }
  773. }