Commit faa3e9d2067913904f8d3d6a5163370a3257de54

Authored by Macartur Sousa
1 parent ac15cbd2

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
colab/widgets/tests/test_widgets.py 0 → 100644
... ... @@ -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)
... ...