github.com/inphi/go-ethereum@v1.9.7/p2p/simulations/network_test.go (about) 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 17 package simulations 18 19 import ( 20 "bytes" 21 "context" 22 "encoding/json" 23 "fmt" 24 "reflect" 25 "strconv" 26 "strings" 27 "testing" 28 "time" 29 30 "github.com/ethereum/go-ethereum/log" 31 "github.com/ethereum/go-ethereum/node" 32 "github.com/ethereum/go-ethereum/p2p/enode" 33 "github.com/ethereum/go-ethereum/p2p/simulations/adapters" 34 ) 35 36 // Tests that a created snapshot with a minimal service only contains the expected connections 37 // and that a network when loaded with this snapshot only contains those same connections 38 func TestSnapshot(t *testing.T) { 39 40 // PART I 41 // create snapshot from ring network 42 43 // this is a minimal service, whose protocol will take exactly one message OR close of connection before quitting 44 adapter := adapters.NewSimAdapter(adapters.Services{ 45 "noopwoop": func(ctx *adapters.ServiceContext) (node.Service, error) { 46 return NewNoopService(nil), nil 47 }, 48 }) 49 50 // create network 51 network := NewNetwork(adapter, &NetworkConfig{ 52 DefaultService: "noopwoop", 53 }) 54 // \todo consider making a member of network, set to true threadsafe when shutdown 55 runningOne := true 56 defer func() { 57 if runningOne { 58 network.Shutdown() 59 } 60 }() 61 62 // create and start nodes 63 nodeCount := 20 64 ids := make([]enode.ID, nodeCount) 65 for i := 0; i < nodeCount; i++ { 66 conf := adapters.RandomNodeConfig() 67 node, err := network.NewNodeWithConfig(conf) 68 if err != nil { 69 t.Fatalf("error creating node: %s", err) 70 } 71 if err := network.Start(node.ID()); err != nil { 72 t.Fatalf("error starting node: %s", err) 73 } 74 ids[i] = node.ID() 75 } 76 77 // subscribe to peer events 78 evC := make(chan *Event) 79 sub := network.Events().Subscribe(evC) 80 defer sub.Unsubscribe() 81 82 // connect nodes in a ring 83 // spawn separate thread to avoid deadlock in the event listeners 84 go func() { 85 for i, id := range ids { 86 peerID := ids[(i+1)%len(ids)] 87 if err := network.Connect(id, peerID); err != nil { 88 t.Fatal(err) 89 } 90 } 91 }() 92 93 // collect connection events up to expected number 94 ctx, cancel := context.WithTimeout(context.TODO(), time.Second) 95 defer cancel() 96 checkIds := make(map[enode.ID][]enode.ID) 97 connEventCount := nodeCount 98 OUTER: 99 for { 100 select { 101 case <-ctx.Done(): 102 t.Fatal(ctx.Err()) 103 case ev := <-evC: 104 if ev.Type == EventTypeConn && !ev.Control { 105 106 // fail on any disconnect 107 if !ev.Conn.Up { 108 t.Fatalf("unexpected disconnect: %v -> %v", ev.Conn.One, ev.Conn.Other) 109 } 110 checkIds[ev.Conn.One] = append(checkIds[ev.Conn.One], ev.Conn.Other) 111 checkIds[ev.Conn.Other] = append(checkIds[ev.Conn.Other], ev.Conn.One) 112 connEventCount-- 113 log.Debug("ev", "count", connEventCount) 114 if connEventCount == 0 { 115 break OUTER 116 } 117 } 118 } 119 } 120 121 // create snapshot of current network 122 snap, err := network.Snapshot() 123 if err != nil { 124 t.Fatal(err) 125 } 126 j, err := json.Marshal(snap) 127 if err != nil { 128 t.Fatal(err) 129 } 130 log.Debug("snapshot taken", "nodes", len(snap.Nodes), "conns", len(snap.Conns), "json", string(j)) 131 132 // verify that the snap element numbers check out 133 if len(checkIds) != len(snap.Conns) || len(checkIds) != len(snap.Nodes) { 134 t.Fatalf("snapshot wrong node,conn counts %d,%d != %d", len(snap.Nodes), len(snap.Conns), len(checkIds)) 135 } 136 137 // shut down sim network 138 runningOne = false 139 sub.Unsubscribe() 140 network.Shutdown() 141 142 // check that we have all the expected connections in the snapshot 143 for nodid, nodConns := range checkIds { 144 for _, nodConn := range nodConns { 145 var match bool 146 for _, snapConn := range snap.Conns { 147 if snapConn.One == nodid && snapConn.Other == nodConn { 148 match = true 149 break 150 } else if snapConn.Other == nodid && snapConn.One == nodConn { 151 match = true 152 break 153 } 154 } 155 if !match { 156 t.Fatalf("snapshot missing conn %v -> %v", nodid, nodConn) 157 } 158 } 159 } 160 log.Info("snapshot checked") 161 162 // PART II 163 // load snapshot and verify that exactly same connections are formed 164 165 adapter = adapters.NewSimAdapter(adapters.Services{ 166 "noopwoop": func(ctx *adapters.ServiceContext) (node.Service, error) { 167 return NewNoopService(nil), nil 168 }, 169 }) 170 network = NewNetwork(adapter, &NetworkConfig{ 171 DefaultService: "noopwoop", 172 }) 173 defer func() { 174 network.Shutdown() 175 }() 176 177 // subscribe to peer events 178 // every node up and conn up event will generate one additional control event 179 // therefore multiply the count by two 180 evC = make(chan *Event, (len(snap.Conns)*2)+(len(snap.Nodes)*2)) 181 sub = network.Events().Subscribe(evC) 182 defer sub.Unsubscribe() 183 184 // load the snapshot 185 // spawn separate thread to avoid deadlock in the event listeners 186 err = network.Load(snap) 187 if err != nil { 188 t.Fatal(err) 189 } 190 191 // collect connection events up to expected number 192 ctx, cancel = context.WithTimeout(context.TODO(), time.Second*3) 193 defer cancel() 194 195 connEventCount = nodeCount 196 197 OuterTwo: 198 for { 199 select { 200 case <-ctx.Done(): 201 t.Fatal(ctx.Err()) 202 case ev := <-evC: 203 if ev.Type == EventTypeConn && !ev.Control { 204 205 // fail on any disconnect 206 if !ev.Conn.Up { 207 t.Fatalf("unexpected disconnect: %v -> %v", ev.Conn.One, ev.Conn.Other) 208 } 209 log.Debug("conn", "on", ev.Conn.One, "other", ev.Conn.Other) 210 checkIds[ev.Conn.One] = append(checkIds[ev.Conn.One], ev.Conn.Other) 211 checkIds[ev.Conn.Other] = append(checkIds[ev.Conn.Other], ev.Conn.One) 212 connEventCount-- 213 log.Debug("ev", "count", connEventCount) 214 if connEventCount == 0 { 215 break OuterTwo 216 } 217 } 218 } 219 } 220 221 // check that we have all expected connections in the network 222 for _, snapConn := range snap.Conns { 223 var match bool 224 for nodid, nodConns := range checkIds { 225 for _, nodConn := range nodConns { 226 if snapConn.One == nodid && snapConn.Other == nodConn { 227 match = true 228 break 229 } else if snapConn.Other == nodid && snapConn.One == nodConn { 230 match = true 231 break 232 } 233 } 234 } 235 if !match { 236 t.Fatalf("network missing conn %v -> %v", snapConn.One, snapConn.Other) 237 } 238 } 239 240 // verify that network didn't generate any other additional connection events after the ones we have collected within a reasonable period of time 241 ctx, cancel = context.WithTimeout(context.TODO(), time.Second) 242 defer cancel() 243 select { 244 case <-ctx.Done(): 245 case ev := <-evC: 246 if ev.Type == EventTypeConn { 247 t.Fatalf("Superfluous conn found %v -> %v", ev.Conn.One, ev.Conn.Other) 248 } 249 } 250 251 // This test validates if all connections from the snapshot 252 // are created in the network. 253 t.Run("conns after load", func(t *testing.T) { 254 // Create new network. 255 n := NewNetwork( 256 adapters.NewSimAdapter(adapters.Services{ 257 "noopwoop": func(ctx *adapters.ServiceContext) (node.Service, error) { 258 return NewNoopService(nil), nil 259 }, 260 }), 261 &NetworkConfig{ 262 DefaultService: "noopwoop", 263 }, 264 ) 265 defer n.Shutdown() 266 267 // Load the same snapshot. 268 err := n.Load(snap) 269 if err != nil { 270 t.Fatal(err) 271 } 272 273 // Check every connection from the snapshot 274 // if it is in the network, too. 275 for _, c := range snap.Conns { 276 if n.GetConn(c.One, c.Other) == nil { 277 t.Errorf("missing connection: %s -> %s", c.One, c.Other) 278 } 279 } 280 }) 281 } 282 283 // TestNetworkSimulation creates a multi-node simulation network with each node 284 // connected in a ring topology, checks that all nodes successfully handshake 285 // with each other and that a snapshot fully represents the desired topology 286 func TestNetworkSimulation(t *testing.T) { 287 // create simulation network with 20 testService nodes 288 adapter := adapters.NewSimAdapter(adapters.Services{ 289 "test": newTestService, 290 }) 291 network := NewNetwork(adapter, &NetworkConfig{ 292 DefaultService: "test", 293 }) 294 defer network.Shutdown() 295 nodeCount := 20 296 ids := make([]enode.ID, nodeCount) 297 for i := 0; i < nodeCount; i++ { 298 conf := adapters.RandomNodeConfig() 299 node, err := network.NewNodeWithConfig(conf) 300 if err != nil { 301 t.Fatalf("error creating node: %s", err) 302 } 303 if err := network.Start(node.ID()); err != nil { 304 t.Fatalf("error starting node: %s", err) 305 } 306 ids[i] = node.ID() 307 } 308 309 // perform a check which connects the nodes in a ring (so each node is 310 // connected to exactly two peers) and then checks that all nodes 311 // performed two handshakes by checking their peerCount 312 action := func(_ context.Context) error { 313 for i, id := range ids { 314 peerID := ids[(i+1)%len(ids)] 315 if err := network.Connect(id, peerID); err != nil { 316 return err 317 } 318 } 319 return nil 320 } 321 check := func(ctx context.Context, id enode.ID) (bool, error) { 322 // check we haven't run out of time 323 select { 324 case <-ctx.Done(): 325 return false, ctx.Err() 326 default: 327 } 328 329 // get the node 330 node := network.GetNode(id) 331 if node == nil { 332 return false, fmt.Errorf("unknown node: %s", id) 333 } 334 335 // check it has exactly two peers 336 client, err := node.Client() 337 if err != nil { 338 return false, err 339 } 340 var peerCount int64 341 if err := client.CallContext(ctx, &peerCount, "test_peerCount"); err != nil { 342 return false, err 343 } 344 switch { 345 case peerCount < 2: 346 return false, nil 347 case peerCount == 2: 348 return true, nil 349 default: 350 return false, fmt.Errorf("unexpected peerCount: %d", peerCount) 351 } 352 } 353 354 timeout := 30 * time.Second 355 ctx, cancel := context.WithTimeout(context.Background(), timeout) 356 defer cancel() 357 358 // trigger a check every 100ms 359 trigger := make(chan enode.ID) 360 go triggerChecks(ctx, ids, trigger, 100*time.Millisecond) 361 362 result := NewSimulation(network).Run(ctx, &Step{ 363 Action: action, 364 Trigger: trigger, 365 Expect: &Expectation{ 366 Nodes: ids, 367 Check: check, 368 }, 369 }) 370 if result.Error != nil { 371 t.Fatalf("simulation failed: %s", result.Error) 372 } 373 374 // take a network snapshot and check it contains the correct topology 375 snap, err := network.Snapshot() 376 if err != nil { 377 t.Fatal(err) 378 } 379 if len(snap.Nodes) != nodeCount { 380 t.Fatalf("expected snapshot to contain %d nodes, got %d", nodeCount, len(snap.Nodes)) 381 } 382 if len(snap.Conns) != nodeCount { 383 t.Fatalf("expected snapshot to contain %d connections, got %d", nodeCount, len(snap.Conns)) 384 } 385 for i, id := range ids { 386 conn := snap.Conns[i] 387 if conn.One != id { 388 t.Fatalf("expected conn[%d].One to be %s, got %s", i, id, conn.One) 389 } 390 peerID := ids[(i+1)%len(ids)] 391 if conn.Other != peerID { 392 t.Fatalf("expected conn[%d].Other to be %s, got %s", i, peerID, conn.Other) 393 } 394 } 395 } 396 397 func createTestNodes(count int, network *Network) (nodes []*Node, err error) { 398 for i := 0; i < count; i++ { 399 nodeConf := adapters.RandomNodeConfig() 400 node, err := network.NewNodeWithConfig(nodeConf) 401 if err != nil { 402 return nil, err 403 } 404 if err := network.Start(node.ID()); err != nil { 405 return nil, err 406 } 407 408 nodes = append(nodes, node) 409 } 410 411 return nodes, nil 412 } 413 414 func createTestNodesWithProperty(property string, count int, network *Network) (propertyNodes []*Node, err error) { 415 for i := 0; i < count; i++ { 416 nodeConf := adapters.RandomNodeConfig() 417 nodeConf.Properties = append(nodeConf.Properties, property) 418 419 node, err := network.NewNodeWithConfig(nodeConf) 420 if err != nil { 421 return nil, err 422 } 423 if err := network.Start(node.ID()); err != nil { 424 return nil, err 425 } 426 427 propertyNodes = append(propertyNodes, node) 428 } 429 430 return propertyNodes, nil 431 } 432 433 // TestGetNodeIDs creates a set of nodes and attempts to retrieve their IDs,. 434 // It then tests again whilst excluding a node ID from being returned. 435 // If a node ID is not returned, or more node IDs than expected are returned, the test fails. 436 func TestGetNodeIDs(t *testing.T) { 437 adapter := adapters.NewSimAdapter(adapters.Services{ 438 "test": newTestService, 439 }) 440 network := NewNetwork(adapter, &NetworkConfig{ 441 DefaultService: "test", 442 }) 443 defer network.Shutdown() 444 445 numNodes := 5 446 nodes, err := createTestNodes(numNodes, network) 447 if err != nil { 448 t.Fatalf("Could not creat test nodes %v", err) 449 } 450 451 gotNodeIDs := network.GetNodeIDs() 452 if len(gotNodeIDs) != numNodes { 453 t.Fatalf("Expected %d nodes, got %d", numNodes, len(gotNodeIDs)) 454 } 455 456 for _, node1 := range nodes { 457 match := false 458 for _, node2ID := range gotNodeIDs { 459 if bytes.Equal(node1.ID().Bytes(), node2ID.Bytes()) { 460 match = true 461 break 462 } 463 } 464 465 if !match { 466 t.Fatalf("A created node was not returned by GetNodes(), ID: %s", node1.ID().String()) 467 } 468 } 469 470 excludeNodeID := nodes[3].ID() 471 gotNodeIDsExcl := network.GetNodeIDs(excludeNodeID) 472 if len(gotNodeIDsExcl) != numNodes-1 { 473 t.Fatalf("Expected one less node ID to be returned") 474 } 475 for _, nodeID := range gotNodeIDsExcl { 476 if bytes.Equal(excludeNodeID.Bytes(), nodeID.Bytes()) { 477 t.Fatalf("GetNodeIDs returned the node ID we excluded, ID: %s", nodeID.String()) 478 } 479 } 480 } 481 482 // TestGetNodes creates a set of nodes and attempts to retrieve them again. 483 // It then tests again whilst excluding a node from being returned. 484 // If a node is not returned, or more nodes than expected are returned, the test fails. 485 func TestGetNodes(t *testing.T) { 486 adapter := adapters.NewSimAdapter(adapters.Services{ 487 "test": newTestService, 488 }) 489 network := NewNetwork(adapter, &NetworkConfig{ 490 DefaultService: "test", 491 }) 492 defer network.Shutdown() 493 494 numNodes := 5 495 nodes, err := createTestNodes(numNodes, network) 496 if err != nil { 497 t.Fatalf("Could not creat test nodes %v", err) 498 } 499 500 gotNodes := network.GetNodes() 501 if len(gotNodes) != numNodes { 502 t.Fatalf("Expected %d nodes, got %d", numNodes, len(gotNodes)) 503 } 504 505 for _, node1 := range nodes { 506 match := false 507 for _, node2 := range gotNodes { 508 if bytes.Equal(node1.ID().Bytes(), node2.ID().Bytes()) { 509 match = true 510 break 511 } 512 } 513 514 if !match { 515 t.Fatalf("A created node was not returned by GetNodes(), ID: %s", node1.ID().String()) 516 } 517 } 518 519 excludeNodeID := nodes[3].ID() 520 gotNodesExcl := network.GetNodes(excludeNodeID) 521 if len(gotNodesExcl) != numNodes-1 { 522 t.Fatalf("Expected one less node to be returned") 523 } 524 for _, node := range gotNodesExcl { 525 if bytes.Equal(excludeNodeID.Bytes(), node.ID().Bytes()) { 526 t.Fatalf("GetNodes returned the node we excluded, ID: %s", node.ID().String()) 527 } 528 } 529 } 530 531 // TestGetNodesByID creates a set of nodes and attempts to retrieve a subset of them by ID 532 // If a node is not returned, or more nodes than expected are returned, the test fails. 533 func TestGetNodesByID(t *testing.T) { 534 adapter := adapters.NewSimAdapter(adapters.Services{ 535 "test": newTestService, 536 }) 537 network := NewNetwork(adapter, &NetworkConfig{ 538 DefaultService: "test", 539 }) 540 defer network.Shutdown() 541 542 numNodes := 5 543 nodes, err := createTestNodes(numNodes, network) 544 if err != nil { 545 t.Fatalf("Could not create test nodes: %v", err) 546 } 547 548 numSubsetNodes := 2 549 subsetNodes := nodes[0:numSubsetNodes] 550 var subsetNodeIDs []enode.ID 551 for _, node := range subsetNodes { 552 subsetNodeIDs = append(subsetNodeIDs, node.ID()) 553 } 554 555 gotNodesByID := network.GetNodesByID(subsetNodeIDs) 556 if len(gotNodesByID) != numSubsetNodes { 557 t.Fatalf("Expected %d nodes, got %d", numSubsetNodes, len(gotNodesByID)) 558 } 559 560 for _, node1 := range subsetNodes { 561 match := false 562 for _, node2 := range gotNodesByID { 563 if bytes.Equal(node1.ID().Bytes(), node2.ID().Bytes()) { 564 match = true 565 break 566 } 567 } 568 569 if !match { 570 t.Fatalf("A created node was not returned by GetNodesByID(), ID: %s", node1.ID().String()) 571 } 572 } 573 } 574 575 // TestGetNodesByProperty creates a subset of nodes with a property assigned. 576 // GetNodesByProperty is then checked for correctness by comparing the nodes returned to those initially created. 577 // If a node with a property is not found, or more nodes than expected are returned, the test fails. 578 func TestGetNodesByProperty(t *testing.T) { 579 adapter := adapters.NewSimAdapter(adapters.Services{ 580 "test": newTestService, 581 }) 582 network := NewNetwork(adapter, &NetworkConfig{ 583 DefaultService: "test", 584 }) 585 defer network.Shutdown() 586 587 numNodes := 3 588 _, err := createTestNodes(numNodes, network) 589 if err != nil { 590 t.Fatalf("Failed to create nodes: %v", err) 591 } 592 593 numPropertyNodes := 3 594 propertyTest := "test" 595 propertyNodes, err := createTestNodesWithProperty(propertyTest, numPropertyNodes, network) 596 if err != nil { 597 t.Fatalf("Failed to create nodes with property: %v", err) 598 } 599 600 gotNodesByProperty := network.GetNodesByProperty(propertyTest) 601 if len(gotNodesByProperty) != numPropertyNodes { 602 t.Fatalf("Expected %d nodes with a property, got %d", numPropertyNodes, len(gotNodesByProperty)) 603 } 604 605 for _, node1 := range propertyNodes { 606 match := false 607 for _, node2 := range gotNodesByProperty { 608 if bytes.Equal(node1.ID().Bytes(), node2.ID().Bytes()) { 609 match = true 610 break 611 } 612 } 613 614 if !match { 615 t.Fatalf("A created node with property was not returned by GetNodesByProperty(), ID: %s", node1.ID().String()) 616 } 617 } 618 } 619 620 // TestGetNodeIDsByProperty creates a subset of nodes with a property assigned. 621 // GetNodeIDsByProperty is then checked for correctness by comparing the node IDs returned to those initially created. 622 // If a node ID with a property is not found, or more nodes IDs than expected are returned, the test fails. 623 func TestGetNodeIDsByProperty(t *testing.T) { 624 adapter := adapters.NewSimAdapter(adapters.Services{ 625 "test": newTestService, 626 }) 627 network := NewNetwork(adapter, &NetworkConfig{ 628 DefaultService: "test", 629 }) 630 defer network.Shutdown() 631 632 numNodes := 3 633 _, err := createTestNodes(numNodes, network) 634 if err != nil { 635 t.Fatalf("Failed to create nodes: %v", err) 636 } 637 638 numPropertyNodes := 3 639 propertyTest := "test" 640 propertyNodes, err := createTestNodesWithProperty(propertyTest, numPropertyNodes, network) 641 if err != nil { 642 t.Fatalf("Failed to created nodes with property: %v", err) 643 } 644 645 gotNodeIDsByProperty := network.GetNodeIDsByProperty(propertyTest) 646 if len(gotNodeIDsByProperty) != numPropertyNodes { 647 t.Fatalf("Expected %d nodes with a property, got %d", numPropertyNodes, len(gotNodeIDsByProperty)) 648 } 649 650 for _, node1 := range propertyNodes { 651 match := false 652 id1 := node1.ID() 653 for _, id2 := range gotNodeIDsByProperty { 654 if bytes.Equal(id1.Bytes(), id2.Bytes()) { 655 match = true 656 break 657 } 658 } 659 660 if !match { 661 t.Fatalf("Not all nodes IDs were returned by GetNodeIDsByProperty(), ID: %s", id1.String()) 662 } 663 } 664 } 665 666 func triggerChecks(ctx context.Context, ids []enode.ID, trigger chan enode.ID, interval time.Duration) { 667 tick := time.NewTicker(interval) 668 defer tick.Stop() 669 for { 670 select { 671 case <-tick.C: 672 for _, id := range ids { 673 select { 674 case trigger <- id: 675 case <-ctx.Done(): 676 return 677 } 678 } 679 case <-ctx.Done(): 680 return 681 } 682 } 683 } 684 685 // \todo: refactor to implement shapshots 686 // and connect configuration methods once these are moved from 687 // swarm/network/simulations/connect.go 688 func BenchmarkMinimalService(b *testing.B) { 689 b.Run("ring/32", benchmarkMinimalServiceTmp) 690 } 691 692 func benchmarkMinimalServiceTmp(b *testing.B) { 693 694 // stop timer to discard setup time pollution 695 args := strings.Split(b.Name(), "/") 696 nodeCount, err := strconv.ParseInt(args[2], 10, 16) 697 if err != nil { 698 b.Fatal(err) 699 } 700 701 for i := 0; i < b.N; i++ { 702 // this is a minimal service, whose protocol will close a channel upon run of protocol 703 // making it possible to bench the time it takes for the service to start and protocol actually to be run 704 protoCMap := make(map[enode.ID]map[enode.ID]chan struct{}) 705 adapter := adapters.NewSimAdapter(adapters.Services{ 706 "noopwoop": func(ctx *adapters.ServiceContext) (node.Service, error) { 707 protoCMap[ctx.Config.ID] = make(map[enode.ID]chan struct{}) 708 svc := NewNoopService(protoCMap[ctx.Config.ID]) 709 return svc, nil 710 }, 711 }) 712 713 // create network 714 network := NewNetwork(adapter, &NetworkConfig{ 715 DefaultService: "noopwoop", 716 }) 717 defer network.Shutdown() 718 719 // create and start nodes 720 ids := make([]enode.ID, nodeCount) 721 for i := 0; i < int(nodeCount); i++ { 722 conf := adapters.RandomNodeConfig() 723 node, err := network.NewNodeWithConfig(conf) 724 if err != nil { 725 b.Fatalf("error creating node: %s", err) 726 } 727 if err := network.Start(node.ID()); err != nil { 728 b.Fatalf("error starting node: %s", err) 729 } 730 ids[i] = node.ID() 731 } 732 733 // ready, set, go 734 b.ResetTimer() 735 736 // connect nodes in a ring 737 for i, id := range ids { 738 peerID := ids[(i+1)%len(ids)] 739 if err := network.Connect(id, peerID); err != nil { 740 b.Fatal(err) 741 } 742 } 743 744 // wait for all protocols to signal to close down 745 ctx, cancel := context.WithTimeout(context.TODO(), time.Second) 746 defer cancel() 747 for nodid, peers := range protoCMap { 748 for peerid, peerC := range peers { 749 log.Debug("getting ", "node", nodid, "peer", peerid) 750 select { 751 case <-ctx.Done(): 752 b.Fatal(ctx.Err()) 753 case <-peerC: 754 } 755 } 756 } 757 } 758 } 759 760 func TestNode_UnmarshalJSON(t *testing.T) { 761 t.Run( 762 "test unmarshal of Node up field", 763 func(t *testing.T) { 764 runNodeUnmarshalJSON(t, casesNodeUnmarshalJSONUpField()) 765 }, 766 ) 767 t.Run( 768 "test unmarshal of Node Config field", 769 func(t *testing.T) { 770 runNodeUnmarshalJSON(t, casesNodeUnmarshalJSONConfigField()) 771 }, 772 ) 773 } 774 775 func runNodeUnmarshalJSON(t *testing.T, tests []nodeUnmarshalTestCase) { 776 t.Helper() 777 for _, tt := range tests { 778 t.Run(tt.name, func(t *testing.T) { 779 var got Node 780 if err := got.UnmarshalJSON([]byte(tt.marshaled)); err != nil { 781 expectErrorMessageToContain(t, err, tt.wantErr) 782 } 783 expectNodeEquality(t, got, tt.want) 784 }) 785 } 786 } 787 788 type nodeUnmarshalTestCase struct { 789 name string 790 marshaled string 791 want Node 792 wantErr string 793 } 794 795 func expectErrorMessageToContain(t *testing.T, got error, want string) { 796 t.Helper() 797 if got == nil && want == "" { 798 return 799 } 800 801 if got == nil && want != "" { 802 t.Errorf("error was expected, got: nil, want: %v", want) 803 return 804 } 805 806 if !strings.Contains(got.Error(), want) { 807 t.Errorf( 808 "unexpected error message, got %v, want: %v", 809 want, 810 got, 811 ) 812 } 813 } 814 815 func expectNodeEquality(t *testing.T, got Node, want Node) { 816 t.Helper() 817 if !reflect.DeepEqual(got, want) { 818 t.Errorf("Node.UnmarshalJSON() = %v, want %v", got, want) 819 } 820 } 821 822 func casesNodeUnmarshalJSONUpField() []nodeUnmarshalTestCase { 823 return []nodeUnmarshalTestCase{ 824 { 825 name: "empty json", 826 marshaled: "{}", 827 want: Node{ 828 up: false, 829 }, 830 }, 831 { 832 name: "a stopped node", 833 marshaled: "{\"up\": false}", 834 want: Node{ 835 up: false, 836 }, 837 }, 838 { 839 name: "a running node", 840 marshaled: "{\"up\": true}", 841 want: Node{ 842 up: true, 843 }, 844 }, 845 { 846 name: "invalid JSON value on valid key", 847 marshaled: "{\"up\": foo}", 848 wantErr: "invalid character", 849 }, 850 { 851 name: "invalid JSON key and value", 852 marshaled: "{foo: bar}", 853 wantErr: "invalid character", 854 }, 855 { 856 name: "bool value expected but got something else (string)", 857 marshaled: "{\"up\": \"true\"}", 858 wantErr: "cannot unmarshal string into Go struct", 859 }, 860 } 861 } 862 863 func casesNodeUnmarshalJSONConfigField() []nodeUnmarshalTestCase { 864 // Don't do a big fuss around testing, as adapters.NodeConfig should 865 // handle it's own serialization. Just do a sanity check. 866 return []nodeUnmarshalTestCase{ 867 { 868 name: "Config field is omitted", 869 marshaled: "{}", 870 want: Node{ 871 Config: nil, 872 }, 873 }, 874 { 875 name: "Config field is nil", 876 marshaled: "{\"config\": nil}", 877 want: Node{ 878 Config: nil, 879 }, 880 }, 881 { 882 name: "a non default Config field", 883 marshaled: "{\"config\":{\"name\":\"node_ecdd0\",\"port\":44665}}", 884 want: Node{ 885 Config: &adapters.NodeConfig{ 886 Name: "node_ecdd0", 887 Port: 44665, 888 }, 889 }, 890 }, 891 } 892 }