github.com/hzck/speedroute@v0.0.0-20201115191102-403b7d0e443f/model/edge_test.go (about)

     1  package model
     2  
     3  import (
     4  	"fmt"
     5  	"testing"
     6  )
     7  
     8  // TestNodeFromEdges tests that when creating an edge, from and to node gets correctly populated.
     9  func TestNodeFromEdges(t *testing.T) {
    10  	startNode := new(Node)
    11  	endNode := new(Node)
    12  	CreateEdge(startNode, endNode)
    13  	if len(startNode.FromEdges()) != 1 {
    14  		t.Fail()
    15  		fmt.Println("No start node from edge")
    16  	}
    17  	if len(endNode.ToEdges()) != 1 {
    18  		t.Fail()
    19  		fmt.Println("No end node to edge")
    20  	}
    21  }