def setUp(self):
beacon.config['TESTING'] = True
logging.disable(logging.CRITICAL)
self.beacon = beacon.test_client()
# Create Records in DB
SQL_Base.metadata.create_all(sql_engine)
self.session = SQL_Session()
self.customer = Customer(name="Foo Bar", sites=["foo.com"])
self.session.add(self.customer)
self.session.commit()
self.beacon_url = "/beacon.js?id=%s&dl=http%%3A//foo.com" % self.customer.id
def get(cls, id):
"""Return Customer object with id"""
session = SQL_Session()
result = session.query(Customer).filter(Customer.id == id).first()
session.close()
return result
@classmethod
def get(cls, id):
"""Return Customer object with id"""
session = SQL_Session()
result = session.query(Customer).filter(Customer.id == id).first()
session.close()
return result
if __name__ == "__main__":
# Build DB
SQL_Base.metadata.create_all(sql_engine)
# Get a session object
session = SQL_Session()
session.add(Customer(name="Apple", sites=["apple.com"]))
session.add(Customer(name="Google", sites=["google.com", "google.co.uk"]))
session.add(Customer(name="Amazon"))
# Save everything
session.commit()
# Print everything
for row in session.query(Customer).all():
print row
# Close the session
session.close()
def setUp(self):
SQL_Base.metadata.create_all(sql_engine)
self.session = SQL_Session()
