def assertEqualQueries(self, first_query, second_query, msg=None):
"""Fail if the two objects are unequal as determined by the '=='
operator.
"""
first_query = clean_query(first_query)
second_query = clean_query(second_query)
assertion_func = self._getAssertEqualityFunc(first_query, second_query)
assertion_func(first_query, second_query, msg=msg)
def test_from_table(self):
scenarios = [('foo', ), (
'foo',
'bar',
)]
for scenario in scenarios:
expected = clean_query('FROM ' + ', '.join(scenario))
statement = From(*scenario)
actual = clean_query(statement.parse())
self.assertEqual(expected, actual)
def test_statement(self):
statement = Statement('FROM', 'foo', 'bar').add('pip')
expected = clean_query('FROM foo, bar, pip')
actual = clean_query(statement.parse())
self.assertEqual(expected, actual)
def test_statement_with_sep(self):
statement = Statement('FROM', 'foo', 'bar', sep=' AND').add('pip')
expected = clean_query('FROM foo AND bar AND pip')
actual = clean_query(statement.parse())
self.assertEqual(expected, actual)
def test_from_table_add(self):
statement = From().add('foo').add('bar').add('pip')
actual = clean_query(statement.parse())
expected = clean_query('FROM foo, bar, pip')
self.assertEqual(expected, actual)