def test_module_file(self): m1 = Mock() m1.__file__ = '/opt/foo/xyz.pyc' self.assertEqual(module_file(m1), '/opt/foo/xyz.py') m2 = Mock() m2.__file__ = '/opt/foo/xyz.py' self.assertEqual(module_file(m1), '/opt/foo/xyz.py')
def test_prepare_prog_name(self): x = CeleryCommand(app=self.app) main = Mock(name='__main__') main.__file__ = '/opt/foo.py' with patch.dict(sys.modules, __main__=main): self.assertEqual(x.prepare_prog_name('__main__.py'), '/opt/foo.py') self.assertEqual(x.prepare_prog_name('celery'), 'celery')
def test_prepare_prog_name(self): x = CeleryCommand(app=self.app) main = Mock(name="__main__") main.__file__ = "/opt/foo.py" with patch.dict(sys.modules, __main__=main): self.assertEqual(x.prepare_prog_name("__main__.py"), "/opt/foo.py") self.assertEqual(x.prepare_prog_name("celery"), "celery")