Пример #1
0
 def setUp(self):
     self.assertTrue(
         current_app.config["TESTING"], "Testing is not set. Are you sure you are using the right config?"
     )
     current_app.config["WTF_CSRF_ENABLED"] = False
     ResetDB.drop_collections()
     PopulateDB.create_roles()
     PopulateDB.create_users()
     self.client = self.app.test_client()
Пример #2
0
 def setUp(self):
     self.assertTrue(
         current_app.config['TESTING'],
         'Testing is not set. Are you sure you are using the right config?')
     current_app.config['WTF_CSRF_ENABLED'] = False
     ResetDB.drop_collections()
     PopulateDB.create_roles()
     PopulateDB.create_users()
     self.client = self.app.test_client()
Пример #3
0
from flask.ext.script import Manager
from flask.ext.script.commands import Server, Shell, ShowUrls, Clean
from flask.ext.security.script import CreateUserCommand, AddRoleCommand,\
    RemoveRoleCommand, ActivateUserCommand, DeactivateUserCommand

from flask_application import app
from flask_application.script import ResetDB, PopulateDB
from flask_application.tests.script import RunTests


manager = Manager(app)
manager.add_command("shell", Shell(use_ipython=True))
manager.add_command("runserver", Server(use_reloader=True))
manager.add_command("show_urls", ShowUrls())
manager.add_command("clean", Clean())

manager.add_command("reset_db", ResetDB())
manager.add_command("populate_db", PopulateDB())

manager.add_command('create_user', CreateUserCommand())
manager.add_command('add_role', AddRoleCommand())
manager.add_command('remove_role', RemoveRoleCommand())
manager.add_command('deactivate_user', DeactivateUserCommand())
manager.add_command('activate_user', ActivateUserCommand())

manager.add_command('run_tests', RunTests())


if __name__ == "__main__":
    manager.run()
Пример #4
0
 def tearDown(self):
     ResetDB.drop_collections()
Пример #5
0
 def tearDown(self):
     ResetDB.drop_collections()