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")
