def test_basic_identifier(self):
l1 = lexer.Token('C', '-- :name identifier :1', at(1, 1))
l2 = lexer.Token('Q', 'select * from {{table_name}} limit 1', at(2, 1))
self.assertEqual([l1, l2],
lexer.lex(
open('tests/sql/identifier.sql', 'r').read(),
ctx))
def test_basic(self):
l1 = lexer.Token('C', '-- :name username_for_id :1', at(1, 1))
l2 = lexer.Token(
'Q', 'select username from users where user_id = :user_id',
at(2, 1))
self.assertEqual(
[l1, l2],
lexer.lex(open('tests/sql/basic.sql', 'r').read(), ctx))
def test_whitespace(self):
l1 = lexer.Token('C', '-- :name username_for_id :1', at(1, 2))
l2 = lexer.Token(
'Q', 'select username from users where user_id = :user_id',
at(2, 2))
self.assertEqual(
[l1, l2],
lexer.lex(
' -- :name username_for_id :1 \n'
' select username from users where user_id = :user_id ', ctx))
def test_blank_lines(self):
l1 = lexer.Token('C', '-- :name username_for_id :1', at(1, 1))
l2 = lexer.Token('Q', '', at(2, 1))
l3 = lexer.Token(
'Q', 'select username from users where user_id = :user_id',
at(3, 1))
self.assertEqual(
[l1, l2, l3],
lexer.lex(
'-- :name username_for_id :1 \n'
'\n'
'select username from users where user_id = :user_id ', ctx))
def test_basic(self):
self.assertEqual([
('C', '-- :name username_for_id :1'),
('Q', 'select username from users where user_id = :user_id'),
], lexer.lex(open('tests/sql/basic.sql', 'r').read()))
