#!/usr/bin/env vpython3
# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""File for testing modification_waiter.py."""
import unittest
import unittest.mock as mock
from modification_waiter import ModificationWaiter
class ModificationWaiterTest(unittest.TestCase):
"""Test ModificationWaiter."""
@mock.patch('modification_waiter.time.sleep')
@mock.patch('modification_waiter.time.time', side_effect=[0, 60])
def test_timeout(self, time_patch, sleep_patch) -> None:
"""The behavior when timeout happens, it shouldn't call other functions
in the case."""
ModificationWaiter('/').__exit__(None, None, None)
self.assertEqual(time_patch.call_count, 2)
self.assertEqual(sleep_patch.call_count, 0)
@mock.patch('modification_waiter.time.sleep')
@mock.patch('modification_waiter.time.time', side_effect=[0, 5])
@mock.patch('modification_waiter.os.path.getmtime', side_effect=[0])
def test_quiet(self, getmtime_patch, time_patch, sleep_patch) -> None:
"""The behavior when no modifications happen during the last 5 seconds.
"""
ModificationWaiter('/').__exit__(None, None, None)
self.assertEqual(time_patch.call_count, 2)
self.assertEqual(getmtime_patch.call_count, 1)
self.assertEqual(sleep_patch.call_count, 0)
@mock.patch('modification_waiter.time.sleep')
@mock.patch('modification_waiter.time.time', side_effect=[5, 10])
@mock.patch('modification_waiter.os.path.getmtime', side_effect=[0])
def test_mod_before_now(self, getmtime_patch, time_patch,
sleep_patch) -> None:
"""The behavior when no modifications happen before current time.
"""
ModificationWaiter('/').__exit__(None, None, None)
self.assertEqual(time_patch.call_count, 2)
self.assertEqual(getmtime_patch.call_count, 1)
self.assertEqual(sleep_patch.call_count, 0)
@mock.patch('modification_waiter.time.sleep')
@mock.patch('modification_waiter.time.time', side_effect=[0, 5, 10])
@mock.patch('modification_waiter.os.path.getmtime', side_effect=[5, 5])
def test_mod_after_now(self, getmtime_patch, time_patch,
sleep_patch) -> None:
"""The behavior when a modification happens after current time.
"""
ModificationWaiter('/').__exit__(None, None, None)
self.assertEqual(time_patch.call_count, 3)
self.assertEqual(getmtime_patch.call_count, 2)
self.assertEqual(sleep_patch.call_count, 1)
@mock.patch('modification_waiter.time.sleep')
@mock.patch('modification_waiter.time.time', side_effect=[0, 5, 10, 15])
@mock.patch('modification_waiter.os.path.getmtime',
side_effect=[5, 10, 10])
def test_mod_twice_after_now(self, getmtime_patch, time_patch,
sleep_patch) -> None:
"""The behavior when a modification happens after current time.
"""
ModificationWaiter('/').__exit__(None, None, None)
self.assertEqual(time_patch.call_count, 4)
self.assertEqual(getmtime_patch.call_count, 3)
self.assertEqual(sleep_patch.call_count, 2)
@mock.patch('modification_waiter.time.sleep')
@mock.patch('modification_waiter.time.time', side_effect=[10, 5, 10, 15])
@mock.patch('modification_waiter.os.path.getmtime',
side_effect=[5, 10, 10])
def test_decreased_time(self, getmtime_patch, time_patch,
sleep_patch) -> None:
"""The behavior when time.time() returns decreased values."""
ModificationWaiter('/').__exit__(None, None, None)
self.assertEqual(time_patch.call_count, 4)
self.assertEqual(getmtime_patch.call_count, 3)
self.assertEqual(sleep_patch.call_count, 2)
@mock.patch('modification_waiter.time.sleep')
@mock.patch('modification_waiter.time.time', side_effect=[0, 5, 10, 15])
@mock.patch('modification_waiter.os.path.getmtime', side_effect=[5, 10, 5])
def test_decreased_mtime(self, getmtime_patch, time_patch,
sleep_patch) -> None:
"""The behavior when path.getmtime returns decreased values."""
ModificationWaiter('/').__exit__(None, None, None)
self.assertEqual(time_patch.call_count, 4)
self.assertEqual(getmtime_patch.call_count, 3)
self.assertEqual(sleep_patch.call_count, 2)
if __name__ == '__main__':
unittest.main()