32 lines
990 B
Python
32 lines
990 B
Python
import pytest
|
|
from gnuradio_mcp.middlewares.platform import BlockModel, PlatformMiddleware
|
|
|
|
class MockBlock:
|
|
def __init__(self, label, key):
|
|
self.label = label
|
|
self.key = key
|
|
|
|
class MockPlatform:
|
|
def __init__(self, blocks):
|
|
self.blocks = {block.key: block for block in blocks}
|
|
|
|
def test_block_model_from_block():
|
|
block = MockBlock(label="Test Block", key="test_key")
|
|
model = BlockModel.from_block(block)
|
|
assert model.label == "Test Block"
|
|
assert model.key == "test_key"
|
|
|
|
def test_platform_middleware_blocks():
|
|
blocks = [
|
|
MockBlock(label="Block A", key="a"),
|
|
MockBlock(label="Block B", key="b"),
|
|
]
|
|
platform = MockPlatform(blocks)
|
|
middleware = PlatformMiddleware(platform)
|
|
block_models = middleware.blocks
|
|
assert len(block_models) == 2
|
|
assert block_models[0].label == "Block A"
|
|
assert block_models[0].key == "a"
|
|
assert block_models[1].label == "Block B"
|
|
assert block_models[1].key == "b"
|