"""Tests for Message Handlers."""
from src.logic.msg_handlers import OP_UPDATE
from src.logic.msg_handlers import get_operation_type
from src.logic.msg_handlers import prepare_message

CRM_ID = '2131231'


def test_get_operation_type_update(message_updated_full):
    """Test operation is correct in update messages."""
    operation = get_operation_type(message_updated_full)
    assert operation == OP_UPDATE


def test_get_operation_type_delete(message_deleted):
    """Test operation is correct in delete messages."""
    operation = get_operation_type(message_deleted)
    assert operation is None


def test_prepare_message_update(subscription_active_off):
    """Test a message is correctly prepared for updates."""
    msg = prepare_message(subscription_active_off, CRM_ID)
    assert msg.Action__c == 'Unsub'