/*
* (c) 2007, Quest Software, Inc. All rights reserved.
*
* This file is part of XScreenSaver,
* Copyright (c) 1993-2004 Jamie Zawinski <jwz@jwz.org>
*
* Permission to use, copy, modify, distribute, and sell this software and its
* documentation for any purpose is hereby granted without fee, provided that
* the above copyright notice appear in all copies and that both that
* copyright notice and this permission notice appear in supporting
* documentation. No representations are made about the suitability of this
* software for any purpose. It is provided "as is" without express or
* implied warranty.
*/
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include "mlstring.c" /* hokey, but whatever */
#define WRAP_WIDTH_PX 100
#undef Bool
#undef True
#undef False
typedef int Bool;
#define True 1
#define False 0
#define SKIPPED -1
#define SUCCESS 0
#define FAILURE 1
#define FAIL(msg, ...) \
do { \
++failcount; \
fprintf(stderr, "[FAIL] "); \
fprintf(stderr, msg, __VA_ARGS__); \
putc('\n', stderr); \
return FAILURE; \
} while (0)
#define SUCCEED(testname) \
do { \
fprintf(stderr, "[SUCCESS] %s\n", (testname)); \
} while (0)
#define SKIP(testname) \
do { \
fprintf(stderr, "[SKIPPED] %s\n", (testname)); \
} while (0)
extern mlstring* mlstring_allocate(const char *msg);
extern void mlstring_wrap(mlstring *mstr, XFontStruct *font, Dimension width);
static int failcount = 0;
static char *mlstring_to_cstr(const mlstring *mlstr) {
char *cstr;
size_t cstrlen = 0, alloclen = 1024;
const struct mlstr_line *line;
cstr = malloc(alloclen);
if (!cstr)
return NULL;
cstr[0] = '\0';
for (line = mlstr->lines; line; line = line->next_line) {
/* Extend the buffer if necessary. */
if (cstrlen + strlen(line->line) + 1 > alloclen) {
cstr = realloc(cstr, alloclen *= 2);
if (!cstr)
return NULL;
}
/* If this is not the first line */
if (line != mlstr->lines) {
/* Append a newline character */
cstr[cstrlen] = '\n';
++cstrlen;
cstr[cstrlen] = '\0';
}
strcat(cstr, line->line);
cstrlen += strlen(line->line);
}
return cstr;
}
/* Pass -1 for expect_min or expect_exact to not check that value.
* expect_empty_p means an empty line is expected at some point in the string.
* Also ensures that the string was not too wide after wrapping. */
static int mlstring_expect_lines(const mlstring *mlstr, int expect_min, int expect_exact, Bool expect_empty_p)
{
int count;
Bool got_empty_line = False;
const struct mlstr_line *line = mlstr->lines;
for (count = 0; line; line = line->next_line) {
if (line->line[0] == '\0') {
if (!expect_empty_p)
FAIL("Not expecting empty lines, but got one on line %d of [%s]", count + 1, mlstring_to_cstr(mlstr));
got_empty_line = True;
}
++count;
}
if (expect_empty_p && !got_empty_line)
FAIL("Expecting an empty line, but none found in [%s]", mlstring_to_cstr(mlstr));
if (expect_exact != -1 && expect_exact != count)
FAIL("Expected %d lines, got %d", expect_exact, count);
if (expect_min != -1 && count < expect_min)
FAIL("Expected at least %d lines, got %d", expect_min, count);
return SUCCESS;
}
static int mlstring_expect(const char *msg, int expect_lines, const mlstring *mlstr, Bool expect_empty_p)
{
char *str, *str_top;
const struct mlstr_line *cur;
int linecount = 0;
/* Duplicate msg so we can chop it up */
str_top = strdup(msg);
if (!str_top)
return SKIPPED;
/* Replace all newlines with NUL */
str = str_top;
while ((str = strchr(str, '\n')))
*str++ = '\0';
/* str is now used to point to the expected string */
str = str_top;
for (cur = mlstr->lines; cur; cur = cur->next_line)
{
++linecount;
if (strcmp(cur->line, str))
FAIL("lines didn't match; expected [%s], got [%s]", str, cur->line);
str += strlen(str) + 1; /* Point to the next expected string */
}
free(str_top);
return mlstring_expect_lines(mlstr, -1, expect_lines, expect_empty_p);
}
/* Ensures that the width has been set properly after wrapping */
static int check_width(const char *msg, const mlstring *mlstr) {
if (mlstr->overall_width == 0)
FAIL("Overall width was zero for string [%s]", msg);
if (mlstr->overall_width > WRAP_WIDTH_PX)
FAIL("Overall width was %hu but the maximum wrap width was %d", mlstr->overall_width, WRAP_WIDTH_PX);
return SUCCESS;
}
/* FAIL() actually returns the wrong return codes in main, but it
* prints a message which is what we want. */
#define TRY_NEW(str, numl, expect_empty) \
do { \
mlstr = mlstring_allocate((str)); \
if (!mlstr) \
FAIL("%s", #str); \
if (SUCCESS == mlstring_expect((str), (numl), mlstr, (expect_empty))) \
SUCCEED(#str); \
free(mlstr); \
} while (0)
/* Expects an XFontStruct* font, and tries to wrap to 100px */
#define TRY_WRAP(str, minl, expect_empty) \
do { \
mltest = mlstring_allocate((str)); \
if (!mltest) \
SKIP(#str); \
else { \
mlstring_wrap(mltest, font, WRAP_WIDTH_PX); \
check_width((str), mltest); \
if (SUCCESS == mlstring_expect_lines(mltest, (minl), -1, (expect_empty))) \
SUCCEED(#str); \
free(mltest); \
mltest = NULL; \
} \
} while (0)
/* Ideally this function would use stub functions rather than real Xlib.
* Then it would be possible to test for exact line counts, which would be
* more reliable.
* It also doesn't handle Xlib errors.
*
* Don't print anything based on the return value of this function, it only
* returns a value so that I can use the FAIL() macro without warning.
*
* Anyone who understands this function wins a cookie ;)
*/
static int test_wrapping(void)
{
Display *dpy = NULL;
XFontStruct *font = NULL;
mlstring *mltest = NULL;
int ok = 0;
int chars_per_line, chars_first_word, i;
const char *test_short = "a";
const char *test_hardwrap = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa";
const char *test_withnewlines = "a\nb";
char *test_softwrap = NULL;
dpy = XOpenDisplay(NULL);
if (!dpy)
goto end;
font = XLoadQueryFont(dpy, "fixed");
if (!font)
goto end;
TRY_WRAP(test_short, 1, False);
TRY_WRAP(test_hardwrap, 2, False);
TRY_WRAP(test_withnewlines, 2, False);
/* See if wrapping splits on word boundaries like it should */
chars_per_line = WRAP_WIDTH_PX / font->max_bounds.width;
if (chars_per_line < 3)
goto end;
/* Allocate for 2 lines + \0 */
test_softwrap = malloc(chars_per_line * 2 + 1);
if (!test_softwrap)
goto end;
/* 2 = strlen(' a'); that is, the minimum space required to start a new word
* on the same line. */
chars_first_word = chars_per_line - 2;
for (i = 0; i < chars_first_word; ++i) {
test_softwrap[i] = 'a'; /* first word */
test_softwrap[i + chars_per_line] = 'b'; /* second word */
}
/* space between first & second words */
test_softwrap[chars_first_word] = ' ';
/* first char of second word (last char of first line) */
test_softwrap[chars_first_word + 1] = 'b';
/* after second word */
test_softwrap[chars_per_line * 2] = '\0';
mltest = mlstring_allocate(test_softwrap);
mlstring_wrap(mltest, font, WRAP_WIDTH_PX);
/* reusing 'i' for a moment here to make freeing mltest easier */
i = strlen(mltest->lines->line);
free(mltest);
if (i != chars_first_word)
FAIL("Soft wrap failed, expected the first line to be %d chars, but it was %d.", chars_first_word, i);
SUCCEED("Soft wrap");
ok = 1;
end:
if (test_softwrap)
free(test_softwrap);
if (font)
XFreeFont(dpy, font);
if (dpy)
XCloseDisplay(dpy);
if (!ok)
SKIP("wrapping");
return ok ? SUCCESS : SKIPPED; /* Unused, actually */
}
int main(int argc, char *argv[])
{
const char *oneline = "1Foo";
const char *twolines = "2Foo\nBar";
const char *threelines = "3Foo\nBar\nWhippet";
const char *trailnewline = "4Foo\n";
const char *trailnewlines = "5Foo\n\n";
const char *embeddednewlines = "6Foo\n\nBar";
mlstring *mlstr;
TRY_NEW(oneline, 1, False);
TRY_NEW(twolines, 2, False);
TRY_NEW(threelines, 3, False);
TRY_NEW(trailnewline, 2, True);
TRY_NEW(trailnewlines, 3, True);
TRY_NEW(embeddednewlines, 3, True);
(void) test_wrapping();
fprintf(stdout, "%d test failures.\n", failcount);
return !!failcount;
}
/* vim:ts=8:sw=2:noet
*/