import pytest

from core.contracts import ContractRegistry


def test_contract_registry_basic():
    reg = ContractRegistry()
    reg.add_contract("LocalProblem", "v1", {"fields": ["asset_id", "payload"]})
    reg.add_contract("SharedVariables", "v1", {"fields": ["iter_id", "values"]})

    c1 = reg.get_contract("LocalProblem", "v1")
    c2 = reg.get_contract("SharedVariables", "v1")
    assert c1 is not None
    assert c2 is not None
    assert c1.name == "LocalProblem" and c1.version == "v1"
    assert c2.name == "SharedVariables" and c2.version == "v1"