From 483e2fcf41adcdddad5543c04d3ad42c60def334 Mon Sep 17 00:00:00 2001 From: chenjiankun Date: Tue, 17 Jan 2017 15:40:17 +0000 Subject: Add unittest framework for Yardstick API JIRA: YARDSTICK-538 Currently it is hard to test API, So I add a base class as flask document do. In this framework I will mock a temp sqlite database and a server. Change-Id: If881233cb22655617c07ad018201b8ee08492d06 Signed-off-by: chenjiankun --- api/database/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'api/database/__init__.py') diff --git a/api/database/__init__.py b/api/database/__init__.py index d7cf4f9c4..753b34684 100644 --- a/api/database/__init__.py +++ b/api/database/__init__.py @@ -13,10 +13,12 @@ from sqlalchemy import create_engine from sqlalchemy.orm import scoped_session, sessionmaker from sqlalchemy.ext.declarative import declarative_base +from yardstick.common import constants as consts + logger = logging.getLogger(__name__) logger.setLevel(logging.DEBUG) -engine = create_engine('sqlite:////tmp/yardstick.db', convert_unicode=True) +engine = create_engine(consts.SQLITE, convert_unicode=True) db_session = scoped_session(sessionmaker(autocommit=False, autoflush=False, bind=engine)) -- cgit 1.2.3-korg