134 lines
4.0 KiB
Python
134 lines
4.0 KiB
Python
import patch_put
|
|
import pytest
|
|
import time
|
|
import threading
|
|
import builtins
|
|
from io import StringIO
|
|
import colorama
|
|
from contextlib import contextmanager
|
|
|
|
from slic.utils.pv import PV
|
|
from morbidissimo import MorIOC
|
|
'''
|
|
# Utils
|
|
@pytest.fixture
|
|
def capture_stdout(monkeypatch):
|
|
# Fixture to capture sys.stdout into a buffer
|
|
buf = StringIO()
|
|
monkeypatch.setattr("sys.stdout", buf)
|
|
return buf
|
|
|
|
@contextmanager
|
|
def use_callback(pv, callback):
|
|
# Context manager to add and remove a callback around a block
|
|
cbid = pv.add_callback(callback)
|
|
try:
|
|
yield
|
|
finally:
|
|
pv.remove_callback(cbid)
|
|
|
|
import re
|
|
|
|
def strip_ansi(text):
|
|
# Remove ANSI escape sequences from text
|
|
ansi_escape = re.compile(r'\x1b\[[0-9;]*m')
|
|
return ansi_escape.sub('', text)
|
|
|
|
# Tests
|
|
@pytest.mark.parametrize("value_new, value_before, show_bar, expected_color", [
|
|
(25, 0, True, colorama.Fore.GREEN),
|
|
(50, 25, True, colorama.Fore.GREEN),
|
|
(75, 50, False, colorama.Fore.GREEN),
|
|
(100, 75, True, colorama.Fore.GREEN),
|
|
(150, 100, False, colorama.Fore.GREEN),
|
|
(-50, 150, True, colorama.Fore.GREEN)
|
|
])
|
|
def test_put_with_progress_and_repr(value_new, value_before, show_bar, expected_color):
|
|
pv = PV("TEST:VAL")
|
|
|
|
pv.put(value_before)
|
|
assert pv.get() == value_before
|
|
|
|
# Capture all printed lines in a list
|
|
printed_lines = []
|
|
|
|
def fake_print(*args, **kwargs):
|
|
line = " ".join(str(a) for a in args)
|
|
printed_lines.append(line)
|
|
|
|
# Monkeypatch print only inside this context
|
|
original_print = builtins.print
|
|
builtins.print = fake_print
|
|
try:
|
|
pv.put(value_new, show_progress=show_bar)
|
|
finally:
|
|
builtins.print = original_print
|
|
|
|
printed_lines = [line for line in printed_lines if line.strip()]
|
|
cleaned_lines = [strip_ansi(line) for line in printed_lines]
|
|
|
|
# Check progress bar output
|
|
if show_bar is True:
|
|
# Verify empty bar was shown at least once
|
|
matches = [line for line in cleaned_lines if f"| |" in line]
|
|
assert matches, f"Expected empty bar not found in:\n" + "\n".join(printed_lines)
|
|
|
|
# Verify filled bar was shown at least once
|
|
matches = [line for line in cleaned_lines if f"|██████████████████████████████|" in line]
|
|
assert matches, f"Expected filled bar not found in:\n" + "\n".join(printed_lines)
|
|
|
|
# Verify color code was used at least once
|
|
color_matches = [line for line in printed_lines if expected_color in line]
|
|
assert color_matches, "Expected color code not found"
|
|
|
|
# Verify old and new values appear in printed lines
|
|
assert all(f"{value_new}" in line for line in printed_lines), "new value not in all lines"
|
|
assert all(f"{value_before}" in line for line in printed_lines), "old value not in all lines"
|
|
|
|
else:
|
|
# When show_bar=False, nothing should be printed
|
|
assert len(printed_lines) == 0
|
|
|
|
# Verify final PV value is updated
|
|
assert pv.get() == value_new
|
|
|
|
|
|
def test_use_callback_context_manager():
|
|
pv = PV("TEST:VAL")
|
|
|
|
pv.put(0.0)
|
|
seen_values = []
|
|
|
|
def callback(value=None, **kwargs):
|
|
seen_values.append(value)
|
|
|
|
# Use context manager, callback should be triggered
|
|
with pv.use_callback(callback):
|
|
pv.put(42.0)
|
|
assert 42.0 in seen_values
|
|
|
|
with pv.use_callback(callback):
|
|
pv.put(24.0)
|
|
assert 24.0 in seen_values
|
|
|
|
with pv.use_callback(callback):
|
|
pv.put(75.0)
|
|
assert 75.0 in seen_values
|
|
|
|
|
|
def test_orig_repr_is_not_custom_repr():
|
|
pv = PV("TEST:VAL", connection_timeout=2.0)
|
|
custom = repr(pv)
|
|
original = pv.orig_repr()
|
|
|
|
# Custom repr should not be the same as original repr
|
|
assert custom != original
|
|
|
|
# Original repr should look like a standard epics PV repr
|
|
assert original.startswith("<PV ")
|
|
assert "TEST:VAL" in original
|
|
|
|
# Custom repr should include PV name
|
|
assert f'PV "TEST:VAL" at' in custom
|
|
'''
|