File: test_plugin.py

package info (click to toggle)
pytest-freezer 0.4.9-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 92 kB
  • sloc: python: 52; makefile: 7; sh: 5
file content (97 lines) | stat: -rw-r--r-- 2,761 bytes parent folder | download
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
def test_freezer_move_to_method(pytester):
    pytester.makepyfile(
        """
        from datetime import datetime

        def test_freezer_move_to_method(freezer):
            freezer.move_to("2021-01-01")
            assert datetime.now().year == 2021
            freezer.move_to("2022-01-01")
            assert datetime.now().year == 2022
        """
    )
    pytester.runpytest().assert_outcomes(passed=1)


def test_mark_decorator(pytester):
    pytester.makepyfile(
        """
        import pytest
        from datetime import datetime
        
        @pytest.mark.freeze_time("2022-10-17T12:34:56")
        def test_mark_decorator():
            assert datetime.now().isoformat() == "2022-10-17T12:34:56"
        """
    )
    pytester.runpytest().assert_outcomes(passed=1)


def test_frozen_by_default_with_fixture(pytester):
    pytester.makepyfile(
        """
        import time
        from datetime import datetime

        def test_frozen_by_default_with_fixture(freezer):
            dt1 = datetime.now()
            time.sleep(0.01)
            dt2 = datetime.now()
            assert dt1 == dt2
        """
    )
    pytester.runpytest().assert_outcomes(passed=1)


def test_not_frozen_with_no_fixture(pytester):
    pytester.makepyfile(
        """
        import time
        from datetime import datetime

        def test_not_frozen_with_no_fixture():
            dt1 = datetime.now()
            time.sleep(0.01)
            dt2 = datetime.now()
            assert dt1 < dt2
        """
    )
    pytester.runpytest().assert_outcomes(passed=1)


def test_tick_method(pytester):
    pytester.makepyfile(
        """
        from datetime import datetime
        
        def test_tick_method(freezer):
            freezer.move_to("2022-10-17")
            assert datetime.now() == datetime(2022, 10, 17)
            freezer.tick()
            assert datetime.now() == datetime(2022, 10, 17, 0, 0, 1)
            freezer.tick(delta=12)
            assert datetime.now() == datetime(2022, 10, 17, 0, 0, 13)
        """
    )
    pytester.runpytest().assert_outcomes(passed=1)


def test_durations(pytester):
    pytester.makepyfile(
        """
        def test_durations(freezer):
            freezer.move_to('2000-01-01')
        """
    )
    result = pytester.runpytest("--durations=3", "-vv")
    durations = {}
    for line in result.outlines:
        if "test_durations.py::test_durations" not in line:
            continue
        for stage in "setup", "call", "teardown":
            if stage in line:
                duration = line.split()[0]
                durations[stage] = float(duration[:-1])
    assert 0 <= durations["setup"] <= 1
    assert 0 <= durations["call"] <= 1
    assert 0 <= durations["teardown"] <= 1