43 lines
1.3 KiB
Python
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')
|