|
@@ -0,0 +1,29 @@
|
|
|
+# -*- coding: utf-8 -*-
|
|
|
+"""Tests for ``celery report`` command."""
|
|
|
+from __future__ import absolute_import, unicode_literals
|
|
|
+
|
|
|
+from case import Mock, call, patch
|
|
|
+
|
|
|
+from celery.bin.celery import report
|
|
|
+from celery.five import WhateverIO
|
|
|
+
|
|
|
+
|
|
|
+class test_report:
|
|
|
+ """Test report command class."""
|
|
|
+
|
|
|
+ def test_run(self):
|
|
|
+ out = WhateverIO()
|
|
|
+ with patch(
|
|
|
+ 'celery.loaders.base.BaseLoader.import_default_modules'
|
|
|
+ ) as import_default_modules:
|
|
|
+ with patch(
|
|
|
+ 'celery.app.base.Celery.bugreport'
|
|
|
+ ) as bugreport:
|
|
|
+ # Method call order mock obj
|
|
|
+ mco = Mock()
|
|
|
+ mco.attach_mock(import_default_modules, 'idm')
|
|
|
+ mco.attach_mock(bugreport, 'br')
|
|
|
+ a = report(app=self.app, stdout=out)
|
|
|
+ a.run()
|
|
|
+ calls = [call.idm(), call.br()]
|
|
|
+ mco.assert_has_calls(calls)
|