aboutsummaryrefslogtreecommitdiffstats
path: root/host/tests/graph.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'host/tests/graph.hpp')
-rw-r--r--host/tests/graph.hpp6
1 files changed, 4 insertions, 2 deletions
diff --git a/host/tests/graph.hpp b/host/tests/graph.hpp
index 0c2be0347..894c436b3 100644
--- a/host/tests/graph.hpp
+++ b/host/tests/graph.hpp
@@ -43,8 +43,10 @@ private:
void connect_nodes(uhd::rfnoc::source_node_ctrl::sptr A, uhd::rfnoc::sink_node_ctrl::sptr B)
{
- A->connect_downstream(B);
- B->connect_upstream(A);
+ const size_t actual_src_port = A->connect_downstream(B);
+ const size_t actual_dst_port = B->connect_upstream(A);
+ A->set_downstream_port(actual_src_port, actual_dst_port);
+ B->set_upstream_port(actual_dst_port, actual_src_port);
}
#endif /* INCLUDED_TEST_GRAPH_HPP */