summaryrefslogtreecommitdiff
path: root/test/test_date_time.ml
blob: 6f58b1ef60af24b3d70718db01a4fb4d3f609913 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
(*---------------------------------------------------------------------------
   Copyright (c) 2015 The ptime programmers. All rights reserved.
   Distributed under the ISC license, see terms at the end of the file.
  ---------------------------------------------------------------------------*)

open Testing
open Testing_ptime

let stamp_of_date_time d =
  (Ptime.of_date_time $ raw_date_time @-> ret_get_option stamp) d

let valid_date_time d =
  ignore ((Ptime.of_date_time $ raw_date_time @-> ret_some stamp) d)

let wrong_date_time d =
  ignore ((Ptime.of_date_time $ raw_date_time @-> ret_none stamp) d)

let span_of_d_ps s = match Ptime.Span.of_d_ps s with
| None -> invalid_arg ""
| Some s -> s

let time_bounds = test "Date-time time field bounds" @@ fun () ->
  let min_date = Ptime.to_date Ptime.min in
  let min_utc t = min_date, (t, 0) in
  (* Check hour bounds *)
  wrong_date_time (min_utc (-2, 00, 00));
  wrong_date_time (min_utc (-1, 00, 00));
  valid_date_time (min_utc (00, 00, 00));
  valid_date_time (min_utc (01, 00, 00));
  valid_date_time (min_utc (23, 00, 00));
  wrong_date_time (min_utc (24, 00, 00));
  (* Check minute bounds *)
  wrong_date_time (min_utc (00, -2, 00));
  wrong_date_time (min_utc (00, -1, 00));
  valid_date_time (min_utc (00, 00, 00));
  valid_date_time (min_utc (00, 01, 00));
  valid_date_time (min_utc (00, 59, 00));
  wrong_date_time (min_utc (00, 60, 00));
  (* Check second bounds *)
  wrong_date_time (min_utc (00, 00, -2));
  wrong_date_time (min_utc (00, 00, -1));
  valid_date_time (min_utc (00, 00, 00));
  valid_date_time (min_utc (00, 00, 01));
  valid_date_time (min_utc (00, 00, 59));
  valid_date_time (min_utc (00, 00, 60));
  wrong_date_time (min_utc (00, 00, 61));
  ()

let tz = test "Testing date-time time zone calculations" @@ fun () ->
  (* Timestamps with tz offsets around Ptime.{max,min} *)
  wrong_date_time ((0000, 01, 01), ((00, 00, 00), +1));
  valid_date_time ((0000, 01, 01), ((00, 00, 00), +0));
  valid_date_time ((0000, 01, 01), ((00, 00, 00), -1));
  wrong_date_time ((9999, 12, 31), ((23, 59, 59), -1));
  wrong_date_time ((9999, 12, 31), ((23, 59, 60), +0));
  valid_date_time ((9999, 12, 31), ((23, 59, 60), +1));
  (* Convert time zones *)
  let nyc_tz = -4 * 3600 in
  let cam_tz = +1 * 3600 in
  let lau_tz = +2 * 3600 in
  let new_york  = ((2015, 06, 27), ((18, 30, 01), nyc_tz)) in
  let cambridge = ((2015, 06, 27), ((23, 30, 01), cam_tz)) in
  let lausanne  = ((2015, 06, 28), ((00, 30, 01), lau_tz)) in
  let nyc_stamp = stamp_of_date_time new_york in
  let cam_stamp = stamp_of_date_time cambridge in
  let lau_stamp = stamp_of_date_time lausanne in
  eq_stamp nyc_stamp cam_stamp;
  eq_stamp cam_stamp lau_stamp;
  eq_date_time (Ptime.to_date_time ~tz_offset_s:nyc_tz nyc_stamp) new_york;
  eq_date_time (Ptime.to_date_time ~tz_offset_s:cam_tz nyc_stamp) cambridge;
  eq_date_time (Ptime.to_date_time ~tz_offset_s:lau_tz nyc_stamp) lausanne;
  ()

let subsecond = test "Subsecond stamp to date-time" @@ fun () ->
  let add, sub =
    let add t ps = Ptime.(add_span t (span_of_d_ps (0, ps))) in
    let sub t ps = Ptime.(sub_span t (span_of_d_ps (0, ps))) in
    add $ stamp @-> int64 @-> ret_get_option stamp,
    sub $ stamp @-> int64 @-> ret_get_option stamp
  in
  let b0 = sub Ptime.epoch 750_000_000_000L in
  let b1 = sub Ptime.epoch 500_000_000_000L in
  let b2 = sub Ptime.epoch 250_000_000_000L in
  let b = (1969, 12, 31), ((23, 59, 59), +0) in
  eq_date_time b (Ptime.to_date_time b0);
  eq_date_time b (Ptime.to_date_time b1);
  eq_date_time b (Ptime.to_date_time b2);
  let a0 = add Ptime.epoch 750_000_000_000L in
  let a1 = add Ptime.epoch 500_000_000_000L in
  let a2 = add Ptime.epoch 250_000_000_000L in
  let a = (1970, 01, 01), ((00, 00, 00), +0) in
  eq_date_time a (Ptime.to_date_time a0);
  eq_date_time a (Ptime.to_date_time a1);
  eq_date_time a (Ptime.to_date_time a2);
  ()

let leap_sec = test "Testing leap second date-times" @@ fun () ->
  let after_leap_sec = (1999, 01, 01), ((00, 00, 00), 0) in
  let t0 = stamp_of_date_time ((1998, 12, 31), ((23, 59, 59), 0)) in
  let t1 = stamp_of_date_time ((1998, 12, 31), ((23, 59, 60), 0)) in
  let t2 = stamp_of_date_time after_leap_sec in
  eq_stamp t1 t2; (* leap sec is represented by second that comes after *)
  eq_stamp_opt (Some t1) Ptime.(add_span t0 (Span.of_int_s 1));
  eq_date_time after_leap_sec (Ptime.to_date_time t1);
  eq_date_time after_leap_sec (Ptime.to_date_time t2);
  eq_span (Ptime.diff t2 t0) (Ptime.Span.of_int_s 1);
  eq_span (Ptime.diff t1 t0) (Ptime.Span.of_int_s 1);
  eq_span (Ptime.diff t2 t1) (Ptime.Span.of_int_s 0);
  ()

let stamp_trips = test "Random stamps to date-time round trips" @@ fun () ->
  let stamp_of_posix_s =
    Ptime.of_float_s $ pp_float @-> (ret_get_option stamp)
  in
  let trip ?tz_offset_s t =
    let back = stamp_of_posix_s (floor (Ptime.to_float_s t)) in
    let trip = stamp_of_date_time (Ptime.to_date_time ?tz_offset_s t) in
    eq_stamp back trip
  in
  for i = 1 to Test_rand.loop_len () do
    trip ~tz_offset_s:0 (* UTC *) (Test_rand.float_stamp ());
    trip ~tz_offset_s:(Test_rand.tz_offset_s ()) (Test_rand.float_stamp ())
  done

let round_trips =
  test "Random valid date-times to stamp round trips" @@ fun () ->
  let is_leap_sec = function
  | (_, _, _), ((_, _, 60), _) -> true
  | _ -> false
  in
  let rec rand_date_time_stamp () = (* biased *)
    let date = Test_rand.date () in
    let time = Test_rand.time () in
    let tz = Test_rand.tz_offset_s () in
    let dt = (date, (time, tz)) in
    match Ptime.of_date_time dt with
    | Some _ -> dt
    | None ->
        let dt = date, (time, 0) (* try in UTC *) in
        begin match Ptime.of_date_time dt with
        | None -> rand_date_time_stamp () (* start again *)
        | Some _ -> dt
        end
  in
  let add_posix_s =
    let span = Ptime.Span.of_float_s $ pp_float @-> ret_get_option span in
    let add_posix_s t s = Ptime.(add_span t (span s)) in
    add_posix_s $ stamp @-> pp_float @-> ret_get_option stamp
  in
  for i = 1 to Test_rand.loop_len () do
    let (_, (_, tz_offset_s) as dt) = rand_date_time_stamp () in
    let stamp = stamp_of_date_time dt in
    if not (is_leap_sec dt)
    then eq_date_time dt (Ptime.to_date_time ~tz_offset_s stamp)
    else begin
      (* Verify we map the leap sec on the the second after. *)
      let before_leap_dt = match dt with
      | date, ((hh, ss, 60), tz) -> date, ((hh, ss, 59), tz)
      | _ -> assert false
      in
      let stamp' = add_posix_s (stamp_of_date_time before_leap_dt) 1. in
      eq_stamp stamp stamp'
    end
  done;
  ()

let weekday =
  test "Ptime.weekday" @@ fun () ->
  let pp_weekday ppf v = Format.pp_print_string ppf begin match v with
    | `Mon -> "`Mon" | `Tue -> "`Tue" | `Wed -> "`Wed" | `Thu -> "`Thu"
    | `Fri -> "`Fri" | `Sat -> "`Sat" | `Sun -> "`Sun"
    end
  in
  let eq_weekday v v' = eq ~eq:(=) ~pp:pp_weekday v v' in
  let eq ?tz_offset_s c wday =
    eq_weekday (Ptime.weekday ?tz_offset_s
                  (stamp_of_date_time (c, ((0, 0, 0), 0)))) wday
  in
  eq (1970, 01, 01) `Thu;
  eq ~tz_offset_s:(-1) (1970, 01, 01) `Wed;
  eq ~tz_offset_s:86400 (1970, 01, 01) `Fri;
  eq (1871, 03, 18) `Sat;
  eq ~tz_offset_s:(-1) (1871, 03, 18) `Fri;
  eq ~tz_offset_s:86400 (1871, 03, 18) `Sun;
  eq (1995, 09, 12) `Tue;
  eq ~tz_offset_s:(-1) (1995, 09, 12) `Mon;
  eq ~tz_offset_s:86400 (1995, 09, 12) `Wed;
  eq ~tz_offset_s:172800 (1995, 09, 12) `Thu;
  ()

let suite = suite "Ptime date-time tests"
    [ time_bounds;
      tz;
      subsecond;
      leap_sec;
      stamp_trips;
      round_trips;
      weekday; ]

(*---------------------------------------------------------------------------
   Copyright (c) 2015 The ptime programmers

   Permission to use, copy, modify, and/or distribute this software for any
   purpose with or without fee is hereby granted, provided that the above
   copyright notice and this permission notice appear in all copies.

   THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
   MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
   ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
   WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
   ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
   OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  ---------------------------------------------------------------------------*)