From 23138d395000a322bd3e9218fa87d26f4c80fdba Mon Sep 17 00:00:00 2001 From: severindellsperger Date: Mon, 15 Jul 2024 19:19:29 +0200 Subject: [PATCH] add missing test case --- pkg/graph/network_graph.go | 4 +++- pkg/graph/network_graph_test.go | 4 ++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/pkg/graph/network_graph.go b/pkg/graph/network_graph.go index 2a4126e..ac95171 100644 --- a/pkg/graph/network_graph.go +++ b/pkg/graph/network_graph.go @@ -113,7 +113,9 @@ func (graph *NetworkGraph) addNodesToSubgraph(newSubGraphs map[uint32]*NetworkGr func (graph *NetworkGraph) addEdgesToSubgraph(newSubGraphs map[uint32]*NetworkGraph) { for _, edge := range graph.edges { for flexAlgo := range edge.GetFlexibleAlgorithms() { - _ = newSubGraphs[flexAlgo].AddEdge(edge) + if err := newSubGraphs[flexAlgo].AddEdge(edge); err != nil { + graph.log.Errorf("Error adding edge to subgraph: %s", err) + } } } } diff --git a/pkg/graph/network_graph_test.go b/pkg/graph/network_graph_test.go index 13a4d54..27845f6 100644 --- a/pkg/graph/network_graph_test.go +++ b/pkg/graph/network_graph_test.go @@ -760,6 +760,7 @@ func TestNetworkGraph_addEdgesToSubgraph(t *testing.T) { }, ), }, + wantErr: true, }, } @@ -773,6 +774,9 @@ func TestNetworkGraph_addEdgesToSubgraph(t *testing.T) { graph.edges = tt.edges newSubGraphs := make(map[uint32]*NetworkGraph) graph.addNodesToSubgraph(newSubGraphs) + if tt.wantErr { + graph.nodes = nil + } graph.addEdgesToSubgraph(newSubGraphs) assert.Equal(t, 2, len(newSubGraphs[128].edges)) assert.Equal(t, 0, len(newSubGraphs[129].edges))