src/test/test-journal-importer.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2 /***
   3   Copyright © 2016 Zbigniew Jędrzejewski-Szmek
   4 ***/
   5
   6 #include <sys/types.h>
   7 #include <sys/stat.h>
   8 #include <fcntl.h>
   9
  10 #include "log.h"
  11 #include "journal-importer.h"
  12 #include "string-util.h"
  13 #include "tests.h"
  14
  15 static void assert_iovec_entry(const struct iovec *iovec, const char* content) {
  16         assert_se(strlen(content) == iovec->iov_len);
  17         assert_se(memcmp(content, iovec->iov_base, iovec->iov_len) == 0);
  18 }
  19
  20 #define COREDUMP_PROC_GROUP                                             \
  21         "COREDUMP_PROC_CGROUP=1:name=systemd:/\n"                       \
  22         "0::/user.slice/user-1002.slice/user@1002.service/gnome-terminal-server.service\n"
  23
  24 static void test_basic_parsing(void) {
  25         _cleanup_(journal_importer_cleanup) JournalImporter imp = {};
  26         int r;
  27
  28         imp.fd = open(get_testdata_dir("/journal-data/journal-1.txt"), O_RDONLY|O_CLOEXEC);
  29         assert_se(imp.fd >= 0);
  30
  31         do
  32                 r = journal_importer_process_data(&imp);
  33         while (r == 0 && !journal_importer_eof(&imp));
  34         assert_se(r == 1);
  35
  36         /* We read one entry, so we should get EOF on next read, but not yet */
  37         assert_se(!journal_importer_eof(&imp));
  38
  39         assert_se(imp.iovw.count == 6);
  40         assert_iovec_entry(&imp.iovw.iovec[0], "_BOOT_ID=1531fd22ec84429e85ae888b12fadb91");
  41         assert_iovec_entry(&imp.iovw.iovec[1], "_TRANSPORT=journal");
  42         assert_iovec_entry(&imp.iovw.iovec[2], COREDUMP_PROC_GROUP);
  43         assert_iovec_entry(&imp.iovw.iovec[3], "COREDUMP_RLIMIT=-1");
  44         assert_iovec_entry(&imp.iovw.iovec[4], COREDUMP_PROC_GROUP);
  45         assert_iovec_entry(&imp.iovw.iovec[5], "_SOURCE_REALTIME_TIMESTAMP=1478389147837945");
  46
  47         /* Let's check if we get EOF now */
  48         r = journal_importer_process_data(&imp);
  49         assert_se(r == 0);
  50         assert_se(journal_importer_eof(&imp));
  51 }
  52
  53 static void test_bad_input(void) {
  54         _cleanup_(journal_importer_cleanup) JournalImporter imp = {};
  55         int r;
  56
  57         imp.fd = open(get_testdata_dir("/journal-data/journal-2.txt"), O_RDONLY|O_CLOEXEC);
  58         assert_se(imp.fd >= 0);
  59
  60         do
  61                 r = journal_importer_process_data(&imp);
  62         while (!journal_importer_eof(&imp));
  63         assert_se(r == 0); /* If we don't have enough input, 0 is returned */
  64
  65         assert_se(journal_importer_eof(&imp));
  66 }
  67
  68 int main(int argc, char **argv) {
  69         log_set_max_level(LOG_DEBUG);
  70         log_parse_environment();
  71
  72         test_basic_parsing();
  73         test_bad_input();
  74
  75         return 0;
  76 }