def handle(self, *args, **kwargs): patch_sae_restful_mysql() kwargs.setdefault('database', DEFAULT_DB_ALIAS) super(Command, self).handle(*args, **kwargs)
def test_patch_sae_restful_mysql(self): settings.patch_sae_restful_mysql() self.assertEqual(sys.modules['MySQLdb'], sae._restful_mysql)