[checkmk-commits] Add some basic unit tests for cmk.cpu_tracking

Module: check_mk Branch: master Commit: 82ac49a36583d662ddc9ca49eea66eaa01bc85ce URL: http://git.mathias-kettner.de/git/?p=check_mk.git;a=commit;h=82ac49a36583d6… Author: Lars Michelsen <lm(a)mathias-kettner.de> Date: Fri Jul 27 13:41:15 2018 +0200 Add some basic unit tests for cmk.cpu_tracking Change-Id: I5a27c43e092df07143e122263c1d5634b76d6355 --- tests/unit/cmk/test_cpu_tracking.py | 85 +++++++++++++++++++++++++++++++++++++ 1 file changed, 85 insertions(+) diff --git a/tests/unit/cmk/test_cpu_tracking.py b/tests/unit/cmk/test_cpu_tracking.py new file mode 100644 index 0000000..48e29ed --- /dev/null +++ b/tests/unit/cmk/test_cpu_tracking.py @@ -0,0 +1,85 @@ +#!/usr/bin/env python + +import time + +import cmk.cpu_tracking as cpu_tracking + +def test_cpu_tracking_initial_times(): + assert cpu_tracking.get_times() == {} + + +def test_cpu_tracking_initial_state(): + assert cpu_tracking._is_not_tracking() + + +def test_pop_without_tracking(): + assert cpu_tracking._is_not_tracking() + cpu_tracking.pop_phase() + assert cpu_tracking._is_not_tracking() + + +def test_push_without_tracking(): + cpu_tracking.push_phase("bla") + assert cpu_tracking._is_not_tracking() + + +def test_cpu_tracking_simple(monkeypatch): + monkeypatch.setattr("time.time", lambda: 0.0) + cpu_tracking.start("busy") + assert cpu_tracking.get_times() == {} + monkeypatch.setattr("time.time", lambda: 1.0) + cpu_tracking.end() + + times = cpu_tracking.get_times() + + assert len(times) == 2 + assert len(times["TOTAL"]) == 5 + assert times["TOTAL"][4] == 1.0 + assert times["busy"][4] == 1.0 + + +def test_cpu_tracking_multiple_phases(monkeypatch): + monkeypatch.setattr("time.time", lambda: 0.0) + cpu_tracking.start("busy") + monkeypatch.setattr("time.time", lambda: 2.0) + + cpu_tracking.push_phase("agent") + monkeypatch.setattr("time.time", lambda: 5.0) + cpu_tracking.pop_phase() + + cpu_tracking.push_phase("snmp") + monkeypatch.setattr("time.time", lambda: 7.0) + cpu_tracking.pop_phase() + + cpu_tracking.end() + + times = cpu_tracking.get_times() + assert len(times) == 4 + + assert times["TOTAL"][4] == 7.0 + assert times["busy"][4] == 2.0 + assert times["snmp"][4] == 2.0 + assert times["agent"][4] == 3.0 + + +def test_cpu_tracking_add_times(monkeypatch): + monkeypatch.setattr("time.time", lambda: 0.0) + cpu_tracking.start("busy") + monkeypatch.setattr("time.time", lambda: 2.0) + + cpu_tracking.push_phase("agent") + monkeypatch.setattr("time.time", lambda: 5.0) + cpu_tracking.pop_phase() + + cpu_tracking.push_phase("agent") + monkeypatch.setattr("time.time", lambda: 7.0) + cpu_tracking.pop_phase() + + cpu_tracking.end() + + times = cpu_tracking.get_times() + assert len(times) == 3 + + assert times["TOTAL"][4] == 7.0 + assert times["busy"][4] == 2.0 + assert times["agent"][4] == 5.0