def test_set_values(self): s = SQLGenerator() self.assertEqual(s.set(['nom']), 'nom = %(nom)s') self.assertEqual(s.set(['nom', 'prenom']), 'nom = %(nom)s, prenom = %(prenom)s') params = {'nom': 'dupont', 'prenom': 'jean'} self.assertEqual(s.set(params), 'nom = %(nom)s, prenom = %(prenom)s') self.assertEqual(params, {'nom': 'dupont', 'prenom': 'jean'})
def test_set_functions(self): s = SQLGenerator() params = { 'nom': 'dupont', 'prenom': 'jean', 'age': SQLExpression('YEARS(%(date)s)', date='2013/01/01') } self.assertEqual( s.set(params), 'age = YEARS(%(date)s), nom = %(nom)s, prenom = %(prenom)s') self.assertEqual(params, { 'nom': 'dupont', 'prenom': 'jean', 'date': '2013/01/01' })