111 lines
4.1 KiB
Python
111 lines
4.1 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# Copyright 2022 The Pigweed Authors
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
||
|
|
# use this file except in compliance with the License. You may obtain a copy of
|
||
|
|
# the License at
|
||
|
|
#
|
||
|
|
# https://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||
|
|
# License for the specific language governing permissions and limitations under
|
||
|
|
# the License.
|
||
|
|
"""Tests for the timestamp analyzer."""
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
from pw_chrono.timestamp_analyzer import process_snapshot
|
||
|
|
from pw_chrono_protos import chrono_pb2
|
||
|
|
|
||
|
|
|
||
|
|
class TimestampTest(unittest.TestCase):
|
||
|
|
"""Test for the timestamp analyzer."""
|
||
|
|
|
||
|
|
def test_no_timepoint(self):
|
||
|
|
time_stamps = chrono_pb2.SnapshotTimestamps()
|
||
|
|
self.assertEqual('', str(process_snapshot(time_stamps)))
|
||
|
|
|
||
|
|
def test_timestamp_unknown_epoch_type(self):
|
||
|
|
time_stamps = chrono_pb2.SnapshotTimestamps()
|
||
|
|
|
||
|
|
time_point = chrono_pb2.TimePoint()
|
||
|
|
unkown = chrono_pb2.EpochType.Enum.UNKNOWN
|
||
|
|
time_point.clock_parameters.epoch_type = unkown
|
||
|
|
|
||
|
|
time_stamps.timestamps.append(time_point)
|
||
|
|
|
||
|
|
expected = '\n'.join(
|
||
|
|
(
|
||
|
|
'Snapshot capture timestamp',
|
||
|
|
' Time since unknown epoch 0: unknown',
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertEqual(expected, str(process_snapshot(time_stamps)))
|
||
|
|
|
||
|
|
def test_timestamp_with_time_since_boot(self):
|
||
|
|
time_stamps = chrono_pb2.SnapshotTimestamps()
|
||
|
|
|
||
|
|
time_point = chrono_pb2.TimePoint()
|
||
|
|
time_since_boot = chrono_pb2.EpochType.Enum.TIME_SINCE_BOOT
|
||
|
|
time_point.clock_parameters.epoch_type = time_since_boot
|
||
|
|
time_point.timestamp = 100
|
||
|
|
time_point.clock_parameters.tick_period_seconds_numerator = 1
|
||
|
|
time_point.clock_parameters.tick_period_seconds_denominator = 1000
|
||
|
|
|
||
|
|
time_stamps.timestamps.append(time_point)
|
||
|
|
|
||
|
|
expected = '\n'.join(
|
||
|
|
('Snapshot capture timestamp', ' Time since boot: 2:24:00')
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertEqual(expected, str(process_snapshot(time_stamps)))
|
||
|
|
|
||
|
|
def test_timestamp_with_utc_wall_clock(self):
|
||
|
|
time_stamps = chrono_pb2.SnapshotTimestamps()
|
||
|
|
|
||
|
|
time_point = chrono_pb2.TimePoint()
|
||
|
|
utc_wall_clock = chrono_pb2.EpochType.Enum.UTC_WALL_CLOCK
|
||
|
|
time_point.clock_parameters.epoch_type = utc_wall_clock
|
||
|
|
time_point.timestamp = 100
|
||
|
|
time_point.clock_parameters.tick_period_seconds_numerator = 1
|
||
|
|
time_point.clock_parameters.tick_period_seconds_denominator = 1000
|
||
|
|
|
||
|
|
time_stamps.timestamps.append(time_point)
|
||
|
|
|
||
|
|
expected = '\n'.join(
|
||
|
|
('Snapshot capture timestamp', ' UTC time: 1970-01-01 02:24:00')
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertEqual(expected, str(process_snapshot(time_stamps)))
|
||
|
|
|
||
|
|
def test_timestamp_with_time_since_boot_and_utc_wall_clock(self):
|
||
|
|
time_stamps = chrono_pb2.SnapshotTimestamps()
|
||
|
|
|
||
|
|
time_point = chrono_pb2.TimePoint()
|
||
|
|
time_since_boot = chrono_pb2.EpochType.Enum.TIME_SINCE_BOOT
|
||
|
|
time_point.clock_parameters.epoch_type = time_since_boot
|
||
|
|
time_point.timestamp = 100
|
||
|
|
time_point.clock_parameters.tick_period_seconds_numerator = 1
|
||
|
|
time_point.clock_parameters.tick_period_seconds_denominator = 1000
|
||
|
|
time_stamps.timestamps.append(time_point)
|
||
|
|
|
||
|
|
time_point = chrono_pb2.TimePoint()
|
||
|
|
utc_wall_clock = chrono_pb2.EpochType.Enum.UTC_WALL_CLOCK
|
||
|
|
time_point.clock_parameters.epoch_type = utc_wall_clock
|
||
|
|
time_point.timestamp = 100
|
||
|
|
time_point.clock_parameters.tick_period_seconds_numerator = 1
|
||
|
|
time_point.clock_parameters.tick_period_seconds_denominator = 1000
|
||
|
|
time_stamps.timestamps.append(time_point)
|
||
|
|
|
||
|
|
expected = '\n'.join(
|
||
|
|
(
|
||
|
|
'Snapshot capture timestamps',
|
||
|
|
' Time since boot: 2:24:00',
|
||
|
|
' UTC time: 1970-01-01 02:24:00',
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertEqual(expected, str(process_snapshot(time_stamps)))
|