Added dao, test framework and tests for dao.

This commit is contained in:
Nicholas Staples
2016-01-08 12:18:12 +00:00
parent a327702ad0
commit 5bcc615825
9 changed files with 118 additions and 17 deletions

View File

@@ -10,7 +10,6 @@ from utils import logging
db = SQLAlchemy()
api_user = LocalProxy(lambda: _request_ctx_stack.top.api_user)

View File

@@ -6,11 +6,11 @@ from app import db
from app.models import Service
def create_new_service(service_name,
user,
limit=1000,
active=False,
restricted=True):
def create_service(service_name,
user,
limit=1000,
active=False,
restricted=True):
service = Service(name=service_name,
created_at=datetime.now(),
limit=limit,
@@ -22,7 +22,10 @@ def create_new_service(service_name,
return service.id
def get_services(user, service_id=None):
def get_services(service_id=None, user_id=None):
# TODO need better mapping from function params to sql query.
if service_id:
return Service.query.filter_by(user=user, service_id=service_id).one()
return Service.query.filter_by(user=user).all()
return Service.query.filter_by(id=service_id).one()
elif user_id:
return Service.query.filter(Service.users.any(id=user_id)).all()
return Service.query.all()

View File

@@ -6,7 +6,7 @@ from app import db
from app.models import User
def create_new_user(email_address):
def create_user(email_address):
user = User(email_address=email_address,
created_at=datetime.now())
db.session.add(user)
@@ -16,5 +16,5 @@ def create_new_user(email_address):
def get_users(user_id=None):
if user_id:
return User.query.filter_by(user_id=user_id).one()
return User.query.filter_by(id=user_id).one()
return User.query.filter_by().all()