File: test_tuned_manager.py

package info (click to toggle)
qtile 0.34.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 5,004 kB
  • sloc: python: 49,959; ansic: 4,371; xml: 324; sh: 260; makefile: 218
file content (53 lines) | stat: -rw-r--r-- 1,670 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
from unittest.mock import MagicMock, patch

from libqtile.widget.tuned_manager import TunedManager


def test_find_mode():
    # Mocking subprocess.run to return a specific output
    with patch("subprocess.run") as mock_run:
        mock_run.return_value.stdout = "Current active profile: balanced-battery\n"

        widget = TunedManager()
        mode = widget.find_mode()

        assert mode == "balanced-battery"
        assert mock_run.call_count == 2  # Called during init and explicitly


def test_update_bar():
    with (
        patch("subprocess.run") as mock_run,
        patch.object(TunedManager, "bar", create=True) as mock_bar,
    ):
        mock_run.return_value.stdout = "Current active profile: powersave\n"
        mock_bar.draw = MagicMock()

        widget = TunedManager()
        widget.update_bar()

        assert widget.current_mode == "powersave"
        assert widget.text == "powersave"
        mock_bar.draw.assert_called_once()


def test_next_mode():
    with patch.object(TunedManager, "execute_command") as mock_execute_command:
        widget = TunedManager()
        widget.modes = ["powersave", "balanced-battery", "throughput-performance"]
        widget.current_mode = "powersave"

        widget.next_mode()

        mock_execute_command.assert_called_once_with(1)


def test_previous_mode():
    with patch.object(TunedManager, "execute_command") as mock_execute_command:
        widget = TunedManager()
        widget.modes = ["powersave", "balanced-battery", "throughput-performance"]
        widget.current_mode = "balanced-battery"

        widget.previous_mode()

        mock_execute_command.assert_called_once_with(0)