예제 #1
0
 def get_full_count(self, session=None):
     """Returns the full size of the dataset."""
     q = SelectQuery(from_=self.class_.__table_name__, columns=('COUNT(*)',))
     session = get_session(session)
     cursor = session.cursor()
     cursor.execute(*q.render())
     return cursor.fetchall()[0][0]
예제 #2
0
 def test_with_columns(self):
     """Test a simplest query"""
     s = SelectQuery('eggs', columns=['id', 'name'])
     self.assertEqual(
         s.render(),
         ('SELECT id, name FROM eggs WHERE 1 = 1;',())
     )
예제 #3
0
 def test_joins(self):
     """Test if joins are rendered"""
     s = SelectQuery('eggs', joins=[
         ('LEFT', 'sausages', 'sausages.egg_id', 'eggs.id')
     ])
     self.assertEqual(
         s.render(),
         ('SELECT * FROM eggs LEFT JOIN sausages ON sausages.egg_id = '
         'eggs.id WHERE 1 = 1;',())
     )
예제 #4
0
 def test_simplest(self):
     """Test a simplest query"""
     s = SelectQuery('eggs')
     self.assertEqual(s.render(), ('SELECT * FROM eggs WHERE 1 = 1;',()))