Commit faa3e9d2067913904f8d3d6a5163370a3257de54
1 parent
ac15cbd2
Exists in
master
and in
4 other branches
Testing import_widgets template tag
Signed-off-by: Lucas Moura <lucas.moura128@gmail.com> Signed-off-by: Luiz Oliveira <ziuloliveira@gmail.com> Signed-off-by: Macartur Sousa <macartur.sc@gmail.com> Signed-off-by: Matheus Faria <matheus.sousa.faria@gmail.com>
Showing
1 changed file
with
29 additions
and
0 deletions
Show diff stats
| ... | ... | @@ -0,0 +1,29 @@ |
| 1 | +import unittest | |
| 2 | +from mock import patch | |
| 3 | + | |
| 4 | +from colab.widgets.templatetags.widgets_tag import import_widgets | |
| 5 | +from colab.widgets.widget_manager import WidgetManager | |
| 6 | + | |
| 7 | + | |
| 8 | +class WidgetsTest(unittest.TestCase): | |
| 9 | + @patch.object(WidgetManager, 'get_widgets') | |
| 10 | + def test_import_widgets_tag(self, get_widgets): | |
| 11 | + return_list = [1, 2, 3] | |
| 12 | + get_widgets.return_value = return_list | |
| 13 | + | |
| 14 | + context = {'request': ""} | |
| 15 | + import_widgets(context, 'area') | |
| 16 | + | |
| 17 | + self.assertIn('widgets_area', context) | |
| 18 | + self.assertEquals(context['widgets_area'], return_list) | |
| 19 | + | |
| 20 | + @patch.object(WidgetManager, 'get_widgets') | |
| 21 | + def test_import_widgets_tag_with_named_var(self, get_widgets): | |
| 22 | + return_list = [1, 2, 3] | |
| 23 | + get_widgets.return_value = return_list | |
| 24 | + | |
| 25 | + context = {'request': ""} | |
| 26 | + import_widgets(context, 'area', 'var') | |
| 27 | + | |
| 28 | + self.assertIn('var', context) | |
| 29 | + self.assertEquals(context['var'], return_list) | ... | ... |