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