mirror of
https://github.com/Threnklyn/dungeon-sheets.git
synced 2026-05-19 04:33:26 +02:00
Mark Wolfman
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
from unittest import TestCase
|
|
|
|
|
|
from dungeonsheets.content_registry import ContentRegistry
|
|
from dungeonsheets import monsters
|
|
|
|
|
|
class TestContentRegistry(TestCase):
|
|
def test_add_module(self):
|
|
creg = ContentRegistry()
|
|
creg.add_module(monsters)
|
|
self.assertEqual(len(creg.modules), 1)
|
|
# Check if is indempotent
|
|
creg.add_module(monsters)
|
|
self.assertEqual(len(creg.modules), 1)
|
|
|
|
def test_findattr(self):
|
|
"""Check if the function can find attributes."""
|
|
|
|
class TestClass:
|
|
my_attr = 47
|
|
YourAttr = 53
|
|
|
|
test_module = TestClass()
|
|
creg = ContentRegistry()
|
|
creg.add_module(test_module)
|
|
# Direct access
|
|
self.assertEqual(creg.findattr("my_attr"), test_module.my_attr)
|
|
self.assertEqual(creg.findattr("YourAttr"), test_module.YourAttr)
|
|
# Swapping spaces for capitalization
|
|
self.assertEqual(creg.findattr("my attr"), test_module.my_attr)
|
|
self.assertEqual(creg.findattr("your attr"), test_module.YourAttr)
|
|
# Check for extra functuation
|
|
self.assertEqual(creg.findattr("my attr"), test_module.my_attr)
|
|
self.assertEqual(creg.findattr("Your/Attr"), test_module.YourAttr)
|
|
|
|
def test_findattr_valid_classes(self):
|
|
"""Check if the function can find attributes."""
|
|
|
|
class TestClass:
|
|
my_attr = 47
|
|
YourAttr = 53
|
|
|
|
class TestClassB:
|
|
my_attr = 48.0
|
|
|
|
test_module = TestClass()
|
|
creg = ContentRegistry()
|
|
creg.add_module(test_module)
|
|
creg.add_module(TestClassB)
|
|
# Direct access
|
|
self.assertEqual(creg.findattr("my_attr", valid_classes=[int]), test_module.my_attr)
|
|
|