ic-query 0.8.1

//! Module: sns::report::view::tests
//!
//! Responsibility: cover SNS report view filtering and sorting behavior.
//! Does not own: report assembly, cache IO, live source conversion, or text rendering.
//! Boundary: exercises in-memory row transformations through the view module.

use super::{
    proposal_matches_before, proposal_matches_eligibility, proposal_matches_proposer,
    proposal_matches_query, proposal_matches_status, proposal_matches_topic,
    sort_mainnet_sns_instances, sort_sns_neurons, sort_sns_proposal_rows,
};
use crate::sns::report::{
    SNS_PROPOSAL_DECISION_DECIDED, SNS_PROPOSAL_DECISION_EXECUTED, SNS_PROPOSAL_DECISION_FAILED,
    SNS_PROPOSAL_DECISION_OPEN, SNS_PROPOSAL_STATUS_ADOPTED_CODE,
    SNS_PROPOSAL_STATUS_REJECTED_CODE, SnsListSort, SnsNeuronRow, SnsNeuronsSort,
    SnsProposalEligibilityFilter, SnsProposalRow, SnsProposalSortDirection,
    SnsProposalStatusFilter, SnsProposalTally, SnsProposalTopicFilter, SnsProposalsSort,
    source::MainnetSns,
};

#[test]
fn proposal_id_sort_orders_newest_id_first() {
    let mut proposals = vec![
        proposal_row(2, 100),
        proposal_row(10, 50),
        proposal_row(1, 200),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::Id,
        SnsProposalSortDirection::Desc,
    );

    assert_eq!(proposal_ids(&proposals), vec![10, 2, 1]);
}

#[test]
fn proposal_decided_sort_orders_newest_decision_first_and_open_last() {
    let mut proposals = vec![
        proposal_row_with_decision(2, Some(100)),
        proposal_row_with_decision(10, None),
        proposal_row_with_decision(1, Some(200)),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::Decided,
        SnsProposalSortDirection::Desc,
    );

    assert_eq!(proposal_ids(&proposals), vec![1, 2, 10]);
}

#[test]
fn proposal_status_sort_orders_lifecycle_states_first() {
    let mut proposals = vec![
        proposal_with_decision_state_and_id(2, SNS_PROPOSAL_DECISION_EXECUTED),
        proposal_with_decision_state_and_id(10, SNS_PROPOSAL_DECISION_OPEN),
        proposal_with_decision_state_and_id(1, SNS_PROPOSAL_DECISION_FAILED),
        proposal_with_decision_state_and_id(9, SNS_PROPOSAL_DECISION_DECIDED),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::Status,
        SnsProposalSortDirection::Asc,
    );

    assert_eq!(proposal_ids(&proposals), vec![10, 9, 2, 1]);
}

#[test]
fn proposal_proposer_sort_orders_present_ids_before_missing_ids() {
    let mut proposals = vec![
        proposal_row_with_proposer(2, Some("ffff")),
        proposal_row_with_proposer(10, None),
        proposal_row_with_proposer(1, Some("aaaa")),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::Proposer,
        SnsProposalSortDirection::Asc,
    );

    assert_eq!(proposal_ids(&proposals), vec![1, 2, 10]);
}

#[test]
fn proposal_topic_sort_orders_present_labels_before_missing_labels() {
    let mut proposals = vec![
        proposal_row_with_topic(2, Some("governance")),
        proposal_row_with_topic(10, None),
        proposal_row_with_topic(1, Some("application-business-logic")),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::Topic,
        SnsProposalSortDirection::Asc,
    );

    assert_eq!(proposal_ids(&proposals), vec![1, 2, 10]);
}

#[test]
fn proposal_decided_ascending_sort_orders_oldest_decision_first_and_open_last() {
    let mut proposals = vec![
        proposal_row_with_decision(2, Some(100)),
        proposal_row_with_decision(10, None),
        proposal_row_with_decision(1, Some(200)),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::Decided,
        SnsProposalSortDirection::Asc,
    );

    assert_eq!(proposal_ids(&proposals), vec![2, 1, 10]);
}

#[test]
fn proposal_executed_sort_orders_newest_execution_first_and_unexecuted_last() {
    let mut proposals = vec![
        proposal_row_with_execution(2, Some(100)),
        proposal_row_with_execution(10, None),
        proposal_row_with_execution(1, Some(200)),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::Executed,
        SnsProposalSortDirection::Desc,
    );

    assert_eq!(proposal_ids(&proposals), vec![1, 2, 10]);
}

#[test]
fn proposal_failed_sort_orders_newest_failure_first_and_non_failed_last() {
    let mut proposals = vec![
        proposal_row_with_failure(2, Some(100)),
        proposal_row_with_failure(10, None),
        proposal_row_with_failure(1, Some(200)),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::Failed,
        SnsProposalSortDirection::Desc,
    );

    assert_eq!(proposal_ids(&proposals), vec![1, 2, 10]);
}

#[test]
fn proposal_title_sort_orders_case_insensitive_with_id_tiebreaker() {
    let mut proposals = vec![
        proposal_row_with_title(2, "Zoo"),
        proposal_row_with_title(10, "alpha"),
        proposal_row_with_title(1, "Alpha"),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::Title,
        SnsProposalSortDirection::Asc,
    );

    assert_eq!(proposal_ids(&proposals), vec![1, 10, 2]);
}

#[test]
fn proposal_action_sort_orders_descending_with_id_tiebreaker() {
    let mut proposals = vec![
        proposal_row_with_action(2, "motion"),
        proposal_row_with_action(10, "upgrade-sns-controlled-canister"),
        proposal_row_with_action(1, "motion"),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::Action,
        SnsProposalSortDirection::Desc,
    );

    assert_eq!(proposal_ids(&proposals), vec![10, 2, 1]);
}

#[test]
fn proposal_total_votes_sort_orders_highest_tally_first() {
    let mut proposals = vec![
        proposal_row_with_tally(2, Some((100, 10, 20, 30))),
        proposal_row_with_tally(10, None),
        proposal_row_with_tally(1, Some((200, 50, 60, 110))),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::TotalVotes,
        SnsProposalSortDirection::Desc,
    );

    assert_eq!(proposal_ids(&proposals), vec![1, 2, 10]);
}

#[test]
fn proposal_tally_time_sort_orders_newest_tally_first() {
    let mut proposals = vec![
        proposal_row_with_tally(2, Some((100, 10, 20, 30))),
        proposal_row_with_tally(10, None),
        proposal_row_with_tally(1, Some((200, 50, 60, 110))),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::TallyTime,
        SnsProposalSortDirection::Desc,
    );

    assert_eq!(proposal_ids(&proposals), vec![1, 2, 10]);
}

#[test]
fn proposal_yes_sort_orders_ascending_with_id_tiebreaker() {
    let mut proposals = vec![
        proposal_row_with_tally(2, Some((100, 10, 20, 30))),
        proposal_row_with_tally(10, Some((200, 10, 30, 40))),
        proposal_row_with_tally(1, Some((300, 50, 60, 110))),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::Yes,
        SnsProposalSortDirection::Asc,
    );

    assert_eq!(proposal_ids(&proposals), vec![2, 10, 1]);
}

#[test]
fn proposal_ballots_sort_orders_highest_count_first() {
    let mut proposals = vec![
        proposal_row_with_ballot_count(2, 1),
        proposal_row_with_ballot_count(10, 5),
        proposal_row_with_ballot_count(1, 5),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::Ballots,
        SnsProposalSortDirection::Desc,
    );

    assert_eq!(proposal_ids(&proposals), vec![10, 1, 2]);
}

#[test]
fn proposal_reject_cost_sort_orders_lowest_cost_first_when_ascending() {
    let mut proposals = vec![
        proposal_row_with_reject_cost(2, 300),
        proposal_row_with_reject_cost(10, 100),
        proposal_row_with_reject_cost(1, 200),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::RejectCost,
        SnsProposalSortDirection::Asc,
    );

    assert_eq!(proposal_ids(&proposals), vec![10, 1, 2]);
}

#[test]
fn proposal_reward_round_sort_orders_highest_round_first() {
    let mut proposals = vec![
        proposal_row_with_reward_round(2, 1),
        proposal_row_with_reward_round(10, 5),
        proposal_row_with_reward_round(1, 5),
    ];

    sort_sns_proposal_rows(
        &mut proposals,
        SnsProposalsSort::RewardRound,
        SnsProposalSortDirection::Desc,
    );

    assert_eq!(proposal_ids(&proposals), vec![10, 1, 2]);
}

#[test]
fn sns_name_sort_preserves_stable_id_tiebreaker() {
    let mut instances = vec![
        mainnet_sns(3, "beta"),
        mainnet_sns(1, "Alpha"),
        mainnet_sns(2, "alpha"),
    ];

    sort_mainnet_sns_instances(&mut instances, SnsListSort::Name);

    assert_eq!(sns_instance_ids(&instances), vec![1, 2, 3]);
}

#[test]
fn neuron_stake_sort_orders_highest_stake_first_with_id_tiebreaker() {
    let mut neurons = vec![
        neuron_row("bb", 10, 1, 1),
        neuron_row("aa", 10, 2, 2),
        neuron_row("cc", 20, 1, 3),
    ];

    sort_sns_neurons(&mut neurons, SnsNeuronsSort::Stake);

    assert_eq!(neuron_ids(&neurons), vec!["cc", "aa", "bb"]);
}

#[test]
fn proposal_before_filter_requires_a_lower_id() {
    assert!(proposal_matches_before(&proposal_row(9, 100), Some(10)));
    assert!(!proposal_matches_before(&proposal_row(10, 100), Some(10)));
    assert!(proposal_matches_before(&proposal_row(10, 100), None));
}

#[test]
fn proposal_status_filter_matches_cache_backed_statuses() {
    assert!(proposal_matches_status(
        &proposal_with_decision_state("executed"),
        SnsProposalStatusFilter::Executed
    ));
    assert!(proposal_matches_status(
        &proposal_with_decision_state("decided"),
        SnsProposalStatusFilter::Decided
    ));
    assert!(!proposal_matches_status(
        &proposal_with_decision_state("open"),
        SnsProposalStatusFilter::Failed
    ));
    assert!(proposal_matches_status(
        &proposal_with_status(Some(SNS_PROPOSAL_STATUS_ADOPTED_CODE)),
        SnsProposalStatusFilter::Adopted
    ));
    assert!(proposal_matches_status(
        &proposal_with_status(Some(SNS_PROPOSAL_STATUS_REJECTED_CODE)),
        SnsProposalStatusFilter::Rejected
    ));
    assert!(!proposal_matches_status(
        &proposal_with_status(None),
        SnsProposalStatusFilter::Adopted
    ));
}

#[test]
fn proposal_topic_filter_matches_cached_topic_labels() {
    assert!(proposal_matches_topic(
        &proposal_with_topic(Some("governance")),
        SnsProposalTopicFilter::Governance
    ));
    assert!(!proposal_matches_topic(
        &proposal_with_topic(Some("treasury-asset-management")),
        SnsProposalTopicFilter::Governance
    ));
    assert!(!proposal_matches_topic(
        &proposal_with_topic(None),
        SnsProposalTopicFilter::Governance
    ));
    assert!(proposal_matches_topic(
        &proposal_with_topic(None),
        SnsProposalTopicFilter::Any
    ));
}

#[test]
fn proposal_eligibility_filter_matches_reward_eligibility() {
    assert!(proposal_matches_eligibility(
        &proposal_with_eligibility(true),
        SnsProposalEligibilityFilter::Yes
    ));
    assert!(proposal_matches_eligibility(
        &proposal_with_eligibility(false),
        SnsProposalEligibilityFilter::No
    ));
    assert!(!proposal_matches_eligibility(
        &proposal_with_eligibility(false),
        SnsProposalEligibilityFilter::Yes
    ));
    assert!(proposal_matches_eligibility(
        &proposal_with_eligibility(false),
        SnsProposalEligibilityFilter::Any
    ));
}

#[test]
fn proposal_proposer_filter_matches_neuron_id_prefix() {
    assert!(proposal_matches_proposer(
        &proposal_with_proposer(Some("0001020304")),
        Some("000102")
    ));
    assert!(!proposal_matches_proposer(
        &proposal_with_proposer(Some("0001020304")),
        Some("ffff")
    ));
    assert!(!proposal_matches_proposer(
        &proposal_with_proposer(None),
        Some("000102")
    ));
    assert!(proposal_matches_proposer(
        &proposal_with_proposer(None),
        None
    ));
}

#[test]
fn proposal_query_filter_matches_searchable_text_fields() {
    let proposal = proposal_with_query_text();

    assert!(proposal_matches_query(&proposal, Some("treasury")));
    assert!(proposal_matches_query(&proposal, Some("UPGRADE")));
    assert!(proposal_matches_query(&proposal, Some("committee")));
    assert!(proposal_matches_query(&proposal, Some("forum")));
    assert!(proposal_matches_query(&proposal, Some("payload")));
    assert!(!proposal_matches_query(&proposal, Some("subnet")));
    assert!(proposal_matches_query(&proposal, None));
}

fn proposal_ids(proposals: &[SnsProposalRow]) -> Vec<u64> {
    proposals
        .iter()
        .map(|proposal| proposal.proposal_id)
        .collect()
}

fn sns_instance_ids(instances: &[MainnetSns]) -> Vec<usize> {
    instances.iter().map(|sns| sns.id).collect()
}

fn mainnet_sns(id: usize, name: &str) -> MainnetSns {
    MainnetSns {
        id,
        name: name.to_string(),
        description: None,
        url: None,
        root_canister_id: format!("{id}-root"),
        governance_canister_id: format!("{id}-governance"),
        ledger_canister_id: format!("{id}-ledger"),
        swap_canister_id: format!("{id}-swap"),
        index_canister_id: format!("{id}-index"),
        metadata_error: None,
    }
}

fn neuron_ids(neurons: &[SnsNeuronRow]) -> Vec<&str> {
    neurons
        .iter()
        .map(|neuron| neuron.neuron_id.as_str())
        .collect()
}

fn neuron_row(
    neuron_id: &str,
    stake_e8s: u64,
    maturity_e8s: u64,
    created_timestamp_seconds: u64,
) -> SnsNeuronRow {
    SnsNeuronRow {
        neuron_id: neuron_id.to_string(),
        cached_neuron_stake_e8s: stake_e8s,
        maturity_e8s_equivalent: maturity_e8s,
        staked_maturity_e8s_equivalent: None,
        created_timestamp_seconds,
        created_at: created_timestamp_seconds.to_string(),
    }
}

fn proposal_with_decision_state(decision_state: &str) -> SnsProposalRow {
    SnsProposalRow {
        decision_state: decision_state.to_string(),
        ..proposal_row(1, 100)
    }
}

fn proposal_with_status(status: Option<i32>) -> SnsProposalRow {
    SnsProposalRow {
        status,
        ..proposal_row(1, 100)
    }
}

fn proposal_with_topic(topic: Option<&str>) -> SnsProposalRow {
    SnsProposalRow {
        topic: topic.map(ToString::to_string),
        ..proposal_row(1, 100)
    }
}

fn proposal_with_eligibility(is_eligible_for_rewards: bool) -> SnsProposalRow {
    SnsProposalRow {
        is_eligible_for_rewards,
        ..proposal_row(1, 100)
    }
}

fn proposal_with_proposer(proposer_neuron_id: Option<&str>) -> SnsProposalRow {
    SnsProposalRow {
        proposer_neuron_id: proposer_neuron_id.map(ToString::to_string),
        ..proposal_row(1, 100)
    }
}

fn proposal_with_query_text() -> SnsProposalRow {
    SnsProposalRow {
        title: "Treasury motion".to_string(),
        action: "upgrade-sns-controlled-canister".to_string(),
        summary: "Committee review".to_string(),
        url: Some("https://forum.example/proposal".to_string()),
        payload_text_rendering: Some("Payload preview".to_string()),
        ..proposal_row(1, 100)
    }
}

fn proposal_row_with_topic(proposal_id: u64, topic: Option<&str>) -> SnsProposalRow {
    SnsProposalRow {
        topic: topic.map(ToString::to_string),
        ..proposal_row(proposal_id, 100)
    }
}

fn proposal_with_decision_state_and_id(proposal_id: u64, decision_state: &str) -> SnsProposalRow {
    SnsProposalRow {
        proposal_id,
        decision_state: decision_state.to_string(),
        ..proposal_row(proposal_id, 100)
    }
}

fn proposal_row_with_title(proposal_id: u64, title: &str) -> SnsProposalRow {
    SnsProposalRow {
        title: title.to_string(),
        ..proposal_row(proposal_id, 100)
    }
}

fn proposal_row_with_action(proposal_id: u64, action: &str) -> SnsProposalRow {
    SnsProposalRow {
        action: action.to_string(),
        ..proposal_row(proposal_id, 100)
    }
}

fn proposal_row_with_proposer(
    proposal_id: u64,
    proposer_neuron_id: Option<&str>,
) -> SnsProposalRow {
    SnsProposalRow {
        proposer_neuron_id: proposer_neuron_id.map(ToString::to_string),
        ..proposal_row(proposal_id, 100)
    }
}

fn proposal_row_with_tally(
    proposal_id: u64,
    tally: Option<(u64, u64, u64, u64)>,
) -> SnsProposalRow {
    SnsProposalRow {
        latest_tally: tally.map(|(timestamp_seconds, yes, no, total)| SnsProposalTally {
            timestamp_seconds,
            yes,
            no,
            total,
        }),
        ..proposal_row(proposal_id, 100)
    }
}

fn proposal_row_with_ballot_count(proposal_id: u64, ballot_count: usize) -> SnsProposalRow {
    SnsProposalRow {
        ballot_count,
        ..proposal_row(proposal_id, 100)
    }
}

fn proposal_row_with_reject_cost(proposal_id: u64, reject_cost_e8s: u64) -> SnsProposalRow {
    SnsProposalRow {
        reject_cost_e8s,
        ..proposal_row(proposal_id, 100)
    }
}

fn proposal_row_with_reward_round(proposal_id: u64, reward_event_round: u64) -> SnsProposalRow {
    SnsProposalRow {
        reward_event_round,
        ..proposal_row(proposal_id, 100)
    }
}

fn proposal_row_with_decision(
    proposal_id: u64,
    decided_timestamp_seconds: Option<u64>,
) -> SnsProposalRow {
    SnsProposalRow {
        proposal_id,
        decided_timestamp_seconds,
        decided_at: decided_timestamp_seconds.map(|value| value.to_string()),
        ..proposal_row(proposal_id, 100)
    }
}

fn proposal_row_with_execution(
    proposal_id: u64,
    executed_timestamp_seconds: Option<u64>,
) -> SnsProposalRow {
    SnsProposalRow {
        proposal_id,
        executed_timestamp_seconds,
        executed_at: executed_timestamp_seconds.map(|value| value.to_string()),
        ..proposal_row(proposal_id, 100)
    }
}

fn proposal_row_with_failure(
    proposal_id: u64,
    failed_timestamp_seconds: Option<u64>,
) -> SnsProposalRow {
    SnsProposalRow {
        proposal_id,
        failed_timestamp_seconds,
        failed_at: failed_timestamp_seconds.map(|value| value.to_string()),
        ..proposal_row(proposal_id, 100)
    }
}

fn proposal_row(proposal_id: u64, created_at_secs: u64) -> SnsProposalRow {
    SnsProposalRow {
        proposal_id,
        action_id: 0,
        action: "motion".to_string(),
        title: String::new(),
        summary: String::new(),
        url: None,
        decision_state: SNS_PROPOSAL_DECISION_OPEN.to_string(),
        status: None,
        topic: None,
        reject_cost_e8s: 0,
        proposal_creation_timestamp_seconds: created_at_secs,
        created_at: created_at_secs.to_string(),
        decided_timestamp_seconds: None,
        decided_at: None,
        executed_timestamp_seconds: None,
        executed_at: None,
        failed_timestamp_seconds: None,
        failed_at: None,
        failure_reason: None,
        reward_event_round: 0,
        reward_event_end_timestamp_seconds: None,
        is_eligible_for_rewards: false,
        latest_tally: None,
        ballot_count: 0,
        ballots: Vec::new(),
        payload_text_rendering: None,
        proposer_neuron_id: None,
    }
}