27 lines
634 B
Python
27 lines
634 B
Python
|
import asyncio
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from libs.RestartController import RestartController
|
||
|
|
|
||
|
|
|
||
|
|
def test_restart_controller_initial_state():
|
||
|
|
rc = RestartController()
|
||
|
|
assert rc.consume() is False
|
||
|
|
|
||
|
|
|
||
|
|
def test_restart_controller_request_and_consume_once():
|
||
|
|
rc = RestartController()
|
||
|
|
rc.request()
|
||
|
|
assert rc.consume() is True
|
||
|
|
# Consuming again should be False until re-requested
|
||
|
|
assert rc.consume() is False
|
||
|
|
|
||
|
|
|
||
|
|
def test_restart_controller_idempotent_requests():
|
||
|
|
rc = RestartController()
|
||
|
|
rc.request()
|
||
|
|
rc.request()
|
||
|
|
# Still only one pending event
|
||
|
|
assert rc.consume() is True
|
||
|
|
assert rc.consume() is False
|