Skip to content

Commit

Permalink
Adding tests for widget render content
Browse files Browse the repository at this point in the history
Signed-off-by: Macartur Sousa <macartur.sc@gmail.com>
  • Loading branch information
macartur committed Nov 16, 2015
1 parent 9787e09 commit da12803
Showing 1 changed file with 11 additions and 5 deletions.
16 changes: 11 additions & 5 deletions colab/widgets/tests/test_widgets.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,27 +2,33 @@
from mock import patch

from colab.widgets.templatetags.widgets_tag import import_widgets
from colab.widgets.widget_manager import WidgetManager
from colab.widgets.widget_manager import WidgetManager, Widget
from django.template import Context


class WigetMock(Widget):
def __init__(self, content=""):
self.content = content


class WidgetsTest(unittest.TestCase):
@patch.object(WidgetManager, 'get_widgets')
def test_import_widgets_tag(self, get_widgets):
return_list = [1, 2, 3]
return_list = [WigetMock(), WigetMock(), WigetMock()]
get_widgets.return_value = return_list

context = {'request': ""}
context = Context({'request': ""})
import_widgets(context, 'area')

self.assertIn('widgets_area', context)
self.assertEquals(context['widgets_area'], return_list)

@patch.object(WidgetManager, 'get_widgets')
def test_import_widgets_tag_with_named_var(self, get_widgets):
return_list = [1, 2, 3]
return_list = [WigetMock(), WigetMock(), WigetMock()]
get_widgets.return_value = return_list

context = {'request': ""}
context = Context({'request': ""})
import_widgets(context, 'area', 'var')

self.assertIn('var', context)
Expand Down

0 comments on commit da12803

Please sign in to comment.