mirror of
https://github.com/ChuckBuilds/LEDMatrix.git
synced 2026-04-10 13:02:59 +00:00
set web ui v2 to be default
This commit is contained in:
Chuck
125
test/test_soccer_logo_fix.py
Normal file
125
test/test_soccer_logo_fix.py
Normal file
@@ -0,0 +1,125 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Test script to verify the soccer logo permissions fix.
|
||||
This script tests the _load_and_resize_logo method to ensure it can create placeholder logos
|
||||
without permission errors.
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
import tempfile
|
||||
import shutil
|
||||
from PIL import Image, ImageDraw, ImageFont
|
||||
import random
|
||||
|
||||
# Add the src directory to the path so we can import the modules
|
||||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))
|
||||
|
||||
try:
|
||||
from cache_manager import CacheManager
|
||||
from soccer_managers import BaseSoccerManager
|
||||
from display_manager import DisplayManager
|
||||
except ImportError as e:
|
||||
print(f"Import error: {e}")
|
||||
print("Make sure you're running this from the LEDMatrix root directory")
|
||||
sys.exit(1)
|
||||
|
||||
def test_soccer_logo_creation():
|
||||
"""Test that soccer placeholder logos can be created without permission errors."""
|
||||
|
||||
print("Testing soccer logo creation...")
|
||||
|
||||
# Create a temporary directory for testing
|
||||
test_dir = tempfile.mkdtemp(prefix="ledmatrix_test_")
|
||||
print(f"Using test directory: {test_dir}")
|
||||
|
||||
try:
|
||||
# Create a minimal config
|
||||
config = {
|
||||
"soccer_scoreboard": {
|
||||
"enabled": True,
|
||||
"logo_dir": "assets/sports/soccer_logos",
|
||||
"update_interval_seconds": 60
|
||||
},
|
||||
"display": {
|
||||
"width": 64,
|
||||
"height": 32
|
||||
}
|
||||
}
|
||||
|
||||
# Create cache manager with test directory
|
||||
cache_manager = CacheManager()
|
||||
# Override cache directory for testing
|
||||
cache_manager.cache_dir = test_dir
|
||||
|
||||
# Create a mock display manager
|
||||
class MockDisplayManager:
|
||||
def __init__(self):
|
||||
self.width = 64
|
||||
self.height = 32
|
||||
self.image = Image.new('RGB', (64, 32), (0, 0, 0))
|
||||
|
||||
display_manager = MockDisplayManager()
|
||||
|
||||
# Create soccer manager
|
||||
soccer_manager = BaseSoccerManager(config, display_manager, cache_manager)
|
||||
|
||||
# Test teams that might not have logos
|
||||
test_teams = ["ATX", "STL", "SD", "CLT", "TEST1", "TEST2"]
|
||||
|
||||
print("\nTesting logo creation for missing teams:")
|
||||
for team in test_teams:
|
||||
print(f" Testing {team}...")
|
||||
try:
|
||||
logo = soccer_manager._load_and_resize_logo(team)
|
||||
if logo:
|
||||
print(f" ✓ Successfully created logo for {team} (size: {logo.size})")
|
||||
else:
|
||||
print(f" ✗ Failed to create logo for {team}")
|
||||
except Exception as e:
|
||||
print(f" ✗ Error creating logo for {team}: {e}")
|
||||
|
||||
# Check if placeholder logos were created in cache
|
||||
placeholder_dir = os.path.join(test_dir, 'placeholder_logos')
|
||||
if os.path.exists(placeholder_dir):
|
||||
placeholder_files = os.listdir(placeholder_dir)
|
||||
print(f"\nPlaceholder logos created in cache: {len(placeholder_files)} files")
|
||||
for file in placeholder_files:
|
||||
print(f" - {file}")
|
||||
else:
|
||||
print("\nNo placeholder logos directory created (using in-memory placeholders)")
|
||||
|
||||
print("\n✓ Soccer logo test completed successfully!")
|
||||
|
||||
except Exception as e:
|
||||
print(f"\n✗ Test failed with error: {e}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
return False
|
||||
|
||||
finally:
|
||||
# Clean up test directory
|
||||
try:
|
||||
shutil.rmtree(test_dir)
|
||||
print(f"Cleaned up test directory: {test_dir}")
|
||||
except Exception as e:
|
||||
print(f"Warning: Could not clean up test directory: {e}")
|
||||
|
||||
return True
|
||||
|
||||
if __name__ == "__main__":
|
||||
print("LEDMatrix Soccer Logo Permissions Fix Test")
|
||||
print("=" * 50)
|
||||
|
||||
success = test_soccer_logo_creation()
|
||||
|
||||
if success:
|
||||
print("\n🎉 All tests passed! The soccer logo fix is working correctly.")
|
||||
print("\nTo apply this fix on your Raspberry Pi:")
|
||||
print("1. Transfer the updated files to your Pi")
|
||||
print("2. Run: chmod +x fix_soccer_logo_permissions.sh")
|
||||
print("3. Run: ./fix_soccer_logo_permissions.sh")
|
||||
print("4. Restart your LEDMatrix application")
|
||||
else:
|
||||
print("\n❌ Tests failed. Please check the error messages above.")
|
||||
sys.exit(1)
|
||||
167
test/test_stock_news_fix.py
Normal file
167
test/test_stock_news_fix.py
Normal file
@@ -0,0 +1,167 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Test script to verify the stock news manager fix.
|
||||
This script tests that the display_news method works correctly without excessive image generation.
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
import time
|
||||
import tempfile
|
||||
import shutil
|
||||
from PIL import Image
|
||||
|
||||
# Add the src directory to the path so we can import the modules
|
||||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))
|
||||
|
||||
try:
|
||||
from cache_manager import CacheManager
|
||||
from stock_news_manager import StockNewsManager
|
||||
from display_manager import DisplayManager
|
||||
except ImportError as e:
|
||||
print(f"Import error: {e}")
|
||||
print("Make sure you're running this from the LEDMatrix root directory")
|
||||
sys.exit(1)
|
||||
|
||||
def test_stock_news_display():
|
||||
"""Test that stock news display works correctly without excessive image generation."""
|
||||
|
||||
print("Testing stock news display fix...")
|
||||
|
||||
# Create a temporary directory for testing
|
||||
test_dir = tempfile.mkdtemp(prefix="ledmatrix_test_")
|
||||
print(f"Using test directory: {test_dir}")
|
||||
|
||||
try:
|
||||
# Create a minimal config
|
||||
config = {
|
||||
"stock_news": {
|
||||
"enabled": True,
|
||||
"scroll_speed": 1,
|
||||
"scroll_delay": 0.1, # Slower for testing
|
||||
"headlines_per_rotation": 2,
|
||||
"max_headlines_per_symbol": 1,
|
||||
"update_interval": 300,
|
||||
"dynamic_duration": True,
|
||||
"min_duration": 30,
|
||||
"max_duration": 300
|
||||
},
|
||||
"stocks": {
|
||||
"symbols": ["AAPL", "GOOGL", "MSFT"],
|
||||
"enabled": True
|
||||
},
|
||||
"display": {
|
||||
"width": 64,
|
||||
"height": 32
|
||||
}
|
||||
}
|
||||
|
||||
# Create cache manager with test directory
|
||||
cache_manager = CacheManager()
|
||||
# Override cache directory for testing
|
||||
cache_manager.cache_dir = test_dir
|
||||
|
||||
# Create a mock display manager
|
||||
class MockDisplayManager:
|
||||
def __init__(self):
|
||||
self.width = 64
|
||||
self.height = 32
|
||||
self.image = Image.new('RGB', (64, 32), (0, 0, 0))
|
||||
self.matrix = type('Matrix', (), {'width': 64, 'height': 32})()
|
||||
self.small_font = None # We'll handle this in the test
|
||||
|
||||
def update_display(self):
|
||||
# Mock update - just pass
|
||||
pass
|
||||
|
||||
display_manager = MockDisplayManager()
|
||||
|
||||
# Create stock news manager
|
||||
news_manager = StockNewsManager(config, display_manager)
|
||||
|
||||
# Mock some news data
|
||||
news_manager.news_data = {
|
||||
"AAPL": [
|
||||
{"title": "Apple reports strong Q4 earnings", "publisher": "Reuters"},
|
||||
{"title": "New iPhone sales exceed expectations", "publisher": "Bloomberg"}
|
||||
],
|
||||
"GOOGL": [
|
||||
{"title": "Google announces new AI features", "publisher": "TechCrunch"},
|
||||
{"title": "Alphabet stock reaches new high", "publisher": "CNBC"}
|
||||
],
|
||||
"MSFT": [
|
||||
{"title": "Microsoft cloud services grow 25%", "publisher": "WSJ"},
|
||||
{"title": "Windows 12 preview released", "publisher": "The Verge"}
|
||||
]
|
||||
}
|
||||
|
||||
print("\nTesting display_news method...")
|
||||
|
||||
# Test multiple calls to ensure it doesn't generate images excessively
|
||||
generation_count = 0
|
||||
original_generate_method = news_manager._generate_background_image
|
||||
|
||||
def mock_generate_method(*args, **kwargs):
|
||||
nonlocal generation_count
|
||||
generation_count += 1
|
||||
print(f" Image generation call #{generation_count}")
|
||||
return original_generate_method(*args, **kwargs)
|
||||
|
||||
news_manager._generate_background_image = mock_generate_method
|
||||
|
||||
# Call display_news multiple times to simulate the display controller
|
||||
for i in range(10):
|
||||
print(f" Call {i+1}: ", end="")
|
||||
try:
|
||||
result = news_manager.display_news()
|
||||
if result:
|
||||
print("✓ Success")
|
||||
else:
|
||||
print("✗ Failed")
|
||||
except Exception as e:
|
||||
print(f"✗ Error: {e}")
|
||||
|
||||
print(f"\nTotal image generations: {generation_count}")
|
||||
|
||||
if generation_count <= 3: # Should only generate a few times for different rotations
|
||||
print("✓ Image generation is working correctly (not excessive)")
|
||||
else:
|
||||
print("✗ Too many image generations - fix may not be working")
|
||||
|
||||
print("\n✓ Stock news display test completed!")
|
||||
|
||||
except Exception as e:
|
||||
print(f"\n✗ Test failed with error: {e}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
return False
|
||||
|
||||
finally:
|
||||
# Clean up test directory
|
||||
try:
|
||||
shutil.rmtree(test_dir)
|
||||
print(f"Cleaned up test directory: {test_dir}")
|
||||
except Exception as e:
|
||||
print(f"Warning: Could not clean up test directory: {e}")
|
||||
|
||||
return True
|
||||
|
||||
if __name__ == "__main__":
|
||||
print("LEDMatrix Stock News Manager Fix Test")
|
||||
print("=" * 50)
|
||||
|
||||
success = test_stock_news_display()
|
||||
|
||||
if success:
|
||||
print("\n🎉 Test completed! The stock news manager should now work correctly.")
|
||||
print("\nThe fix addresses the issue where the display_news method was:")
|
||||
print("1. Generating images excessively (every second)")
|
||||
print("2. Missing the actual scrolling display logic")
|
||||
print("3. Causing rapid rotation through headlines")
|
||||
print("\nNow it should:")
|
||||
print("1. Generate images only when needed for new rotations")
|
||||
print("2. Properly scroll the content across the display")
|
||||
print("3. Use the configured dynamic duration properly")
|
||||
else:
|
||||
print("\n❌ Test failed. Please check the error messages above.")
|
||||
sys.exit(1)
|
||||
Reference in New Issue
Block a user