summaryrefslogtreecommitdiff
path: root/ccan/grab_file/test/run-grab.c
blob: c9f242609697423e38483af5b0181a5cd9b3fbc9 (plain)
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
/* This is test for grab_file() function
 */
#include <ccan/grab_file/grab_file.h>
#include <stdlib.h>
#include <stdio.h>
#include <err.h>
#include <sys/stat.h>
#include <ccan/grab_file/grab_file.c>
#include <ccan/tap/tap.h>
#include <string.h>

static char **strsplit(const void *ctx, const char *string, const char *delims)
{
	char **lines = NULL;
	unsigned int max = 64, num = 0;

	lines = talloc_array(ctx, char *, max+1);

	while (*string != '\0') {
		unsigned int len = strcspn(string, delims);
		lines[num] = talloc_array(lines, char, len + 1);
		memcpy(lines[num], string, len);
		lines[num][len] = '\0';
		string += len;
		string += strspn(string, delims) ? 1 : 0;
		if (++num == max)
			lines = talloc_realloc(ctx, lines, char *, max*=2 + 1);
	}
	lines[num] = NULL;

	/* Shrink, so talloc_get_size works */
	return talloc_realloc(ctx, lines, char *, num+1);
}

int 
main(int argc, char *argv[])
{
	unsigned int	i;
	char 		**split, *str;
	int 		length;
	struct 		stat st;

	str = grab_file(NULL, "test/run-grab.c", NULL);
	split = strsplit(str, str, "\n");
	length = strlen(split[0]);
	ok1(!strcmp(split[0], "/* This is test for grab_file() function"));
	for (i = 1; split[i]; i++)	
		length += strlen(split[i]);
	ok1(!strcmp(split[i-1], "/* End of grab_file() test */"));
	if (stat("test/run-grab.c", &st) != 0) 
		/* FIXME: ditto */
		if (stat("ccan/grab_file/test/run-grab.c", &st) != 0) 
			err(1, "Could not stat self");
	ok1(st.st_size == length + i);
	talloc_free(str);
	
	return 0;
}

/* End of grab_file() test */