"""Test for run controller contract schema."""

from royalties.schemas.run_controller_contract import (
    RunControllerContractPutSchema,
)


def test_run_controller_contract_put_schema():
    """Test run controller contract PUT schema."""
    data = {
        'contract_id': 1,
        'run_controller_id': 2,
    }
    result = RunControllerContractPutSchema().dump(data)

    assert result['contract_id'] == data['contract_id']
    assert result['run_controller_id'] == data['run_controller_id']