openstack/deb-python-sqlalchemy-utils
Code Issues Proposed changes

Refactor tests

Browse Source
This commit is contained in:
Konsta Vesterinen
2014-05-14 15:47:22 +03:00
parent 278c9a55c7
commit 4714e013ba
2 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/functions/test_get_columns.py
View File

@@ -1,11 +1,14 @@
import sqlalchemy as sa
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy_utils import get_columns
from tests import TestCase
class TestGetColumns(TestCase):
def create_models(self):
class Building(self.Base):
class TestGetColumns(object):
def setup_method(self, method):
Base = declarative_base()
class Building(Base):
__tablename__ = 'building'
id = sa.Column('_id', sa.Integer, primary_key=True)
name = sa.Column('_name', sa.Unicode(255))

10
tests/functions/test_get_primary_keys.py
View File

@@ -3,13 +3,17 @@ try:
except ImportError:
from ordereddict import OrderedDict
import sqlalchemy as sa
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy_utils import get_primary_keys
from tests import TestCase
class TestGetPrimaryKeys(TestCase):
def create_models(self):
class Building(self.Base):
class TestGetPrimaryKeys(object):
def setup_method(self, method):
Base = declarative_base()
class Building(Base):
__tablename__ = 'building'
id = sa.Column('_id', sa.Integer, primary_key=True)
name = sa.Column('_name', sa.Unicode(255))