AKharytonchyk
8f74f4b786
- Add comprehensive unit tests for TelegramListener and SaveToTelegram nodes - Add project structure validation tests - Add test runner script with coverage support - Add pytest configuration for better test organization - Add GitHub Actions workflow for CI/CD - Add development dependencies and requirements-dev.txt - Add Makefile for common development tasks - Add code formatting and linting configuration - Fix import issues in __init__.py for testing compatibility - Add missing .pyc pattern to .gitignore Test coverage includes: - Node class structure validation - Input/output type checking - Error handling and validation - Integration between listener and sender nodes - Project file structure compliance - ComfyUI Manager compatibility
85 lines
3.1 KiB
Python
85 lines
3.1 KiB
Python
import unittest
|
|
import sys
|
|
import os
|
|
|
|
# Add the parent directory to the path so we can import our modules
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
# Mock telegram imports before importing our module
|
|
from unittest.mock import Mock
|
|
sys.modules['telegram'] = Mock()
|
|
sys.modules['telegram.ext'] = Mock()
|
|
|
|
# Test the __init__.py module
|
|
class TestInit(unittest.TestCase):
|
|
"""Test cases for __init__.py module"""
|
|
|
|
def test_node_mappings_exist(self):
|
|
"""Test that NODE_CLASS_MAPPINGS and NODE_DISPLAY_NAME_MAPPINGS exist"""
|
|
import __init__ as init_module
|
|
|
|
# Check that the mappings exist
|
|
self.assertTrue(hasattr(init_module, 'NODE_CLASS_MAPPINGS'))
|
|
self.assertTrue(hasattr(init_module, 'NODE_DISPLAY_NAME_MAPPINGS'))
|
|
self.assertTrue(hasattr(init_module, 'WEB_DIRECTORY'))
|
|
|
|
def test_node_class_mappings_structure(self):
|
|
"""Test NODE_CLASS_MAPPINGS structure"""
|
|
import __init__ as init_module
|
|
|
|
mappings = init_module.NODE_CLASS_MAPPINGS
|
|
|
|
# Check that required nodes exist
|
|
self.assertIn('TelegramListener', mappings)
|
|
self.assertIn('SaveToTelegram', mappings)
|
|
|
|
# Check that they are callable (classes)
|
|
self.assertTrue(callable(mappings['TelegramListener']))
|
|
self.assertTrue(callable(mappings['SaveToTelegram']))
|
|
|
|
def test_node_display_name_mappings_structure(self):
|
|
"""Test NODE_DISPLAY_NAME_MAPPINGS structure"""
|
|
import __init__ as init_module
|
|
|
|
display_mappings = init_module.NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
# Check that display names exist for all nodes
|
|
self.assertIn('TelegramListener', display_mappings)
|
|
self.assertIn('SaveToTelegram', display_mappings)
|
|
|
|
# Check that display names are strings
|
|
self.assertIsInstance(display_mappings['TelegramListener'], str)
|
|
self.assertIsInstance(display_mappings['SaveToTelegram'], str)
|
|
|
|
# Check specific display names
|
|
self.assertEqual(display_mappings['TelegramListener'], 'Telegram Listener')
|
|
self.assertEqual(display_mappings['SaveToTelegram'], 'Save to Telegram')
|
|
|
|
def test_web_directory_setting(self):
|
|
"""Test WEB_DIRECTORY setting"""
|
|
import __init__ as init_module
|
|
|
|
self.assertEqual(init_module.WEB_DIRECTORY, "./web")
|
|
|
|
def test_version_exists(self):
|
|
"""Test that version is defined"""
|
|
import __init__ as init_module
|
|
|
|
self.assertTrue(hasattr(init_module, '__version__'))
|
|
self.assertIsInstance(init_module.__version__, str)
|
|
self.assertEqual(init_module.__version__, "1.0.0")
|
|
|
|
def test_all_exports(self):
|
|
"""Test __all__ exports"""
|
|
import __init__ as init_module
|
|
|
|
self.assertTrue(hasattr(init_module, '__all__'))
|
|
expected_exports = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|
|
|
|
for export in expected_exports:
|
|
self.assertIn(export, init_module.__all__)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(verbosity=2)
|