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 | 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) | ... | ... |