test_widget_manager.py 325 Bytes
1 2 3 4 5 6 7 8 9 10 11 12 
from django.test import TestCase

from colab.widgets.widget_manager import WidgetManager, Widget

class WidgetManagerTest(TestCase):

    def test_add_widgets_to_key_area(self):
        area = 'profile'
        WidgetManager.register_widget(area, Widget())

        self.assertEqual(len(WidgetManager.get_widgets(area)), 1)