tests: Bluetooth: BAP: Test bt_bap_ep_get_conn in BSIM

Add steps in the BAP tests to verify the correctness of
bt_bap_ep_get_conn.

Signed-off-by: Emil Gydesen <[email protected]>

Author: Thalley

tests/bsim/bluetooth/audio/src/bap_broadcast_sink_test.c

@@ -565,6 +565,7 @@ static void stream_started_cb(struct bt_bap_stream *stream)
 {
 	struct audio_test_stream *test_stream = audio_test_stream_from_bap_stream(stream);
 	struct bt_bap_ep_info info;
+	struct bt_conn *ep_conn;
 	int err;
 
 	memset(&test_stream->last_info, 0, sizeof(test_stream->last_info));
@@ -602,6 +603,12 @@ static void stream_started_cb(struct bt_bap_stream *stream)
 		return;
 	}
 
+	ep_conn = bt_bap_ep_get_conn(stream->ep);
+	if (ep_conn != NULL) {
+		FAIL("Invalid conn from endpoint: %p", ep_conn);
+		return;
+	}
+
 	printk("Stream %p started\n", stream);
 	k_sem_give(&sem_stream_started);
 

tests/bsim/bluetooth/audio/src/bap_broadcast_source_test.c

@@ -178,6 +178,7 @@ static void stream_started_cb(struct bt_bap_stream *stream)
 {
 	struct audio_test_stream *test_stream = audio_test_stream_from_bap_stream(stream);
 	struct bt_bap_ep_info info;
+	struct bt_conn *ep_conn;
 	int err;
 
 	test_stream->seq_num = 0U;
@@ -214,6 +215,12 @@ static void stream_started_cb(struct bt_bap_stream *stream)
 		return;
 	}
 
+	ep_conn = bt_bap_ep_get_conn(stream->ep);
+	if (ep_conn != NULL) {
+		FAIL("Invalid conn from endpoint: %p", ep_conn);
+		return;
+	}
+
 	err = bap_stream_tx_register(stream);
 	if (err != 0) {
 		FAIL("Failed to register stream %p for TX: %d\n", stream, err);

tests/bsim/bluetooth/audio/src/bap_unicast_client_test.c

@@ -69,12 +69,21 @@ CREATE_FLAG(flag_operation_success);
 
 static void stream_configured(struct bt_bap_stream *stream, const struct bt_bap_qos_cfg_pref *pref)
 {
+	struct bt_conn *ep_conn;
+
 	printk("Configured stream %p\n", stream);
 
 	/* TODO: The preference should be used/taken into account when
 	 * setting the QoS
 	 */
 
+	ep_conn = bt_bap_ep_get_conn(stream->ep);
+	if (ep_conn == NULL || stream->conn != ep_conn) {
+		FAIL("Invalid conn from endpoint: %p", ep_conn);
+		return;
+	}
+	bt_conn_unref(ep_conn);
+
 	SET_FLAG(flag_stream_codec_configured);
 }
 

tests/bsim/bluetooth/audio/src/bap_unicast_server_test.c

@@ -114,8 +114,6 @@ static int lc3_config(struct bt_conn *conn, const struct bt_bap_ep *ep, enum bt_
 
 	bt_bap_unicast_server_foreach_ep(conn, print_ase_info, NULL);
 
-	SET_FLAG(flag_stream_configured);
-
 	*pref = qos_pref;
 
 	return 0;
@@ -222,6 +220,23 @@ static const struct bt_bap_unicast_server_cb unicast_server_cb = {
 	.release = lc3_release,
 };
 
+static void stream_configured_cb(struct bt_bap_stream *stream,
+				 const struct bt_bap_qos_cfg_pref *pref)
+{
+	struct bt_conn *ep_conn;
+
+	printk("Configured stream %p\n", stream);
+
+	ep_conn = bt_bap_ep_get_conn(stream->ep);
+	if (ep_conn == NULL || stream->conn != ep_conn) {
+		FAIL("Invalid conn from endpoint: %p", ep_conn);
+		return;
+	}
+	bt_conn_unref(ep_conn);
+
+	SET_FLAG(flag_stream_configured);
+}
+
 static void stream_enabled_cb(struct bt_bap_stream *stream)
 {
 	struct bt_bap_ep_info ep_info;
@@ -279,6 +294,7 @@ static void stream_stopped_cb(struct bt_bap_stream *stream, uint8_t reason)
 }
 
 static struct bt_bap_stream_ops stream_ops = {
+	.configured = stream_configured_cb,
 	.enabled = stream_enabled_cb,
 	.started = stream_started_cb,
 	.stopped = stream_stopped_cb,