Commit da12803043d107891dcc932d2ca143422e39fd3f
1 parent
9787e097
Exists in
master
and in
4 other branches
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) |