diff --git a/colab/plugins/utils/apps.py b/colab/plugins/utils/apps.py index c3fd56a..57d5441 100644 --- a/colab/plugins/utils/apps.py +++ b/colab/plugins/utils/apps.py @@ -10,7 +10,7 @@ class ColabProxiedAppConfig(AppConfig): def __init__(self, app_name, app_module): super(ColabProxiedAppConfig, self).__init__(app_name, app_module) - self.__import_signals(app_name) + self._import_signals(app_name) self.signals.register_signal() diff --git a/colab/signals/tests/test_signals.py b/colab/signals/tests/test_signals.py index 7e65f27..1fcf73f 100644 --- a/colab/signals/tests/test_signals.py +++ b/colab/signals/tests/test_signals.py @@ -6,8 +6,8 @@ Objective: Test parameters, and behavior. from django.test import TestCase from django.apps import apps import django -from colab.signals.tasks import * -from mock import patch +from colab.signals.signals import * +from mock import patch, MagicMock, PropertyMock class SignalsTest(TestCase): @@ -51,12 +51,13 @@ class SignalsTest(TestCase): @patch('colab.signals.signals.Signal.connect') def test_connect_already_registered_signal(self, mock): sender = 'Test' - handling_method = 'Test' + handling_method = MagicMock + type(handling_method).delay = PropertyMock(return_value='Test') signal_name = 'a' register_signal(self.plugin_name, self.list_signal) - connect_signal(signal_name, sender, handling_method) + connect_signal(signal_name, sender, handling_method.delay) args, kwargs = mock.call_args self.assertEqual(args[0], handling_method) -- libgit2 0.21.2