lib: test-ostream-* - Call o_stream_nfinish() after uncorking

diff --git a/src/lib/test-ostream-buffer.c b/src/lib/test-ostream-buffer.c
index 40d5aeb28bea023b3a3cae5b19413e9b7852b966..af5253cdd252e661fbff9a6c1908d48471ffd7bb 100644 (file)
--- a/src/lib/test-ostream-buffer.c
+++ b/src/lib/test-ostream-buffer.c
@@ -38,8 +38,8 @@ static void test_ostream_buffer_random_once(void)
                        test_assert(o_stream_flush(output) > 0);
        }
 
-       test_assert(o_stream_flush(output) > 0);
        o_stream_uncork(output);
+       test_assert(o_stream_nfinish(output) == 0);
 
        i_assert(buffer->used <= MAX_BUFSIZE*4);
        test_assert(memcmp(buf, buffer->data, buffer->used) == 0);

diff --git a/src/lib/test-ostream-file.c b/src/lib/test-ostream-file.c
index ef492483502c89c76cdde59ad11c2534361d433b..4383ba4f2a8d11670fc4dd5bd3d7df37dfe606fe 100644 (file)
--- a/src/lib/test-ostream-file.c
+++ b/src/lib/test-ostream-file.c
@@ -45,8 +45,8 @@ static void test_ostream_file_random_once(void)
                        test_assert(o_stream_flush(output) > 0);
        }
 
-       test_assert(o_stream_flush(output) > 0);
        o_stream_uncork(output);
+       test_assert(o_stream_nfinish(output) == 0);
        ret = pread(fd, buf2, sizeof(buf2), 0);
        if (ret < 0)
                i_fatal("pread() failed: %m");