X-Git-Url: http://git.meshlink.io/?p=meshlink;a=blobdiff_plain;f=test%2Fchannels-no-partial.c;h=8216a5568bb9260f4b9e862fba3478732f82983b;hp=caa308fbb69cba4a2b9c5c3a1d5c67d4a28a4e92;hb=83c20878a6ef6e1e3a1abad17961e02137f16ab5;hpb=6eb1cfb2b112baf54d7c91e6d303f823ddb8acdd

diff --git a/test/channels-no-partial.c b/test/channels-no-partial.c
index caa308fb..8216a556 100644
--- a/test/channels-no-partial.c
+++ b/test/channels-no-partial.c
@@ -36,13 +36,17 @@ int main(int argc, char *argv[]) {
 	meshlink_node_t *b = meshlink_get_node(mesh_a, "b");
 	assert(b);
 
-	meshlink_channel_t *channel = meshlink_channel_open_ex(mesh_a, b, 1, NULL, NULL, 0, MESHLINK_CHANNEL_NO_PARTIAL);
+	meshlink_channel_t *channel = meshlink_channel_open_ex(mesh_a, b, 1, NULL, NULL, 0, MESHLINK_CHANNEL_TCP | MESHLINK_CHANNEL_NO_PARTIAL);
 	assert(channel);
 
+	// Stop a to ensure we get deterministic behaviour for the channel send queue.
+
+	meshlink_stop(mesh_a);
+
 	// Verify that no partial sends succeed.
 	// If rejected sends would fit an empty send buffer, 0 should be returned, otherwise -1.
 
-	char buf[256] = "";
+	char buf[512] = "data";
 
 	meshlink_set_channel_sndbuf(mesh_a, channel, 256);
 	assert(meshlink_channel_send(mesh_a, channel, buf, 257) == -1);
@@ -56,6 +60,10 @@ int main(int argc, char *argv[]) {
 	assert(meshlink_channel_send(mesh_a, channel, buf, 29) == 0);
 	assert(meshlink_channel_send(mesh_a, channel, buf, 513) == -1);
 
+	// Restart a to ensure it gets to flush the channel send queue.
+
+	meshlink_start(mesh_a);
+
 	assert_after(!meshlink_channel_get_sendq(mesh_a, channel), 30);
 	assert(meshlink_channel_send(mesh_a, channel, buf, 512) == 512);