def __init__(self, config=None, initialize=True): # FIXME: remove this 'initialize' check # it's used the tests, but I could simply override the class instead if initialize: if not config: config = Config() # FIXME: use nwu.server.app instead when it's available db_bind(config.get("database", "connection_string", "sqlite:////var/lib/nwu/nwu.db"))
from nwu.server.rpc_agents import RPC from nwu.server.db.model import metadata, db_bind from nwu.server.db.model import Computer, CurrentPackages from nwu.server.cli import Commands log = logging.getLogger() HDLR = logging.StreamHandler() log.addHandler(HDLR) log.setLevel(logging.DEBUG) CONFIG = { 'connection_string' : 'sqlite:///' } CLI = Commands(config=CONFIG, initialize=False) PASSWORD = '******' UNIQ = 'dsadsad1921832918312weee' TOKEN = hmac.new(PASSWORD, UNIQ).hexdigest() print "Conncted to database? %s" % repr(db_bind(CONFIG['connection_string'])) create_all() class TestServer: def test_add_computer(self): assert Local.check_token(UNIQ, TOKEN) == False # no servers in the database zero = Computer.query.count() assert zero == 0 # server doesn't exist yet session = RPC.session_setup(UNIQ, TOKEN) assert session == False