X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=test%2Fchannels-no-partial.c;h=c765456abff08e3de1b9c8a5151182e59c6ddd4d;hb=HEAD;hp=d8a8bf1eb5262d1b3af29a63e6b3f677870b7294;hpb=9a2520c36431a8a5fd90451e97f488c22f4decc5;p=meshlink

diff --git a/test/channels-no-partial.c b/test/channels-no-partial.c
index d8a8bf1e..43792be3 100644
--- a/test/channels-no-partial.c
+++ b/test/channels-no-partial.c
@@ -1,3 +1,7 @@
+#ifdef NDEBUG
+#undef NDEBUG
+#endif
+
 #include <assert.h>
 #include <stdio.h>
 #include <unistd.h>
@@ -64,11 +68,24 @@ int main(int argc, char *argv[]) {
 
 	// Restart a to ensure it gets to flush the channel send queue.
 
-	meshlink_start(mesh_a);
+	assert(meshlink_start(mesh_a));
 
 	assert_after(!meshlink_channel_get_sendq(mesh_a, channel), 30);
 	assert(meshlink_channel_send(mesh_a, channel, buf, 512) == 512);
 
+	// Check that we can change the NO_PARTIAL flag
+
+	assert_after(!meshlink_channel_get_sendq(mesh_a, channel), 30);
+	meshlink_set_channel_sndbuf(mesh_a, channel, 256);
+	assert(meshlink_channel_send(mesh_a, channel, buf, 257) == -1);
+	meshlink_set_channel_flags(mesh_a, channel, 0);
+	assert(meshlink_channel_send(mesh_a, channel, buf, 257) == 256);
+
+	assert_after(!meshlink_channel_get_sendq(mesh_a, channel), 30);
+	meshlink_set_channel_flags(mesh_a, channel, MESHLINK_CHANNEL_NO_PARTIAL);
+	assert(meshlink_channel_send(mesh_a, channel, buf, 257) == -1);
+	assert(meshlink_channel_send(mesh_a, channel, buf, 256) == 256);
+
 	// Clean up.
 
 	close_meshlink_pair(mesh_a, mesh_b);