def test_drop_database(self, connection_method):
connection_method.return_value = Mock(__enter__=Mock, __exit__=Mock())
with patch.object(rethink, 'db_drop', return_value=Mock()) as \
db_drop_method:
drop_database()
db_drop_method.assert_called_once_with(DEFAULT_DB_NAME)
def test_drop_database(self, connection_method):
connection_method.return_value = Mock(__enter__=Mock, __exit__=Mock())
with patch.object(rethink, 'db_drop', return_value=Mock()) as \
db_drop_method:
drop_database()
db_drop_method.assert_called_once()
db_drop_method.run.assert_called_once()
db_drop_method.assert_called_with(DEFAULT_DB_NAME)
def main(arguments=None): # pragma: no cover
parser = argparse.ArgumentParser(description='Run the PyWebHooks app')
parser.add_argument('--initdb', dest='initdb', action='store_true')
args = parser.parse_args(arguments)
if args.initdb:
print('Dropping database...')
try:
drop_database()
except Exception as ex:
print(str(ex))
print('Creating database...')
create_database()
print('Adding admin account')
print(create_admin_account())
print('Complete')
else:
app.run(debug=True, port=8081, host='0.0.0.0')
CELERY.conf.update(flask_app.config)
return flask_app
CELERY = Celery(__name__, broker=CELERY_BROKER_URL)
app = create_wsgi_app()
@app.before_request
def before_request():
if request.headers['content-type'].lower().find('application/json'):
return 'Unsupported Media Type', 415
if __name__ == '__main__': # pragma: no cover
parser = argparse.ArgumentParser(description='Run the PyWebHooks app')
parser.add_argument('--initdb', dest='initdb', action='store_true')
args = parser.parse_args()
if args.initdb:
print('Dropping database...')
drop_database()
print('Creating database...')
create_database()
print('Adding admin account')
print(create_admin_account())
print('Complete')
else:
app.run(debug=True, port=8081)
def initdb():
drop_database()
create_database()
return create_admin_account()
def test_drop_database_throws_rql_runtime_error(self, _):
with self.assertRaises(RqlRuntimeError) as cm:
drop_database()
self.assertEqual(cm.exception, RqlRuntimeError(None, None, None))
def test_drop_database_throws_rql_driver_error(self, _):
with self.assertRaises(RqlDriverError) as cm:
drop_database()
self.assertEqual(cm.exception, RqlDriverError(None))