class XBasicsTestCase( TestCase ):
def setUp( self ):
self.app = CreateApp( "testing" )
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
def tearDown( self ):
db.session.remove()
db.drop_all()
self.app_context.pop()
def test_app_exists( self ):
self.assertFalse( current_app is None )
def test_app_is_testing( self ):
self.assertTrue( current_app.config[ "TESTING" ] )
import time
import psutil
import sys
import signal
from concurrent.futures import ThreadPoolExecutor
from app import (
app,
CreateApp,
)
from app.symbol import RunSymbol
from app.general.exchange_info import exchange_info
from system.db import Mysql
CreateApp()
async def startup():
print("-" * 5, " STARTUP ", "-" * 5)
return [await exchange_info()]
async def shutdown():
print("-" * 5, " SHUTDOWN ", "-" * 5)
await app.db.execute("TRUNCATE TABLE running")
async def active_symbols():
system_pids = [str(x) for x in psutil.pids()]
from app import CreateApp app = CreateApp()
# -*- coding: utf-8 -*- # @Time : 2018/4/24 下午2:21 # @author : Xmchx # @File : celery_worker.py from app import CreateApp from extensions import celery app = CreateApp() app.app_context().push()
import sys
from app import CreateApp
from time import sleep
from flask import Flask, render_template
import logging
app = CreateApp('config')
if __name__ == '__main__':
app.run(host=app.config['HOST'], port=app.config['PORT'], debug=app.config['DEBUG'])
import sys
from app import CreateApp
from time import sleep
from flask import Flask, render_template
import logging
app = None
if (sys.argv[1] == "db"):
appType = 0
else:
appType = int(sys.argv[1])
print(appType)
if appType == 0:
app = CreateApp('configclient', appType)
logging.basicConfig(filename='errorClient.txt', level=logging.DEBUG)
else:
app = CreateApp('configadmin', appType)
logging.basicConfig(filename='errorAdmin.txt', level=logging.DEBUG)
print(app.config['PORT'])
if __name__ == '__main__':
app.run(host=app.config['HOST'],
port=app.config['PORT'],
debug=app.config['DEBUG'])
from app import db, CreateApp
from models import *
app = CreateApp()
with app.app_context():
db.create_all()
def setUp( self ):
self.app = CreateApp( "testing" )
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
class XUserModelTestCase( TestCase ):
def setUp( self ):
self.app = CreateApp( "testing" )
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
XRole.InsertRoles()
def tearDown( self ):
db.session.remove()
db.drop_all()
self.app_context.pop()
def test_password_setter( self ):
u = XUser( password="******" )
self.assertTrue( u.password_hash is not None )
def test_no_password_getter( self ):
u = XUser( password="******" )
with self.assertRaises( AttributeError ):
u.password
def test_password_verification( self ):
u = XUser( password="******" )
self.assertTrue( u.VerifyPassword( "fox" ) )
self.assertFalse( u.VerifyPassword( "cat" ) )
def test_password_salts_are_random( self ):
u1 = XUser( password="******" )
u2 = XUser( password="******" )
self.assertTrue( u1.password_hash != u2.password_hash )
def test_valid_confirmation_token( self ):
u = XUser( password="******" )
db.session.add(u)
db.session.commit()
token = u.GenerateConfirmationToken()
self.assertTrue( u.Confirm( token ) )
def test_invalid_reset_token( self ):
u1 = XUser( password="******" )
u2 = XUser( password="******" )
db.session.add( u1 )
db.session.add( u2 )
token = u1.GenerateResetToken()
self.assertFalse( u2.ResetPassword( token, "horse" ) )
self.assertTrue( u2.VerifyPassword( "dog" ) )
def test_valid_email_change_token( self ):
u = XUser( email="*****@*****.**", password="******" )
db.session.add( u )
db.session.commit()
token = u.GenerateEmailChangeToken( "*****@*****.**" )
self.assertTrue( u.ChangeEmail( token ) )
self.assertTrue( u.email == "*****@*****.**" )
def test_invalid_email_change_token( self ):
u1 = XUser( email="*****@*****.**", password="******" )
u2 = XUser( email="*****@*****.**", password="******" )
db.session.add( u1 )
db.session.add( u2 )
db.session.commit()
token = u1.GenerateEmailChangeToken( "*****@*****.**" )
self.assertFalse( u2.ChangeEmail( token ) )
self.assertTrue( u2.email == "*****@*****.**" )
def test_duplicate_email_change_token( self ):
u1 = XUser( email="*****@*****.**", password="******" )
u2 = XUser( email="*****@*****.**", password="******" )
db.session.add( u1 )
db.session.add( u2 )
db.session.commit()
token = u2.GenerateEmailChangeToken( "*****@*****.**" )
self.assertFalse( u2.ChangeEmail( token ) )
self.assertTrue( u2.email == "*****@*****.**" )
def test_roles_and_permissions( self ):
u = XUser( email="*****@*****.**", password="******" )
self.assertTrue( u.Can( XPermission.WRITE_ARTICLES ) )
self.assertFalse( u.Can( XPermission.MODERATE_COMMENTS ) )
def test_anonymous_user( self ):
u = XAnonymousUser()
self.assertFalse( u.Can( XPermission.FOLLOW ) )
def test_timestamps( self ):
u = XUser( password="******" )
db.session.add( u )
db.session.commit()
self.assertTrue(
( datetime.utcnow() - u.member_since ).total_seconds() < 3
)
self.assertTrue(
(datetime.utcnow() - u.last_seen).total_seconds() < 3
)
def test_ping( self ):
u = XUser( password="******" )
db.session.add( u )
db.session.commit()
last_seen_before = u.last_seen
time.sleep( 2 )
u.Ping()
self.assertTrue( u.last_seen > last_seen_before )
