from gridverse_open_low_code_platform_for_cro.registry import ContractRegistry


def test_registry_register_and_get():
    reg = ContractRegistry()
    contract = {
        "name": "LocalProblemContract",
        "version": "0.1.0",
        "schema": {"type": "object", "properties": {"id": {"type": "string"}}},
    }
    reg.register("local_problem", contract)
    assert reg.get("local_problem")["name"] == "LocalProblemContract"