import pytest

from .models import Author, Work, Quote, QuoteTag

class Test_MassImport():
    @pytest.fixture(scope='function')
    def q1(self, db):
        a1 = Author.objects.create(name="JFK")
        w1 = Work.objects.create(name="Berlin speech", author=a1)
        q1 = Quote.objects.create(text="<p>Ich bin...</p>", work=w1)
        return q1

    @pytest.mark.django_db
    def test_massimport_1(self, q1, c_adm):
        allquotes = """\
Ich bin ein Berliner
---
JFK
---
Berlin speech
===
To be or not to be, that is the question
---
William Shakespeare
---
Hamlet
===
To thine own self, be true
---
William Shakespeare
---
Hamlet
---
tag1, tag2, tag3
===
A rose by any other name...
---
William Shakespeare
---
Romeo and Juliet
---
tag1, tag555
==="""

        results = c_adm.postPage('massimport/', {'quotes': allquotes})

        assert "rejected" not in results

        assert Quote.objects.get(text="<p>To thine own self, be true</p>")
        assert Quote.objects.get(text__contains="To be or not to be")

        assert Author.objects.get(name="JFK")
        assert Author.objects.get(name="William Shakespeare")
        assert Author.objects.all().count() == 2

        hamlet = Work.objects.get(name="Hamlet")
        assert hamlet
        assert hamlet.author == Author.objects.get(name="William Shakespeare")

        for quote in Quote.objects.all():
            assert c_adm.getPage(quote.get_absolute_url())

    @pytest.mark.django_db
    def test_massimport_2(self, q1, c_adm):
        allquotes = """\
<script>somethingevil()</script>
A rose by any other name...
---
William Shakespeare
---
Romeo and Juliet
---
tag1, tag555
==="""

        results = c_adm.postPage('massimport/', {'quotes': allquotes})
        assert "<script>" not in results
        with pytest.raises(Quote.DoesNotExist):
            Quote.objects.get(text__contains="<script>")

    @pytest.mark.django_db
    def test_massimport_3(self, c_adm):
        """Whitespace and stuff"""
        allquotes = """\
A rose by any other name...
---
William Shakespeare
---
Romeo and Juliet
---
tag1, tag555
===

To be, or not to be, that is the question:
Whether 'tis Nobler in the mind to suffer
The Slings and Arrows of outrageous Fortune,
Or to take Arms against a Sea of troubles,
And by opposing end them: to die, to sleep
No more;

---
   William Shakespeare  

---
Hamlet  

"""

        results = c_adm.postPage('massimport/', {'quotes': allquotes})
        assert "rejected" not in results
        assert Quote.objects.get(text="<p>A rose by any other name...</p>")
        assert Author.objects.get(name="William Shakespeare")
        assert Work.objects.get(name="Romeo and Juliet")
        assert QuoteTag.objects.get(tag="tag1")
        assert QuoteTag.objects.get(tag="tag555")

        tirade = Quote.objects.get(text__contains="To be, or not to be")
        assert tirade.work.name == "Hamlet"
        assert tirade.work.author.name == "William Shakespeare"
        assert Author.objects.filter(name__contains="Shakespeare").count() == 1

    def test_massimport_noaccess(self, c):
        c.getPage('massimport/', exp_status=302)
        c.postPage('massimport/', {}, exp_status=302)