AutoGPT

1
import time
2

3
from autogpt.app.spinner import Spinner
4

5
ALMOST_DONE_MESSAGE = "Almost done..."
6
PLEASE_WAIT = "Please wait..."
7

8

9
def test_spinner_initializes_with_default_values():
10
    """Tests that the spinner initializes with default values."""
11
    with Spinner() as spinner:
12
        assert spinner.message == "Loading..."
13
        assert spinner.delay == 0.1
14

15

16
def test_spinner_initializes_with_custom_values():
17
    """Tests that the spinner initializes with custom message and delay values."""
18
    with Spinner(message=PLEASE_WAIT, delay=0.2) as spinner:
19
        assert spinner.message == PLEASE_WAIT
20
        assert spinner.delay == 0.2
21

22

23
#
24
def test_spinner_stops_spinning():
25
    """Tests that the spinner starts spinning and stops spinning without errors."""
26
    with Spinner() as spinner:
27
        time.sleep(1)
28
    assert not spinner.running
29

30

31
def test_spinner_can_be_used_as_context_manager():
32
    """Tests that the spinner can be used as a context manager."""
33
    with Spinner() as spinner:
34
        assert spinner.running
35
    assert not spinner.running
36