from django.test import TestCase
from wagtail.core.models import Collection

from vbv_lernwelt.core.create_default_users import create_default_users
from vbv_lernwelt.core.tests.helpers import create_locales_for_wagtail
from vbv_lernwelt.competence.create_default_documents import create_default_collections, create_default_documents
from vbv_lernwelt.competence.models import LibraryDocument


class TestCreateDefaultDocuments(TestCase):
    def setUp(self) -> None:
        create_default_users()
        create_locales_for_wagtail()

    def test_create_default_collections(self):
        create_default_collections()
        qs = Collection.objects.filter(name="Versicherungsvermittler/in")
        self.assertTrue(qs.exists())

    def test_create_default_documents(self):
        create_default_collections()

        create_default_documents()
        qs = LibraryDocument.objects.all()
        self.assertTrue(qs.exists())