def save(response_elements):
session = DBSession()
network = LinkShareNetwork(**response_elements['network'])
if not session.query(LinkShareNetwork).filter_by(id=response_elements['network']['id']):
session.add(network)
merchant = LinkShareMerchant(network_id=network.id, **response_elements['merchant'])
session.add(merchant)
deal = LinkShareDeal(network_id=network.id, merchant_id=merchant.id, **response_elements['deal'])
deal.categories=[LinkShareCategory(**category) for category in response_elements['categories']]
deal.promotion_types=[LinkSharePromotionType(**promotion_type) for promotion_type in response_elements['promotion_types']]
session.add(deal)
session.commit()
class BaseTestCase(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.engine = create_engine('sqlite://', echo=False)
cls.connection = cls.engine.connect()
initialize_sql(cls.connection)
def setUp(self):
# begin a non-ORM transaction
self.trans = self.connection.begin()
# bind an individual Session to the connection
self.session = DBSession()
def tearDown(self):
self.trans.rollback()
self.session.close()
def setUp(self):
# begin a non-ORM transaction
self.trans = self.connection.begin()
# bind an individual Session to the connection
self.session = DBSession()
