/* SPDX-License-Identifier: LGPL-2.1-or-later */ #include #include "alloc-util.h" #include "format-table.h" #include "string-util.h" #include "strv.h" #include "time-util.h" static void test_issue_9549(void) { _cleanup_(table_unrefp) Table *table = NULL; _cleanup_free_ char *formatted = NULL; log_info("/* %s */", __func__); assert_se(table = table_new("name", "type", "ro", "usage", "created", "modified")); assert_se(table_set_align_percent(table, TABLE_HEADER_CELL(3), 100) >= 0); assert_se(table_add_many(table, TABLE_STRING, "foooo", TABLE_STRING, "raw", TABLE_BOOLEAN, false, TABLE_SIZE, (uint64_t) (673.7*1024*1024), TABLE_STRING, "Wed 2018-07-11 00:10:33 JST", TABLE_STRING, "Wed 2018-07-11 00:16:00 JST") >= 0); table_set_width(table, 75); assert_se(table_format(table, &formatted) >= 0); printf("%s\n", formatted); assert_se(streq(formatted, "NAME TYPE RO USAGE CREATED MODIFIED\n" "foooo raw no 673.6M Wed 2018-07-11 00:10:33 J… Wed 2018-07-11 00:16:00 JST\n" )); } static void test_multiline(void) { _cleanup_(table_unrefp) Table *table = NULL; _cleanup_free_ char *formatted = NULL; log_info("/* %s */", __func__); assert_se(table = table_new("foo", "bar")); assert_se(table_set_align_percent(table, TABLE_HEADER_CELL(1), 100) >= 0); assert_se(table_add_many(table, TABLE_STRING, "three\ndifferent\nlines", TABLE_STRING, "two\nlines\n") >= 0); table_set_cell_height_max(table, 1); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three… two…\n")); formatted = mfree(formatted); table_set_cell_height_max(table, 2); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three two\n" "different… lines\n")); formatted = mfree(formatted); table_set_cell_height_max(table, 3); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three two\n" "different lines\n" "lines \n")); formatted = mfree(formatted); table_set_cell_height_max(table, (size_t) -1); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three two\n" "different lines\n" "lines \n")); formatted = mfree(formatted); assert_se(table_add_many(table, TABLE_STRING, "short", TABLE_STRING, "a\npair") >= 0); assert_se(table_add_many(table, TABLE_STRING, "short2\n", TABLE_STRING, "a\nfour\nline\ncell") >= 0); table_set_cell_height_max(table, 1); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three… two…\n" "short a…\n" "short2 a…\n")); formatted = mfree(formatted); table_set_cell_height_max(table, 2); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three two\n" "different… lines\n" "short a\n" " pair\n" "short2 a\n" " four…\n")); formatted = mfree(formatted); table_set_cell_height_max(table, 3); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three two\n" "different lines\n" "lines \n" "short a\n" " pair\n" "short2 a\n" " four\n" " line…\n")); formatted = mfree(formatted); table_set_cell_height_max(table, (size_t) -1); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three two\n" "different lines\n" "lines \n" "short a\n" " pair\n" "short2 a\n" " four\n" " line\n" " cell\n")); formatted = mfree(formatted); } static void test_strv(void) { _cleanup_(table_unrefp) Table *table = NULL; _cleanup_free_ char *formatted = NULL; log_info("/* %s */", __func__); assert_se(table = table_new("foo", "bar")); assert_se(table_set_align_percent(table, TABLE_HEADER_CELL(1), 100) >= 0); assert_se(table_add_many(table, TABLE_STRV, STRV_MAKE("three", "different", "lines"), TABLE_STRV, STRV_MAKE("two", "lines")) >= 0); table_set_cell_height_max(table, 1); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three… two…\n")); formatted = mfree(formatted); table_set_cell_height_max(table, 2); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three two\n" "different… lines\n")); formatted = mfree(formatted); table_set_cell_height_max(table, 3); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three two\n" "different lines\n" "lines \n")); formatted = mfree(formatted); table_set_cell_height_max(table, (size_t) -1); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three two\n" "different lines\n" "lines \n")); formatted = mfree(formatted); assert_se(table_add_many(table, TABLE_STRING, "short", TABLE_STRV, STRV_MAKE("a", "pair")) >= 0); assert_se(table_add_many(table, TABLE_STRV, STRV_MAKE("short2"), TABLE_STRV, STRV_MAKE("a", "four", "line", "cell")) >= 0); table_set_cell_height_max(table, 1); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three… two…\n" "short a…\n" "short2 a…\n")); formatted = mfree(formatted); table_set_cell_height_max(table, 2); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three two\n" "different… lines\n" "short a\n" " pair\n" "short2 a\n" " four…\n")); formatted = mfree(formatted); table_set_cell_height_max(table, 3); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three two\n" "different lines\n" "lines \n" "short a\n" " pair\n" "short2 a\n" " four\n" " line…\n")); formatted = mfree(formatted); table_set_cell_height_max(table, (size_t) -1); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three two\n" "different lines\n" "lines \n" "short a\n" " pair\n" "short2 a\n" " four\n" " line\n" " cell\n")); formatted = mfree(formatted); } static void test_strv_wrapped(void) { _cleanup_(table_unrefp) Table *table = NULL; _cleanup_free_ char *formatted = NULL; log_info("/* %s */", __func__); assert_se(table = table_new("foo", "bar")); assert_se(table_set_align_percent(table, TABLE_HEADER_CELL(1), 100) >= 0); assert_se(table_add_many(table, TABLE_STRV_WRAPPED, STRV_MAKE("three", "different", "lines"), TABLE_STRV_WRAPPED, STRV_MAKE("two", "lines")) >= 0); table_set_cell_height_max(table, 1); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three different lines two lines\n")); formatted = mfree(formatted); table_set_cell_height_max(table, 2); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three different lines two lines\n")); formatted = mfree(formatted); table_set_cell_height_max(table, 3); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three different lines two lines\n")); formatted = mfree(formatted); table_set_cell_height_max(table, (size_t) -1); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three different lines two lines\n")); formatted = mfree(formatted); assert_se(table_add_many(table, TABLE_STRING, "short", TABLE_STRV_WRAPPED, STRV_MAKE("a", "pair")) >= 0); assert_se(table_add_many(table, TABLE_STRV_WRAPPED, STRV_MAKE("short2"), TABLE_STRV_WRAPPED, STRV_MAKE("a", "eight", "line", "ćęłł", "___5___", "___6___", "___7___", "___8___")) >= 0); table_set_cell_height_max(table, 1); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three different… two lines\n" "short a pair\n" "short2 a eight line ćęłł…\n")); formatted = mfree(formatted); table_set_cell_height_max(table, 2); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three different two lines\n" "lines \n" "short a pair\n" "short2 a eight line ćęłł\n" " ___5___ ___6___…\n")); formatted = mfree(formatted); table_set_cell_height_max(table, 3); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three different two lines\n" "lines \n" "short a pair\n" "short2 a eight line ćęłł\n" " ___5___ ___6___\n" " ___7___ ___8___\n")); formatted = mfree(formatted); table_set_cell_height_max(table, (size_t) -1); assert_se(table_format(table, &formatted) >= 0); fputs(formatted, stdout); assert_se(streq(formatted, "FOO BAR\n" "three different two lines\n" "lines \n" "short a pair\n" "short2 a eight line ćęłł\n" " ___5___ ___6___\n" " ___7___ ___8___\n")); formatted = mfree(formatted); } int main(int argc, char *argv[]) { _cleanup_(table_unrefp) Table *t = NULL; _cleanup_free_ char *formatted = NULL; assert_se(setenv("SYSTEMD_COLORS", "0", 1) >= 0); assert_se(setenv("COLUMNS", "40", 1) >= 0); assert_se(t = table_new("one", "two", "three")); assert_se(table_set_align_percent(t, TABLE_HEADER_CELL(2), 100) >= 0); assert_se(table_add_many(t, TABLE_STRING, "xxx", TABLE_STRING, "yyy", TABLE_BOOLEAN, true) >= 0); assert_se(table_add_many(t, TABLE_STRING, "a long field", TABLE_STRING, "yyy", TABLE_SET_UPPERCASE, 1, TABLE_BOOLEAN, false) >= 0); assert_se(table_format(t, &formatted) >= 0); printf("%s\n", formatted); assert_se(streq(formatted, "ONE TWO THREE\n" "xxx yyy yes\n" "a long field YYY no\n")); formatted = mfree(formatted); table_set_width(t, 40); assert_se(table_format(t, &formatted) >= 0); printf("%s\n", formatted); assert_se(streq(formatted, "ONE TWO THREE\n" "xxx yyy yes\n" "a long field YYY no\n")); formatted = mfree(formatted); table_set_width(t, 12); assert_se(table_format(t, &formatted) >= 0); printf("%s\n", formatted); assert_se(streq(formatted, "ONE TWO THR…\n" "xxx yyy yes\n" "a … YYY no\n")); formatted = mfree(formatted); table_set_width(t, 5); assert_se(table_format(t, &formatted) >= 0); printf("%s\n", formatted); assert_se(streq(formatted, "… … …\n" "… … …\n" "… … …\n")); formatted = mfree(formatted); table_set_width(t, 3); assert_se(table_format(t, &formatted) >= 0); printf("%s\n", formatted); assert_se(streq(formatted, "… … …\n" "… … …\n" "… … …\n")); formatted = mfree(formatted); table_set_width(t, (size_t) -1); assert_se(table_set_sort(t, (size_t) 0, (size_t) 2, (size_t) -1) >= 0); assert_se(table_format(t, &formatted) >= 0); printf("%s\n", formatted); assert_se(streq(formatted, "ONE TWO THREE\n" "a long field YYY no\n" "xxx yyy yes\n")); formatted = mfree(formatted); table_set_header(t, false); assert_se(table_add_many(t, TABLE_STRING, "fäää", TABLE_STRING, "uuu", TABLE_BOOLEAN, true) >= 0); assert_se(table_add_many(t, TABLE_STRING, "fäää", TABLE_STRING, "zzz", TABLE_BOOLEAN, false) >= 0); assert_se(table_add_many(t, TABLE_EMPTY, TABLE_SIZE, (uint64_t) 4711, TABLE_TIMESPAN, (usec_t) 5*USEC_PER_MINUTE) >= 0); assert_se(table_format(t, &formatted) >= 0); printf("%s\n", formatted); assert_se(streq(formatted, "a long field YYY no\n" "fäää zzz no\n" "fäää uuu yes\n" "xxx yyy yes\n" " 4.6K 5min\n")); formatted = mfree(formatted); assert_se(table_set_display(t, (size_t) 2, (size_t) 0, (size_t) 2, (size_t) 0, (size_t) 0, (size_t) -1) >= 0); assert_se(table_format(t, &formatted) >= 0); printf("%s\n", formatted); if (isatty(STDOUT_FILENO)) assert_se(streq(formatted, " no a long f… no a long f… a long fi…\n" " no fäää no fäää fäää\n" " yes fäää yes fäää fäää\n" " yes xxx yes xxx xxx\n" "5min 5min \n")); else assert_se(streq(formatted, " no a long field no a long field a long field\n" " no fäää no fäää fäää\n" " yes fäää yes fäää fäää\n" " yes xxx yes xxx xxx\n" "5min 5min \n")); test_issue_9549(); test_multiline(); test_strv(); test_strv_wrapped(); return 0; }