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