mirror of
https://github.com/ChuckBuilds/LEDMatrix.git
synced 2026-04-10 13:02:59 +00:00
101 lines
3.6 KiB
Python
101 lines
3.6 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Test script for stock manager toggle_chart functionality.
|
|
This script tests that the toggle_chart setting properly adds/removes charts from the scrolling ticker.
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
import json
|
|
import time
|
|
|
|
# Add the src directory to the path so we can import our modules
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'src'))
|
|
|
|
from stock_manager import StockManager
|
|
from display_manager import DisplayManager
|
|
|
|
def test_toggle_chart_functionality():
|
|
"""Test that toggle_chart properly controls chart display in scrolling ticker."""
|
|
|
|
# Load test configuration
|
|
config = {
|
|
'stocks': {
|
|
'enabled': True,
|
|
'symbols': ['AAPL', 'MSFT', 'GOOGL'],
|
|
'scroll_speed': 1,
|
|
'scroll_delay': 0.01,
|
|
'toggle_chart': False # Start with charts disabled
|
|
},
|
|
'crypto': {
|
|
'enabled': False,
|
|
'symbols': []
|
|
}
|
|
}
|
|
|
|
# Create a mock display manager for testing
|
|
class MockDisplayManager:
|
|
def __init__(self):
|
|
self.matrix = type('Matrix', (), {'width': 64, 'height': 32})()
|
|
self.image = None
|
|
self.regular_font = type('Font', (), {'path': 'assets/fonts/5x7.bdf', 'size': 7})()
|
|
self.small_font = type('Font', (), {'path': 'assets/fonts/4x6.bdf', 'size': 6})()
|
|
|
|
def clear(self):
|
|
pass
|
|
|
|
def update_display(self):
|
|
pass
|
|
|
|
display_manager = MockDisplayManager()
|
|
|
|
# Create stock manager
|
|
stock_manager = StockManager(config, display_manager)
|
|
|
|
print("Testing Stock Manager toggle_chart functionality...")
|
|
print("=" * 50)
|
|
|
|
# Test 1: Verify initial state (charts disabled)
|
|
print(f"1. Initial toggle_chart setting: {stock_manager.toggle_chart}")
|
|
assert stock_manager.toggle_chart == False, "Initial toggle_chart should be False"
|
|
print("✓ Initial state correct")
|
|
|
|
# Test 2: Enable charts
|
|
print("\n2. Enabling charts...")
|
|
stock_manager.set_toggle_chart(True)
|
|
assert stock_manager.toggle_chart == True, "toggle_chart should be True after enabling"
|
|
print("✓ Charts enabled successfully")
|
|
|
|
# Test 3: Disable charts
|
|
print("\n3. Disabling charts...")
|
|
stock_manager.set_toggle_chart(False)
|
|
assert stock_manager.toggle_chart == False, "toggle_chart should be False after disabling"
|
|
print("✓ Charts disabled successfully")
|
|
|
|
# Test 4: Verify cache clearing
|
|
print("\n4. Testing cache clearing...")
|
|
stock_manager.cached_text_image = "test_cache"
|
|
stock_manager.set_toggle_chart(True)
|
|
assert stock_manager.cached_text_image is None, "Cache should be cleared when toggle_chart changes"
|
|
print("✓ Cache clearing works correctly")
|
|
|
|
# Test 5: Test configuration reload
|
|
print("\n5. Testing configuration reload...")
|
|
config['stocks']['toggle_chart'] = True
|
|
stock_manager.config = config
|
|
stock_manager.stocks_config = config['stocks']
|
|
stock_manager._reload_config()
|
|
assert stock_manager.toggle_chart == True, "toggle_chart should be updated from config"
|
|
print("✓ Configuration reload works correctly")
|
|
|
|
print("\n" + "=" * 50)
|
|
print("All tests passed! ✓")
|
|
print("\nSummary:")
|
|
print("- toggle_chart setting properly controls chart display in scrolling ticker")
|
|
print("- Charts are only shown when toggle_chart is True")
|
|
print("- Cache is properly cleared when setting changes")
|
|
print("- Configuration reload works correctly")
|
|
print("- No sleep delays are used in the scrolling ticker")
|
|
|
|
if __name__ == "__main__":
|
|
test_toggle_chart_functionality() |