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