File: cocoa_time.py

package info (click to toggle)
dfdatetime 20190116-1
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 652 kB
  • sloc: python: 5,155; sh: 119; makefile: 30
file content (141 lines) | stat: -rw-r--r-- 4,928 bytes parent folder | download | duplicates (2)
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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""Tests for the Cocoa time implementation."""

from __future__ import unicode_literals

import decimal
import unittest

from dfdatetime import cocoa_time


class CocoaTimeEpochTest(unittest.TestCase):
  """Tests for the Cocoa time epoch."""

  def testInitialize(self):
    """Tests the __init__ function."""
    cocoa_time_epoch = cocoa_time.CocoaTimeEpoch()
    self.assertIsNotNone(cocoa_time_epoch)


class CocoaTimeTest(unittest.TestCase):
  """Tests for the Cocoa timestamp."""

  # pylint: disable=protected-access

  def testProperties(self):
    """Tests the properties."""
    cocoa_time_object = cocoa_time.CocoaTime(timestamp=395011845.0)
    self.assertEqual(cocoa_time_object.timestamp, 395011845.0)

    cocoa_time_object = cocoa_time.CocoaTime()
    self.assertIsNone(cocoa_time_object.timestamp)

  def testGetNormalizedTimestamp(self):
    """Tests the _GetNormalizedTimestamp function."""
    cocoa_time_object = cocoa_time.CocoaTime(timestamp=395011845.0)

    normalized_timestamp = cocoa_time_object._GetNormalizedTimestamp()
    self.assertEqual(normalized_timestamp, decimal.Decimal('1373319045.0'))

    cocoa_time_object = cocoa_time.CocoaTime()

    normalized_timestamp = cocoa_time_object._GetNormalizedTimestamp()
    self.assertIsNone(normalized_timestamp)

  def testCopyFromDateTimeString(self):
    """Tests the CopyFromDateTimeString function."""
    cocoa_time_object = cocoa_time.CocoaTime()

    expected_timestamp = 394934400.0
    cocoa_time_object.CopyFromDateTimeString('2013-07-08')
    self.assertEqual(cocoa_time_object._timestamp, expected_timestamp)

    expected_timestamp = 395011845.0
    cocoa_time_object.CopyFromDateTimeString('2013-07-08 21:30:45')
    self.assertEqual(cocoa_time_object._timestamp, expected_timestamp)

    expected_timestamp = 395011845.546875
    cocoa_time_object.CopyFromDateTimeString('2013-07-08 21:30:45.546875')
    self.assertEqual(cocoa_time_object._timestamp, expected_timestamp)

    expected_timestamp = 395015445.546875
    cocoa_time_object.CopyFromDateTimeString('2013-07-08 21:30:45.546875-01:00')
    self.assertEqual(cocoa_time_object._timestamp, expected_timestamp)

    expected_timestamp = 395008245.546875
    cocoa_time_object.CopyFromDateTimeString('2013-07-08 21:30:45.546875+01:00')
    self.assertEqual(cocoa_time_object._timestamp, expected_timestamp)

    expected_timestamp = 86400.0
    cocoa_time_object.CopyFromDateTimeString('2001-01-02 00:00:00')
    self.assertEqual(cocoa_time_object._timestamp, expected_timestamp)

  def testCopyToDateTimeString(self):
    """Tests the CopyToDateTimeString function."""
    cocoa_time_object = cocoa_time.CocoaTime(timestamp=395011845.546875)

    date_time_string = cocoa_time_object.CopyToDateTimeString()
    self.assertEqual(date_time_string, '2013-07-08 21:30:45.546875')

    epoch_year = cocoa_time_object._EPOCH.year
    cocoa_time_object._EPOCH.year = -1

    with self.assertRaises(ValueError):
      cocoa_time_object.CopyToDateTimeString()

    cocoa_time_object._EPOCH.year = epoch_year

    cocoa_time_object = cocoa_time.CocoaTime()

    date_time_string = cocoa_time_object.CopyToDateTimeString()
    self.assertIsNone(date_time_string)

  def testCopyToDateTimeStringISO8601(self):
    """Tests the CopyToDateTimeStringISO8601 function."""
    cocoa_time_object = cocoa_time.CocoaTime(timestamp=395011845.546875)

    date_time_string = cocoa_time_object.CopyToDateTimeStringISO8601()
    self.assertEqual(date_time_string, '2013-07-08T21:30:45.546875Z')

  def testGetDate(self):
    """Tests the GetDate function."""
    cocoa_time_object = cocoa_time.CocoaTime(timestamp=395011845.546875)

    date_tuple = cocoa_time_object.GetDate()
    self.assertEqual(date_tuple, (2013, 7, 8))

    cocoa_time_object = cocoa_time.CocoaTime()

    date_tuple = cocoa_time_object.GetDate()
    self.assertEqual(date_tuple, (None, None, None))

  # TODO: remove this method when there is no more need for it in plaso.
  def testGetPlasoTimestamp(self):
    """Tests the GetPlasoTimestamp function."""
    cocoa_time_object = cocoa_time.CocoaTime(timestamp=395011845.0)

    micro_posix_timestamp = cocoa_time_object.GetPlasoTimestamp()
    self.assertEqual(micro_posix_timestamp, 1373319045000000)

    cocoa_time_object = cocoa_time.CocoaTime()

    micro_posix_timestamp = cocoa_time_object.GetPlasoTimestamp()
    self.assertIsNone(micro_posix_timestamp)

  def testGetTimeOfDay(self):
    """Tests the GetTimeOfDay function."""
    cocoa_time_object = cocoa_time.CocoaTime(timestamp=395011845.546875)

    time_of_day_tuple = cocoa_time_object.GetTimeOfDay()
    self.assertEqual(time_of_day_tuple, (21, 30, 45))

    cocoa_time_object = cocoa_time.CocoaTime()

    time_of_day_tuple = cocoa_time_object.GetTimeOfDay()
    self.assertEqual(time_of_day_tuple, (None, None, None))


if __name__ == '__main__':
  unittest.main()