# -*- coding: utf-8 -*-

from . import *
from . fixtures import *

from talon import quotations


@patch.object(quotations, 'extract_from_html')
@patch.object(quotations, 'extract_from_plain')
def test_extract_from_respects_content_type(extract_from_plain,
                                            extract_from_html):
    msg_body = "Hi there"

    quotations.extract_from(msg_body, 'text/plain')
    extract_from_plain.assert_called_with(msg_body)

    quotations.extract_from(msg_body, 'text/html')
    extract_from_html.assert_called_with(msg_body)

    eq_(msg_body, quotations.extract_from(msg_body, 'text/blah'))


@patch.object(quotations, 'extract_from_plain', Mock(side_effect=Exception()))
def test_crash_inside_extract_from():
    msg_body = "Hi there"
    eq_(msg_body, quotations.extract_from(msg_body, 'text/plain'))


def test_empty_body():
    eq_('', quotations.extract_from_plain(''))


def test__CRLF_to_LF():
    eq_(('\n\r', True), quotations._CRLF_to_LF('\r\n\r'))
    eq_(('\n\r', False), quotations._CRLF_to_LF('\n\r'))


def test__restore_CRLF():
    eq_('\n', quotations._restore_CRLF('\n', replaced=False))
    eq_('\r\n', quotations._restore_CRLF('\n', replaced=True))    
    # default
    eq_('\r\n', quotations._restore_CRLF('\n'))