from chorus_bot import modality_map


def test_known_channels_map_to_modalities():
    assert modality_map.modality_for_channel("personal-text") == "personal_text"
    assert modality_map.modality_for_channel("voice") == "voice"
    assert modality_map.modality_for_channel("fb-messenger") == "fb_messenger"
    assert modality_map.modality_for_channel("nd-messenger") == "nd_messenger"


def test_unknown_channel_returns_none():
    assert modality_map.modality_for_channel("random-chat") is None


def test_inbox_and_general_channels_are_not_mapped():
    assert modality_map.modality_for_channel("inbox") is None
    assert modality_map.modality_for_channel("general") is None


def test_leads_found_channel_maps_to_lead_found_modality():
    assert modality_map.modality_for_channel("leads-found") == "lead_found"


def test_direction_for_modality_outbound_for_leads():
    assert modality_map.direction_for_modality("lead_found") == "outbound"
    assert modality_map.direction_for_modality("magpie_lead") == "outbound"


def test_direction_for_modality_inbound_for_everything_else():
    assert modality_map.direction_for_modality("bh_email") == "inbound"
    assert modality_map.direction_for_modality("voice") == "inbound"