X-Git-Url: https://git.draconx.ca/gitweb/dxcommon.git/blobdiff_plain/9efff8c7a48509fabab57631f5c64c669183dee0..27f6a9c3d4663286f1bec753fbd035ab1a6eb5de:/t/getline.c

diff --git a/t/getline.c b/t/getline.c
new file mode 100644
index 0000000..912e571
--- /dev/null
+++ b/t/getline.c
@@ -0,0 +1,123 @@
+/*
+ * Copyright © 2024 Nick Bowler
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+void tap_diag(const char *fmt, ...);
+
+#undef HAVE_GETLINE
+#define DX_GETLINE_INITIAL_ALLOC 2
+#include "getline.h"
+#include "tap.h"
+
+#include <errno.h>
+
+#define LONGLINE "this line is pretty long and may require scaling the buffer multiple times"
+
+static const char testdata[] =
+	"\n"
+	LONGLINE "\n"
+	LONGLINE "\n"
+	LONGLINE "\n"
+	"hellorld\n";
+
+int main(void)
+{
+	char *line = NULL;
+	size_t long_size;
+	size_t n = 0;
+	FILE *f;
+
+	if (!(f = tmpfile()))
+		tap_bail_out("tmpfile failed: %s\n", strerror(errno));
+
+	if (fwrite(testdata, sizeof testdata-1, 1, f) != 1)
+		tap_bail_out("fwrite failed: %s\n", strerror(errno));
+
+	if (fflush(f) != 0)
+		tap_bail_out("fflush failed: %s\n", strerror(errno));
+
+	if (fseek(f, 0, SEEK_SET) != 0)
+		tap_bail_out("fseek failed: %s\n", strerror(errno));
+
+	/* First line: empty */
+	tap_result(dx_getline(&line, &n, f) != 0, "dx_getline");
+	if (!tap_result(n == 2, "alloc size")) {
+		tap_diag("Failed, unexpected result");
+		tap_diag("   Received: %lu", (unsigned long)n);
+		tap_diag("   Expected: 2");
+	}
+	if (!tap_result(line[0] == 0, "returned string")) {
+		tap_diag("Failed, unexpected result");
+		tap_diag("   Received: \"%.*s\"", (int)n, line);
+		tap_diag("   Expected: \"\"");
+	}
+
+	/* Next line: long line (buffer needs expanding) */
+	tap_result(dx_getline(&line, &n, f) != 0, "dx_getline");
+	if (!tap_result(n > sizeof LONGLINE, "alloc size")) {
+		tap_diag("Failed, unexpected result");
+		tap_diag("   Received: %lu", (unsigned long)n);
+		tap_diag("   Expected: >%u", (unsigned)(sizeof LONGLINE));
+	}
+	if (!tap_result(!strcmp(line, LONGLINE), "returned string")) {
+		tap_diag("Failed, unexpected result");
+		tap_diag("   Received: \"%.*s\"", (int)n, line);
+		tap_diag("   Expected: \"" LONGLINE "\"");
+	}
+
+	/* Next line: long line (buffer does not need expanding) */
+	long_size = n;
+	tap_result(dx_getline(&line, &n, f) != 0, "dx_getline");
+	if (!tap_result(n == long_size, "alloc size")) {
+		tap_diag("Failed, unexpected result");
+		tap_diag("   Received: %lu", (unsigned long)n);
+		tap_diag("   Expected: %lu", (unsigned long)long_size);
+	}
+	if (!tap_result(!strcmp(line, LONGLINE), "returned string")) {
+		tap_diag("Failed, unexpected result");
+		tap_diag("   Received: \"%.*s\"", (int)n, line);
+		tap_diag("   Expected: \"" LONGLINE "\"");
+	}
+
+	/* Next line: long line (new buffer allocation) */
+	free(line); line = NULL; n = 0;
+	tap_result(dx_getline(&line, &n, f) != 0, "dx_getline");
+	if (!tap_result(n == long_size, "alloc size")) {
+		tap_diag("Failed, unexpected result");
+		tap_diag("   Received: %lu", (unsigned long)n);
+		tap_diag("   Expected: %lu", (unsigned long)long_size);
+	}
+	if (!tap_result(!strcmp(line, LONGLINE), "returned string")) {
+		tap_diag("Failed, unexpected result");
+		tap_diag("   Received: \"%.*s\"", (int)n, line);
+		tap_diag("   Expected: \"" LONGLINE "\"");
+	}
+
+	/* Next line: short line */
+	tap_result(dx_getline(&line, &n, f) != 0, "dx_getline");
+	if (!tap_result(n == long_size, "alloc size")) {
+		tap_diag("Failed, unexpected result");
+		tap_diag("   Received: %lu", (unsigned long)n);
+		tap_diag("   Expected: %lu", (unsigned long)long_size);
+	}
+	if (!tap_result(!strcmp(line, "hellorld"), "returned string")) {
+		tap_diag("Failed, unexpected result");
+		tap_diag("   Received: \"%.*s\"", (int)n, line);
+		tap_diag("   Expected: \"hellorld\"");
+	}
+
+	tap_done();
+}