Skip to content

Commit

Permalink
Merge pull request #5818 from MetRonnie/unit-test
Browse files Browse the repository at this point in the history
Rewrite doctest as unit test
  • Loading branch information
oliver-sanders authored Feb 23, 2024
2 parents 7a813eb + c2e91ac commit acb164f
Show file tree
Hide file tree
Showing 3 changed files with 59 additions and 53 deletions.
2 changes: 2 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -109,3 +109,5 @@ Contributions welcome:
[admin pages](https://cylc.github.io/cylc-admin/#change-proposals).
* Touch base in the
[developers chat](https://matrix.to/#/#cylc-general:matrix.org).

This repository contains some code that was generated by GitHub Copilot.
54 changes: 1 addition & 53 deletions cylc/flow/timer.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,59 +26,7 @@


class Timer:
"""Simple timer class for workflow timers.
Examples:
>>> from time import sleep
# Patch LOG.warning to capture warning messages
>>> warnings = []
>>> orig_log_warning = LOG.warning
>>> LOG.warning = lambda msg: warnings.append(msg)
>>> timer = Timer("bob timeout", 1.0)
# timer attributes
>>> timer.name
'bob timer'
>>> timer.interval
'PT1S'
# start timer
>>> timer.reset()
>>> warnings[-1]
'PT1S bob timer starts NOW'
# check timeout
>>> sleep(2)
>>> timer.timed_out()
True
>>> warnings[-1]
'bob timer timed out after PT1S'
# stop should do nothing after timeou
>>> warnings = []
>>> timer.stop()
>>> warnings
[]
# start timer again, then stop it
>>> timer.reset()
>>> warnings[-1]
'PT1S bob timer starts NOW'
>>> timer.stop()
>>> warnings[-1]
'bob timer stopped'
# another stop should do nothing
>>> warnings = []
>>> timer.stop()
>>> warnings
[]
# Un-patch LOG.warning to avoid breaking subsequent unit tests!
>>> LOG.warning = orig_log_warning
"""
"""Simple timer class for workflow timers."""

def __init__(
self, name: str, interval: float,
Expand Down
56 changes: 56 additions & 0 deletions tests/unit/test_timer.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,56 @@
# THIS FILE IS PART OF THE CYLC WORKFLOW ENGINE.
# Copyright (C) NIWA & British Crown (Met Office) & Contributors.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.

from time import sleep

import pytest

from cylc.flow.timer import Timer


def test_Timer(caplog: pytest.LogCaptureFixture):
"""Test the Timer class."""
caplog.set_level('WARNING')
timer = Timer("bob timeout", 1.0)

# timer attributes
assert timer.name == "bob timer"
assert timer.interval == "PT1S"

# start timer
timer.reset()
assert caplog.records[-1].message == "PT1S bob timer starts NOW"

# check timeout
sleep(2)
assert timer.timed_out()
assert caplog.records[-1].message == "bob timer timed out after PT1S"

# stop should do nothing after timeout
caplog.clear()
timer.stop()
assert not caplog.records

# start timer again, then stop it
timer.reset()
assert caplog.records[-1].message == "PT1S bob timer starts NOW"
timer.stop()
assert caplog.records[-1].message == "bob timer stopped"

# another stop should do nothing
caplog.clear()
timer.stop()
assert not caplog.records

0 comments on commit acb164f

Please sign in to comment.