|  | @@ -54,12 +54,16 @@ def with_dist_not_in_path(fun):
 | 
	
		
			
				|  |  |  class Upgrade(object):
 | 
	
		
			
				|  |  |      old_modules = ("platform", )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -    def run(self):
 | 
	
		
			
				|  |  | -        path = self.detect_existing_installation()
 | 
	
		
			
				|  |  | +    def run(self, dist=False):
 | 
	
		
			
				|  |  | +        detect_ = self.detect_existing_installation
 | 
	
		
			
				|  |  | +        if not dist:
 | 
	
		
			
				|  |  | +            detect = with_dist_not_in_path(detect_)
 | 
	
		
			
				|  |  | +        else:
 | 
	
		
			
				|  |  | +            detect = lambda: detect_(distmeta)
 | 
	
		
			
				|  |  | +        path = detect()
 | 
	
		
			
				|  |  |          if path:
 | 
	
		
			
				|  |  |              self.remove_modules(path)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -    @with_dist_not_in_path
 | 
	
		
			
				|  |  |      def detect_existing_installation(self, celery=None):
 | 
	
		
			
				|  |  |          path = os.path.dirname(celery.__file__)
 | 
	
		
			
				|  |  |          sys.stderr.write("* Upgrading old Celery from: \n\t%r\n" % path)
 | 
	
	
		
			
				|  | @@ -78,13 +82,20 @@ class Upgrade(object):
 | 
	
		
			
				|  |  |              self.try_remove(os.path.join(path, "%s.pyc" % module_name))
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -class quicktest(test):
 | 
	
		
			
				|  |  | +class mytest(test):
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    def run(self, *args, **kwargs):
 | 
	
		
			
				|  |  | +        Upgrade().run(dist=True)
 | 
	
		
			
				|  |  | +        test.run(self, *args, **kwargs)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +class quicktest(mytest):
 | 
	
		
			
				|  |  |      extra_env = dict(SKIP_RLIMITS=1, QUICKTEST=1)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      def run(self, *args, **kwargs):
 | 
	
		
			
				|  |  |          for env_name, env_value in self.extra_env.items():
 | 
	
		
			
				|  |  |              os.environ[env_name] = str(env_value)
 | 
	
		
			
				|  |  | -        test.run(self, *args, **kwargs)
 | 
	
		
			
				|  |  | +        mytest.run(self, *args, **kwargs)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  class upgrade(install):
 | 
	
	
		
			
				|  | @@ -136,7 +147,7 @@ setup(
 | 
	
		
			
				|  |  |      zip_safe=False,
 | 
	
		
			
				|  |  |      install_requires=install_requires,
 | 
	
		
			
				|  |  |      tests_require=['nose', 'nose-cover3', 'unittest2', 'simplejson'],
 | 
	
		
			
				|  |  | -    cmdclass={"install": upgrade, "quicktest": quicktest},
 | 
	
		
			
				|  |  | +    cmdclass={"install": upgrade, "test": mytest, "quicktest": quicktest},
 | 
	
		
			
				|  |  |      test_suite="nose.collector",
 | 
	
		
			
				|  |  |      classifiers=[
 | 
	
		
			
				|  |  |          "Development Status :: 5 - Production/Stable",
 |