def test_get_pks_multiple(self):
Base = declarative_base()
class MyModel(Base):
__tablename__ = 'blah'
id = Column(Integer, primary_key=True)
pk = Column(String, primary_key=True)
value = Column(String)
self.assertAllIn(_get_pks(MyModel), ('id', 'pk',))
def test_get_pks_single(self):
Base = declarative_base()
class MyModel(Base):
__tablename__ = 'blah'
id = Column(Integer, primary_key=True)
value = Column(String)
resp = _get_pks(MyModel)
self.assertAllIn(('id',), resp)
def test_get_pks_single(self):
Base = declarative_base()
class MyModel(Base):
__tablename__ = 'blah'
id = Column(Integer, primary_key=True)
value = Column(String)
resp = _get_pks(MyModel)
self.assertAllIn(('id', ), resp)
def test_get_pks_multiple(self):
Base = declarative_base()
class MyModel(Base):
__tablename__ = 'blah'
id = Column(Integer, primary_key=True)
pk = Column(String, primary_key=True)
value = Column(String)
self.assertAllIn(_get_pks(MyModel), (
'id',
'pk',
))