mod common;

mod get {

    use edc_connector_client::types::contract_negotiation::ContractNegotiationState;

    use crate::common::{
        seed_contract_negotiation, setup_consumer_client, setup_provider_client,
        wait_for_negotiation_state,
    };

    #[tokio::test]
    async fn should_get_a_contract_agreement() {
        let provider = setup_provider_client();
        let consumer = setup_consumer_client();

        let (contract_negotiation_id, _) = seed_contract_negotiation(&consumer, &provider).await;

        wait_for_negotiation_state(
            &consumer,
            &contract_negotiation_id,
            ContractNegotiationState::Finalized,
        )
        .await;

        let agreement_id = consumer
            .contract_negotiations()
            .get(&contract_negotiation_id)
            .await
            .map(|cn| cn.contract_agreement_id().cloned())
            .unwrap()
            .unwrap();

        let contract_agreement = consumer
            .contract_agreements()
            .get(&agreement_id)
            .await
            .unwrap();

        assert_eq!(agreement_id, contract_agreement.id());
    }
}

mod query {
    use edc_connector_client::types::{
        contract_negotiation::ContractNegotiationState, query::Query,
    };

    use crate::common::{
        seed_contract_negotiation, setup_consumer_client, setup_provider_client,
        wait_for_negotiation_state,
    };

    #[tokio::test]
    async fn should_query_contract_agreements() {
        let provider = setup_provider_client();
        let consumer = setup_consumer_client();

        let (contract_negotiation_id, asset_id) =
            seed_contract_negotiation(&consumer, &provider).await;

        wait_for_negotiation_state(
            &consumer,
            &contract_negotiation_id,
            ContractNegotiationState::Finalized,
        )
        .await;

        let agreements = consumer
            .contract_agreements()
            .query(Query::builder().filter("assetId", "=", asset_id).build())
            .await
            .unwrap();

        assert_eq!(1, agreements.len());
    }
}