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 2 from mock import patch
3 3  
4 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 14 class WidgetsTest(unittest.TestCase):
9 15 @patch.object(WidgetManager, 'get_widgets')
10 16 def test_import_widgets_tag(self, get_widgets):
11   - return_list = [1, 2, 3]
  17 + return_list = [WigetMock(), WigetMock(), WigetMock()]
12 18 get_widgets.return_value = return_list
13 19  
14   - context = {'request': ""}
  20 + context = Context({'request': ""})
15 21 import_widgets(context, 'area')
16 22  
17 23 self.assertIn('widgets_area', context)
... ... @@ -19,10 +25,10 @@ class WidgetsTest(unittest.TestCase):
19 25  
20 26 @patch.object(WidgetManager, 'get_widgets')
21 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 29 get_widgets.return_value = return_list
24 30  
25   - context = {'request': ""}
  31 + context = Context({'request': ""})
26 32 import_widgets(context, 'area', 'var')
27 33  
28 34 self.assertIn('var', context)
... ...