sqlalchemy-pervasive/tests/test_base.py
2013-10-23 10:11:55 -07:00

43 lines
1.3 KiB
Python

from unittest import TestCase
from mock import Mock
from sqlalchemy.exc import InvalidRequestError
from sqlalchemy_pervasive import base
class PervasiveCompilerTests(TestCase):
def compiler(self):
return base.PervasiveCompiler(dialect=Mock(paramstyle='named'), statement=Mock())
def test_get_select_precolumns(self):
compiler = self.compiler()
select = Mock(_distinct=None, _limit=None, _offset=None)
self.assertEqual(compiler.get_select_precolumns(select), '')
select._distinct = True
self.assertEqual(compiler.get_select_precolumns(select), 'DISTINCT ')
select._distinct = False
select._limit = 10
self.assertEqual(compiler.get_select_precolumns(select), 'TOP 10 ')
select._limit = None
select._offset = 10
self.assertRaises(InvalidRequestError, compiler.get_select_precolumns, select)
def test_limit_clause(self):
compiler = self.compiler()
self.assertEqual(compiler.limit_clause(Mock()), '')
def test_visit_true(self):
compiler = self.compiler()
self.assertEqual(compiler.visit_true(Mock()), '1')
def test_visit_false(self):
compiler = self.compiler()
self.assertEqual(compiler.visit_false(Mock()), '0')