from django.template.defaultfilters import cut
from django.test import SimpleTestCase
from django.utils.safestring import mark_safe

from ..utils import setup


class CutTests(SimpleTestCase):

    @setup({'cut01': '{% autoescape off %}{{ a|cut:"x" }} {{ b|cut:"x" }}{% endautoescape %}'})
    def test_cut01(self):
        output = self.engine.render_to_string('cut01', {"a": "x&y", "b": mark_safe("x&y")})
        self.assertEqual(output, "&y &y")

    @setup({'cut02': '{{ a|cut:"x" }} {{ b|cut:"x" }}'})
    def test_cut02(self):
        output = self.engine.render_to_string('cut02', {"a": "x&y", "b": mark_safe("x&y")})
        self.assertEqual(output, "&y &y")

    @setup({'cut03': '{% autoescape off %}{{ a|cut:"&" }} {{ b|cut:"&" }}{% endautoescape %}'})
    def test_cut03(self):
        output = self.engine.render_to_string('cut03', {"a": "x&y", "b": mark_safe("x&y")})
        self.assertEqual(output, "xy xamp;y")

    @setup({'cut04': '{{ a|cut:"&" }} {{ b|cut:"&" }}'})
    def test_cut04(self):
        output = self.engine.render_to_string('cut04', {"a": "x&y", "b": mark_safe("x&y")})
        self.assertEqual(output, "xy xamp;y")

    # Passing ';' to cut can break existing HTML entities, so those strings
    # are auto-escaped.
    @setup({'cut05': '{% autoescape off %}{{ a|cut:";" }} {{ b|cut:";" }}{% endautoescape %}'})
    def test_cut05(self):
        output = self.engine.render_to_string('cut05', {"a": "x&y", "b": mark_safe("x&y")})
        self.assertEqual(output, "x&y x&ampy")

    @setup({'cut06': '{{ a|cut:";" }} {{ b|cut:";" }}'})
    def test_cut06(self):
        output = self.engine.render_to_string('cut06', {"a": "x&y", "b": mark_safe("x&y")})
        self.assertEqual(output, "x&y x&ampy")


class FunctionTests(SimpleTestCase):

    def test_character(self):
        self.assertEqual(cut('a string to be mangled', 'a'), ' string to be mngled')

    def test_characters(self):
        self.assertEqual(cut('a string to be mangled', 'ng'), 'a stri to be maled')

    def test_non_matching_string(self):
        self.assertEqual(cut('a string to be mangled', 'strings'), 'a string to be mangled')

    def test_non_string_input(self):
        self.assertEqual(cut(123, '2'), '13')