src/test/test-calendarspec.c

   1 /***
   2   This file is part of systemd.
   3
   4   Copyright 2012 Lennart Poettering
   5
   6   systemd is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as published by
   8   the Free Software Foundation; either version 2.1 of the License, or
   9   (at your option) any later version.
  10
  11   systemd is distributed in the hope that it will be useful, but
  12   WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14   Lesser General Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public License
  17   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  18 ***/
  19
  20 #include <string.h>
  21
  22 #include "alloc-util.h"
  23 #include "calendarspec.h"
  24 #include "string-util.h"
  25 #include "util.h"
  26
  27 static void test_one(const char *input, const char *output) {
  28         CalendarSpec *c;
  29         _cleanup_free_ char *p = NULL, *q = NULL;
  30         usec_t u;
  31         char buf[FORMAT_TIMESTAMP_MAX];
  32         int r;
  33
  34         assert_se(calendar_spec_from_string(input, &c) >= 0);
  35
  36         assert_se(calendar_spec_to_string(c, &p) >= 0);
  37         printf("\"%s\" → \"%s\"\n", input, p);
  38
  39         assert_se(streq(p, output));
  40
  41         u = now(CLOCK_REALTIME);
  42         r = calendar_spec_next_usec(c, u, &u);
  43         printf("Next: %s\n", r < 0 ? strerror(-r) : format_timestamp(buf, sizeof(buf), u));
  44         calendar_spec_free(c);
  45
  46         assert_se(calendar_spec_from_string(p, &c) >= 0);
  47         assert_se(calendar_spec_to_string(c, &q) >= 0);
  48         calendar_spec_free(c);
  49
  50         assert_se(streq(q, p));
  51 }
  52
  53 static void test_next(const char *input, const char *new_tz, usec_t after, usec_t expect) {
  54         CalendarSpec *c;
  55         usec_t u;
  56         char *old_tz;
  57         char buf[FORMAT_TIMESTAMP_MAX];
  58         int r;
  59
  60         old_tz = getenv("TZ");
  61         if (old_tz)
  62                 old_tz = strdupa(old_tz);
  63
  64         if (new_tz)
  65                 assert_se(setenv("TZ", new_tz, 1) >= 0);
  66         else
  67                 assert_se(unsetenv("TZ") >= 0);
  68         tzset();
  69
  70         assert_se(calendar_spec_from_string(input, &c) >= 0);
  71
  72         printf("\"%s\"\n", input);
  73
  74         u = after;
  75         r = calendar_spec_next_usec(c, after, &u);
  76         printf("At: %s\n", r < 0 ? strerror(-r) : format_timestamp_us(buf, sizeof buf, u));
  77         if (expect != (usec_t)-1)
  78                 assert_se(r >= 0 && u == expect);
  79         else
  80                 assert(r == -ENOENT);
  81
  82         calendar_spec_free(c);
  83
  84         if (old_tz)
  85                 assert_se(setenv("TZ", old_tz, 1) >= 0);
  86         else
  87                 assert_se(unsetenv("TZ") >= 0);
  88         tzset();
  89 }
  90
  91 static void test_timestamp(void) {
  92         char buf[FORMAT_TIMESTAMP_MAX];
  93         _cleanup_free_ char *t = NULL;
  94         CalendarSpec *c;
  95         usec_t x, y;
  96
  97         /* Ensure that a timestamp is also a valid calendar specification. Convert forth and back */
  98
  99         x = now(CLOCK_REALTIME);
 100
 101         assert_se(format_timestamp_us(buf, sizeof(buf), x));
 102         printf("%s\n", buf);
 103         assert_se(calendar_spec_from_string(buf, &c) >= 0);
 104         assert_se(calendar_spec_to_string(c, &t) >= 0);
 105         calendar_spec_free(c);
 106         printf("%s\n", t);
 107
 108         assert_se(parse_timestamp(t, &y) >= 0);
 109         assert_se(y == x);
 110 }
 111
 112 static void test_hourly_bug_4031(void) {
 113         CalendarSpec *c;
 114         usec_t n, u, w;
 115         char buf[FORMAT_TIMESTAMP_MAX], zaf[FORMAT_TIMESTAMP_MAX];
 116         int r;
 117
 118         assert_se(calendar_spec_from_string("hourly", &c) >= 0);
 119         n = now(CLOCK_REALTIME);
 120         assert_se((r = calendar_spec_next_usec(c, n, &u)) >= 0);
 121
 122         printf("Now: %s (%"PRIu64")\n", format_timestamp_us(buf, sizeof buf, n), n);
 123         printf("Next hourly: %s (%"PRIu64")\n", r < 0 ? strerror(-r) : format_timestamp_us(buf, sizeof buf, u), u);
 124
 125         assert_se((r = calendar_spec_next_usec(c, u, &w)) >= 0);
 126         printf("Next hourly: %s (%"PRIu64")\n", r < 0 ? strerror(-r) : format_timestamp_us(zaf, sizeof zaf, w), w);
 127
 128         assert_se(n < u);
 129         assert_se(u <= n + USEC_PER_HOUR);
 130         assert_se(u < w);
 131         assert_se(w <= u + USEC_PER_HOUR);
 132 }
 133
 134 int main(int argc, char* argv[]) {
 135         CalendarSpec *c;
 136
 137         test_one("Sat,Thu,Mon-Wed,Sat-Sun", "Mon..Thu,Sat,Sun *-*-* 00:00:00");
 138         test_one("Sat,Thu,Mon..Wed,Sat..Sun", "Mon..Thu,Sat,Sun *-*-* 00:00:00");
 139         test_one("Mon,Sun 12-*-* 2,1:23", "Mon,Sun 2012-*-* 01,02:23:00");
 140         test_one("Wed *-1", "Wed *-*-01 00:00:00");
 141         test_one("Wed-Wed,Wed *-1", "Wed *-*-01 00:00:00");
 142         test_one("Wed..Wed,Wed *-1", "Wed *-*-01 00:00:00");
 143         test_one("Wed, 17:48", "Wed *-*-* 17:48:00");
 144         test_one("Wed-Sat,Tue 12-10-15 1:2:3", "Tue..Sat 2012-10-15 01:02:03");
 145         test_one("Wed..Sat,Tue 12-10-15 1:2:3", "Tue..Sat 2012-10-15 01:02:03");
 146         test_one("*-*-7 0:0:0", "*-*-07 00:00:00");
 147         test_one("10-15", "*-10-15 00:00:00");
 148         test_one("monday *-12-* 17:00", "Mon *-12-* 17:00:00");
 149         test_one("Mon,Fri *-*-3,1,2 *:30:45", "Mon,Fri *-*-01,02,03 *:30:45");
 150         test_one("12,14,13,12:20,10,30", "*-*-* 12,13,14:10,20,30:00");
 151         test_one("mon,fri *-1/2-1,3 *:30:45", "Mon,Fri *-01/2-01,03 *:30:45");
 152         test_one("03-05 08:05:40", "*-03-05 08:05:40");
 153         test_one("08:05:40", "*-*-* 08:05:40");
 154         test_one("05:40", "*-*-* 05:40:00");
 155         test_one("Sat,Sun 12-05 08:05:40", "Sat,Sun *-12-05 08:05:40");
 156         test_one("Sat,Sun 08:05:40", "Sat,Sun *-*-* 08:05:40");
 157         test_one("2003-03-05 05:40", "2003-03-05 05:40:00");
 158         test_one("2003-03-05", "2003-03-05 00:00:00");
 159         test_one("03-05", "*-03-05 00:00:00");
 160         test_one("hourly", "*-*-* *:00:00");
 161         test_one("daily", "*-*-* 00:00:00");
 162         test_one("monthly", "*-*-01 00:00:00");
 163         test_one("weekly", "Mon *-*-* 00:00:00");
 164         test_one("minutely", "*-*-* *:*:00");
 165         test_one("quarterly", "*-01,04,07,10-01 00:00:00");
 166         test_one("semi-annually", "*-01,07-01 00:00:00");
 167         test_one("annually", "*-01-01 00:00:00");
 168         test_one("*:2/3", "*-*-* *:02/3:00");
 169         test_one("2015-10-25 01:00:00 uTc", "2015-10-25 01:00:00 UTC");
 170         test_one("2016-03-27 03:17:00.4200005", "2016-03-27 03:17:00.420001");
 171         test_one("2016-03-27 03:17:00/0.42", "2016-03-27 03:17:00/0.420000");
 172         test_one("2016-03-27 03:17:00/0.42", "2016-03-27 03:17:00/0.420000");
 173         test_one("9..11,13:00,30", "*-*-* 09,10,11,13:00,30:00");
 174         test_one("1..3-1..3 1..3:1..3", "*-01,02,03-01,02,03 01,02,03:01,02,03:00");
 175         test_one("00:00:1.125..2.125", "*-*-* 00:00:01.125000,02.125000");
 176         test_one("00:00:1.0..3.8", "*-*-* 00:00:01,02,03");
 177
 178         test_next("2016-03-27 03:17:00", "", 12345, 1459048620000000);
 179         test_next("2016-03-27 03:17:00", "CET", 12345, 1459041420000000);
 180         test_next("2016-03-27 03:17:00", "EET", 12345, -1);
 181         test_next("2016-03-27 03:17:00 UTC", NULL, 12345, 1459048620000000);
 182         test_next("2016-03-27 03:17:00 UTC", "", 12345, 1459048620000000);
 183         test_next("2016-03-27 03:17:00 UTC", "CET", 12345, 1459048620000000);
 184         test_next("2016-03-27 03:17:00 UTC", "EET", 12345, 1459048620000000);
 185         test_next("2016-03-27 03:17:00.420000001 UTC", "EET", 12345, 1459048620420000);
 186         test_next("2016-03-27 03:17:00.4200005 UTC", "EET", 12345, 1459048620420001);
 187         test_next("2015-11-13 09:11:23.42", "EET", 12345, 1447398683420000);
 188         test_next("2015-11-13 09:11:23.42/1.77", "EET", 1447398683420000, 1447398685190000);
 189         test_next("2015-11-13 09:11:23.42/1.77", "EET", 1447398683419999, 1447398683420000);
 190         test_next("Sun 16:00:00", "CET", 1456041600123456, 1456066800000000);
 191
 192         assert_se(calendar_spec_from_string("test", &c) < 0);
 193         assert_se(calendar_spec_from_string("", &c) < 0);
 194         assert_se(calendar_spec_from_string("7", &c) < 0);
 195         assert_se(calendar_spec_from_string("121212:1:2", &c) < 0);
 196         assert_se(calendar_spec_from_string("2000-03-05.23 00:00:00", &c) < 0);
 197         assert_se(calendar_spec_from_string("2000-03-05 00:00.1:00", &c) < 0);
 198         assert_se(calendar_spec_from_string("00:00:00/0.00000001", &c) < 0);
 199         assert_se(calendar_spec_from_string("00:00:00.0..00.9", &c) < 0);
 200
 201         test_timestamp();
 202         test_hourly_bug_4031();
 203
 204         return 0;
 205 }