Open Menu / tests / envbuf.c
Loading...
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>

#include <darwintest.h>
#include "darwintest_utils.h"

#define IS_FLAG_SET(fp, flag) ((fp)->_flags & (flag))
#define BUFFER_SIZE(fp) ((fp)->_bf._size)

static FILE* test_setup(void)
{
	char* temp_dir;
	temp_dir = dt_tmpdir();
	T_LOG("Temp Dir: %s\n", temp_dir);

	char temp_path[128];
	snprintf(temp_path, sizeof(temp_path), "%s/%s", temp_dir, "test_file");
	T_LOG("Using Temporary File: %s\n", temp_path);
	FILE* fp = fopen(temp_path, "w");
	T_LOG("File opened: %d\n", fileno(fp));
	T_ASSERT_EQ(fileno(fp), 3, "file descriptor matches");
	// Need to prime the buffer
	fprintf(fp, "Test\n");

	return fp;
}

T_DECL(envbuf_PR_38637477_unbuf_all, "Forcing unbuffered through environment variable", T_META_ENVVAR("STDBUF=U"))
{
	FILE* fp = test_setup();

	T_EXPECT_TRUE(IS_FLAG_SET(fp, __SNBF), "unbuffered");
	T_EXPECT_FALSE(IS_FLAG_SET(fp, __SLBF), "not line buffered");
	T_EXPECT_EQ(BUFFER_SIZE(fp), 1, "buffer size 1");
}


T_DECL(envbuf_PR_38637477_line_all, "Forcing line buffering through environment variable", T_META_ENVVAR("STDBUF=L16"))
{
	FILE* fp = test_setup();

	T_EXPECT_FALSE(IS_FLAG_SET(fp, __SNBF), "not unbuffered");
	T_EXPECT_TRUE(IS_FLAG_SET(fp, __SLBF), "line buffered");
	T_EXPECT_EQ(BUFFER_SIZE(fp), 16, "buffer size 16");
}

T_DECL(envbuf_PR_38637477_full_all, "Forcing full buffering through environment variable", T_META_ENVVAR("STDBUF=F16"))
{
	FILE* fp = test_setup();

	T_EXPECT_FALSE(IS_FLAG_SET(fp, __SNBF), "not unbuffered");
	T_EXPECT_FALSE(IS_FLAG_SET(fp, __SLBF), "not line buffered");
	T_EXPECT_EQ(BUFFER_SIZE(fp), 16, "buffer size 16");
}

T_DECL(envbuf_PR_38637477_unbuf_stdout, "Forcing unbuffered through environment variable for stdout", T_META_ENVVAR("STDBUF1=U"))
{

	fprintf(stdout, "Test\n");

	T_EXPECT_TRUE(IS_FLAG_SET(stdout, __SNBF), "unbuffered");
	T_EXPECT_FALSE(IS_FLAG_SET(stdout, __SLBF), "not linebuffered");
	T_EXPECT_EQ(BUFFER_SIZE(stdout), 1, "buffer size 1");
}

T_DECL(envbuf_PR_38637477_line_stdout, "Forcing line buffering through environment variable for stdout", T_META_ENVVAR("STDBUF1=L32"))
{

	fprintf(stdout, "Test\n");

	T_EXPECT_FALSE(IS_FLAG_SET(stdout, __SNBF), "not unbuffered");
	T_EXPECT_TRUE(IS_FLAG_SET(stdout, __SLBF), "line buffered");
	T_EXPECT_EQ(BUFFER_SIZE(stdout), 32, "buffer size 32");
}

T_DECL(envbuf_PR_38637477_full_stdout, "Forcing full buffering through environment variable for stdout", T_META_ENVVAR("STDBUF1=F16"))
{

	fprintf(stdout, "Test\n");

	T_EXPECT_FALSE(IS_FLAG_SET(stdout, __SNBF), "not unbuffered");
	T_EXPECT_FALSE(IS_FLAG_SET(stdout, __SLBF), "not line buffered");
	T_EXPECT_EQ(BUFFER_SIZE(stdout), 16, "buffer size 16");
}