from django.test import TestCase
from django.utils import unittest
from widgy.site import WidgySite
from widgy.models import Node
from widgy.exceptions import ParentChildRejection
from widgy.contrib.page_builder.models import (Table, TableRow,
TableHeaderData, TableHeader,
TableBody, Accordion, Video,
MainContent)
from widgy.contrib.page_builder.forms import CKEditorField
widgy_site = WidgySite()
def refetch(c):
return Node.objects.get(pk=c.node.pk).content
class TestTableWidget(TestCase):
def setUp(self):
self.table = Table.add_root(widgy_site)
def test_add_column(self):
self.table.body.add_child(widgy_site, TableRow)
self.table.body.add_child(widgy_site, TableRow)
self.assertEqual(len(self.table.body.get_children()[0].get_children()),
0)
self.assertEqual(len(self.table.body.get_children()[1].get_children()),
0)
from widgy.site import WidgySite
from widgets.models import HomepageLayout, WhateverpageLayout
class WidgySite(WidgySite):
def valid_root_of(self, owner_class, root_class, root_choices):
from widgy.contrib.widgy_mezzanine.models import WidgyPage
if issubclass(owner_class, WidgyPage):
return issubclass(root_class, (
HomepageLayout,
WhateverpageLayout,
))
else:
return super(WidgySite,
self).valid_root_of(owner_class, root_class,
root_choices)
site = WidgySite()