Commit da12803043d107891dcc932d2ca143422e39fd3f

Authored by Macartur Sousa
1 parent 9787e097

Adding tests for widget render content

Signed-off-by: Macartur Sousa <macartur.sc@gmail.com>
Showing 1 changed file with 11 additions and 5 deletions   Show diff stats
colab/widgets/tests/test_widgets.py
@@ -2,16 +2,22 @@ import unittest @@ -2,16 +2,22 @@ import unittest
2 from mock import patch 2 from mock import patch
3 3
4 from colab.widgets.templatetags.widgets_tag import import_widgets 4 from colab.widgets.templatetags.widgets_tag import import_widgets
5 -from colab.widgets.widget_manager import WidgetManager 5 +from colab.widgets.widget_manager import WidgetManager, Widget
  6 +from django.template import Context
  7 +
  8 +
  9 +class WigetMock(Widget):
  10 + def __init__(self, content=""):
  11 + self.content = content
6 12
7 13
8 class WidgetsTest(unittest.TestCase): 14 class WidgetsTest(unittest.TestCase):
9 @patch.object(WidgetManager, 'get_widgets') 15 @patch.object(WidgetManager, 'get_widgets')
10 def test_import_widgets_tag(self, get_widgets): 16 def test_import_widgets_tag(self, get_widgets):
11 - return_list = [1, 2, 3] 17 + return_list = [WigetMock(), WigetMock(), WigetMock()]
12 get_widgets.return_value = return_list 18 get_widgets.return_value = return_list
13 19
14 - context = {'request': ""} 20 + context = Context({'request': ""})
15 import_widgets(context, 'area') 21 import_widgets(context, 'area')
16 22
17 self.assertIn('widgets_area', context) 23 self.assertIn('widgets_area', context)
@@ -19,10 +25,10 @@ class WidgetsTest(unittest.TestCase): @@ -19,10 +25,10 @@ class WidgetsTest(unittest.TestCase):
19 25
20 @patch.object(WidgetManager, 'get_widgets') 26 @patch.object(WidgetManager, 'get_widgets')
21 def test_import_widgets_tag_with_named_var(self, get_widgets): 27 def test_import_widgets_tag_with_named_var(self, get_widgets):
22 - return_list = [1, 2, 3] 28 + return_list = [WigetMock(), WigetMock(), WigetMock()]
23 get_widgets.return_value = return_list 29 get_widgets.return_value = return_list
24 30
25 - context = {'request': ""} 31 + context = Context({'request': ""})
26 import_widgets(context, 'area', 'var') 32 import_widgets(context, 'area', 'var')
27 33
28 self.assertIn('var', context) 34 self.assertIn('var', context)