def init_db(): """Erase and creates a new database.""" with APP.app_context(): database = get_db() with APP.open_resource('schema.sql', mode='r') as squema: database.cursor().executescript(squema.read()) database.commit()
def send_mail(resetCode, email): with APP.app_context(): try: msg = Message("slackr passwordreset", sender="*****@*****.**", recipients=[email]) msg.body = resetCode mail.send(msg) return "SUCCESS" except Exception: return "ERROR"
def test_route_with_no_worker(self, mocker): """Test index route when there is no worker set.""" client = APP.test_client(mocker) url = '/' redis = mocker.MagicMock() mocker.patch.object(collector.utils, 'REDIS', redis) response = client.get(url) assert response.get_data() == \ b'{"message":"No worker set","status":"Error","version":"1.0"}\n' assert response.status_code == 500
def test_route_with_redis_absent(self, mocker): """Test index route when there is no redis.""" client = APP.test_client(mocker) worker = mocker.MagicMock() mocker.patch.object(collector, 'WORKER', worker) url = '/' response = client.get(url) assert response.get_data() == \ b'{"message":"Required service not operational",' \ b'"status":"Error","version":"1.0"}\n' assert response.status_code == 500
def test_route_with_redis_present(self, mocker): """Test index route when redis is present.""" client = APP.test_client(mocker) worker = mocker.MagicMock() mocker.patch.object(collector, 'WORKER', worker) url = '/' redis = mocker.MagicMock() mocker.patch.object(collector.utils, 'REDIS', redis) response = client.get(url) assert response.get_data() == \ b'{"message":"Up and Running","status":"OK","version":"1.0"}\n' assert response.status_code == 200
def setUp(self): """Set up test application client""" self.app = APP.test_client() self.app.testing = True
""" Точка входа в приложение """ from server import APP if __name__ == "__main__": APP.run()
def test_admin_get(self): response = APP.test_client().get('/admin') self.assertEqual(response.status_code, 200)
( 'admin', 'datex2', ), 'text/xml', ), ) urlsreqstation = [ '/jpglist.html', '/export/image.jpg', #'/export', '/export/vaisalaobs.xml', #'/datex2/content.xml', '/datex2/mst.xml', ] testApp = TestApp(APP.wsgifunc()) # Don't run on production... could do witha better test but 'meh' if dbconfig['host'] in ("localhost", "192.168.32.40"): raise PGDB = web.database(**dbconfig) @nottest def setup_passworddb(): transaction = PGDB.transaction() PGDB.delete('exportws.permissions', where="True") PGDB.delete('exportws.pwdb', where="True") values = [ {
#!/usr/bin/env python3 from server import APP if __name__ == "__main__": APP.run(debug=True, use_debugger=False, use_reloader=False)
""" Primary entry point for the app """ from server import APP if __name__ == "__main__": APP.run(host=APP.config["IP"], port=int(APP.config["PORT"]))
def setUp(self): app.config['TESTING'] = True self.app = app.test_client() self.app.testing = True